def test_query_multiple_user_paged():
with mock.patch(
"umapi_client.connection.requests.Session.get") as mock_get:
mock_get.side_effect = [
MockResponse(
200, {
"result":
"success",
"lastPage":
False,
"users": [{
"name": "n1",
"type": "user"
}, {
"name": "n2",
"type": "user"
}]
}),
MockResponse(
200, {
"result":
"success",
"lastPage":
True,
"users": [{
"name": "n3",
"type": "user"
}, {
"name": "n4",
"type": "user"
}]
})
]
conn = Connection(**mock_connection_params)
assert conn.query_multiple("user", 0) == ([{
"name": "n1",
"type": "user"
}, {
"name": "n2",
"type": "user"
}], False, 0, 0, 1, 0)
assert conn.query_multiple("user", 1) == ([{
"name": "n3",
"type": "user"
}, {
"name": "n4",
"type": "user"
}], True, 0, 0, 1, 0)
def test_query_multiple_user_empty():
with mock.patch("umapi_client.connection.requests.Session.get") as mock_get:
mock_get.return_value = MockResponse(200, {"result": "success",
"lastPage": True,
"users": []})
conn = Connection(**mock_connection_params)
assert conn.query_multiple("user") == ([], True)
def test_query_multiple_user_success():
with mock.patch(
"umapi_client.connection.requests.Session.get") as mock_get:
mock_get.return_value = MockResponse(
200, {
"result":
"success",
"lastPage":
False,
"users": [{
"name": "n1",
"type": "user"
}, {
"name": "n2",
"type": "user"
}]
})
conn = Connection(**mock_connection_params)
assert conn.query_multiple("user") == ([{
"name": "n1",
"type": "user"
}, {
"name": "n2",
"type": "user"
}], False, 0, 0, 1, 0)
def test_query_multiple_usergroup_paged():
with mock.patch(
"umapi_client.connection.requests.Session.get") as mock_get:
mock_get.side_effect = [
MockResponse(
200, [{
"name": "n1",
"type": "user-group"
}, {
"name": "n2",
"type": "user-group"
}], {
"X-Total-Count": "4",
"X-Page-Count": "2",
"X-Current-Page": "1",
"X-Page-Size": "2"
}),
MockResponse(
200, [{
"name": "n3",
"type": "user-group"
}, {
"name": "n4",
"type": "user-group"
}], {
"X-Total-Count": "4",
"X-Page-Count": "2",
"X-Current-Page": "2",
"X-Page-Size": "2"
})
]
conn = Connection(**mock_connection_params)
assert conn.query_multiple("user-group", 0) == ([{
"name": "n1",
"type": "user-group"
}, {
"name": "n2",
"type": "user-group"
}], False, 4, 2, 1, 2)
assert conn.query_multiple("user-group", 1) == ([{
"name": "n3",
"type": "user-group"
}, {
"name": "n4",
"type": "user-group"
}], True, 4, 2, 2, 2)
def test_query_multiple_usergroup_empty():
with mock.patch("umapi_client.connection.requests.Session.get") as mock_get:
mock_get.return_value = MockResponse(200,
[],
{"X-Total-Count": "0",
"X-Page-Count": "1",
"X-Current-Page": "1",
"X-Page-Size:": "0"})
conn = Connection(**mock_connection_params)
assert conn.query_multiple("user-group") == ([], True)
def test_query_multiple_usergroup_success():
with mock.patch("umapi_client.connection.requests.Session.get") as mock_get:
mock_get.return_value = MockResponse(200,
[{"name": "n1", "type": "user-group"},
{"name": "n2", "type": "user-group"}],
{"X-Total-Count": "4",
"X-Page-Count": "2",
"X-Current-Page": "1",
"X-Page-Size:": "2"})
conn = Connection(**mock_connection_params)
assert conn.query_multiple("user-group") == ([{"name": "n1", "type": "user-group"},
{"name": "n2", "type": "user-group"}],
False)
def test_query_multiple_usergroup_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_multiple("user-group") == ([], True)
