def test_get_targets(self): user1 = UserFactory.build(merged_by=None) user2 = UserFactory.build(merged_by=user1) user3 = UserFactory.build() user1.save() user2.save() user3.save() user_list = get_targets() assert user_list is not None assert len(user_list) is 1 user1.merged_by = user3 user1.save() user_list = get_targets() assert len(user_list) is 2
def test_do_migration(self): user1 = UserFactory.build(merged_by=None) user2 = UserFactory.build(merged_by=user1, verification_key="key1") user3 = UserFactory.build(merged_by=user1, verification_key="key2") user2.email_verifications['token'] = {'email': '*****@*****.**'} user3.email_verifications['token'] = {'email': '*****@*****.**'} user1.save() user2.save() user3.save() user_list = get_targets() do_migration(user_list) user2.reload() user3.reload() assert user2.username is None assert user2.password is None assert len(user2.email_verifications) is 0 assert user2.verification_key is None assert user3.username is None assert user3.password is None assert len(user3.email_verifications) is 0 assert user3.verification_key is None