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"