示例#1
0
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"]) == {}