def test_raise_exception_if_trying_to_set_non_handler(): with pytest.raises(TypeError) as e: set_default_request_handler("not a handler") assert str(e.value) == "Attempted to set an invalid request handler as default."
def test_get_set_default_handler(): h = RequestHandler('') set_default_request_handler(h) assert get_default_request_handler() == h ubersmith.api._DEFAULT_REQUEST_HANDLER = None
def init(base_url, username=None, password=None, verify=True): """Initialize ubersmith API module with HTTP request handler.""" handler = RequestHandler(base_url, username, password, verify) set_default_request_handler(handler) return handler