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)