def test_list_for_standard_user(self):
user = factory.make_User()
handler = UserHandler(user, {})
# Other users
for _ in range(3):
factory.make_User()
self.assertItemsEqual([self.dehydrate_user(user)], handler.list({}))
def test_list_for_admin(self):
admin = factory.make_admin()
handler = UserHandler(admin, {}, None)
factory.make_User()
expected_users = [
self.dehydrate_user(user, for_self=(user == admin))
for user in User.objects.exclude(username__in=SYSTEM_USERS)
]
self.assertItemsEqual(expected_users, handler.list({}))
def test_list_with_sshkeys_and_machines(self):
user, keys = factory.make_user_with_keys()
num_machines = 3
for _ in range(num_machines):
node = factory.make_Node(status=NODE_STATUS.READY)
node.acquire(user)
handler = UserHandler(user, {}, None)
users_from_ws = handler.list({})
self.assertEqual([self.dehydrate_user(user, for_self=True)],
users_from_ws)
user_from_ws = users_from_ws[0]
self.assertEqual(user_from_ws["sshkeys_count"], len(keys))
self.assertEqual(user_from_ws["machines_count"], num_machines)
