def test_timeout_is_configurable(self, requests_session_mock): _wb = WBAPI(self._api_uri, connect_timeout=1, read_timeout=2) requests_session = requests_session_mock() _wb.get_apps() requests_session.get.assert_called_with("{}api/apps/".format( self._api_uri), timeout=(1, 2))
def setUp(self, requests_session_mock): self._api_uri = "http://dummy/" self._wb = WBAPI(self._api_uri) self._requests_session = requests_session_mock()
def get_api_before_request(): g.wb_api = WBAPI(current_app.config["API_URI"]) try: g.wb_api.set_token(session["api_token"]) except KeyError: pass