def test_mock_proxy_get(): with mock.patch("umapi_client.connection.requests.Session.get") as mock_get: mock_get.return_value = MockResponse(200, body=["test", "body"]) with mock.patch("umapi_client.connection.os.getenv") as mock_getenv: mock_getenv.return_value = "proxy" conn = Connection(**mock_connection_params) conn.make_call("").json() mock_get.assert_called_with('http://test/', auth='N/A', timeout=120.0)
def test_mock_playback_get(): with mock.patch("umapi_client.connection.requests.Session.get") as mock_get: mock_get.return_value = MockResponse(200, body=["test", "body"]) with mock.patch("umapi_client.connection.os.getenv") as mock_getenv: mock_getenv.return_value = "playback" conn = Connection(**mock_connection_params) conn.make_call("").json() assert mock_get.call_args[0][0] == 'http://test/' assert isinstance(mock_get.call_args[1]['auth'], Auth)
def test_post_success_test_mode(): with mock.patch("umapi_client.connection.requests.Session.post") as mock_post: mock_post.return_value = MockResponse(200, body=["test", "body"]) conn = Connection(test_mode=True, **mock_connection_params) assert conn.make_call("", [3, 5]).json() == ["test", "body"]
def test_get_success(): with mock.patch("umapi_client.connection.requests.Session.get") as mock_get: mock_get.return_value = MockResponse(200, body=["test", "body"]) conn = Connection(**mock_connection_params) assert conn.make_call("").json() == ["test", "body"]