コード例 #1
0
    def test_unequal_keys(self):
        epidb = DeepBlueClient(address="localhost", port=31415)
        self.init(epidb)
        # Note: this test has not 100% hit chance since the problem only
        # occured when two keys were generated within one second.
        # Though 3 users give a high chance for this to happen.

        res, u1 = epidb.add_user("user1", "*****@*****.**", "test",
                                 self.admin_key)
        self.assertSuccess(res, u1)
        res, u2 = epidb.add_user("user2", "*****@*****.**", "test",
                                 self.admin_key)
        self.assertSuccess(res, u2)
        res, u3 = epidb.add_user("user3", "*****@*****.**", "test",
                                 self.admin_key)
        self.assertSuccess(res, u3)

        self.assertTrue(u1[1] != u2[1])  # ensure different user keys
        self.assertTrue(u1[1] != u3[1])
        self.assertTrue(u2[1] != u3[1])

        res, users = epidb.list_users(self.admin_key)
        self.assertSuccess(res, users)

        user_names = epidb.extract_names(users)[1]

        self.assertEqual(len(users), 5)
        self.assertTrue(EPIDB_TEST_ADMIN[0] in user_names)
        self.assertTrue("user1" in user_names)
        self.assertTrue("user2" in user_names)
        self.assertTrue("user3" in user_names)
        self.assertTrue("anonymous" in user_names)
コード例 #2
0
    def test_users(self):
        epidb = DeepBlueClient(address="localhost", port=31415)
        self.init(epidb)

        res, user = epidb.add_user("user1", "*****@*****.**", "test",
                                   self.admin_key)
        self.assertSuccess(res, user)

        res, users = epidb.list_users(self.admin_key)
        self.assertSuccess(res, users)

        user_names = epidb.extract_names(users)[1]
        self.assertEqual(len(users), 3)  # the admin and the newly created user
        self.assertTrue(EPIDB_TEST_ADMIN[0] in user_names)
        self.assertTrue("user1" in user_names)
        self.assertTrue("anonymous" in user_names)