def test_query_single_success(): with mock.patch( "umapi_client.connection.requests.Session.get") as mock_get: mock_get.return_value = MockResponse(200, { "result": "success", "user": { "name": "n1", "type": "user" } }) conn = Connection(**mock_connection_params) assert conn.query_single("user", ["n1"]) == { "name": "n1", "type": "user" }
def test_query_single_not_found(): with mock.patch("umapi_client.connection.requests.Session.get") as mock_get: mock_get.return_value = MockResponse(404, text="404 Object not found") conn = Connection(**mock_connection_params) assert conn.query_single("user", ["n1"]) == {}