test_session.py 文件源码

def setUp(self):

        self.adapter = requests_mock.Adapter()
        self.aos = Session(server=Config.test_server)
        self.aos.api.requests.mount('https://%s' % Config.test_server, self.adapter)

        def get_api_version(request, context):
            context.status_code = 200
            return dict(version=Config.test_server_api_version)

        self.adapter.register_uri('GET', '/api/versions/api', json=get_api_version)

        def get_build_version(request, context):
            context.status_code = 200
            return dict(version=Config.test_server_build_version)

        self.adapter.register_uri('GET', '/api/version', json=get_build_version)

        self.adapter.register_uri(
            'POST', '/api/user/login', json=dict(token=Config.test_auth_token))

        # generally enable the probe step to pass
        patch.object(self.aos.api, 'probe', return_value=True).start()