Ejemplo n.º 1
0
    def test_user_holders(self):
        user_holder = UsersHolder()

        user1 = User(uid="test0", name="test0", secret="1111")
        user2 = User(uid="test1", name="test1", secret="1111")

        # add user 1
        # add user 2
        user_holder.add_user(user=user1)
        user_holder.add_user(user=user2)

        # get all users
        success, all_users_dict = user_holder.get_all()

        assert success
        assert len(all_users_dict) == 2, "unexpected size"
        print "[test_user_holders] test case 1 finished"

        # get specific user
        success, user_get_1 = user_holder.get_user(uid=user1.uid)
        assert success, "user_uid not found"
        assert user1.uid == user_get_1.uid
        print "[test_user_holders] test case 2 finished"

        # update some user
        user1.icon = "wakawaka"
        user_holder.update_user(user=user1)

        success, user_get_1 = user_holder.get_user(uid=user1.uid)
        assert success, "user_uid not found"
        assert user1.uid == user_get_1.uid and user_get_1.icon == user1.icon, "not equal"
        print "[test_user_holders] test case 3 finished"

        # remove one user

        remove_success, info = user_holder.remove_user(user_uid=user2.uid)

        get_success, all_users_dict = user_holder.get_all()

        assert remove_success
        assert len(all_users_dict) == 1, "unexpected size"
        print "[test_user_holders] test case 4 finished"