def test_migrate_all_files_to_new_legacy_user_when_multiple_users(self):
stored = factory.make_file_storage(owner=None)
user1 = factory.make_user()
user2 = factory.make_user()
migrate()
stored = reload_object(stored)
self.assertNotIn(stored.owner, {user1, user2, None})
def test_migrate_ancillary_data_to_legacy_user_when_multiple_users(self):
factory.make_FileStorage(owner=None)
# Create two users, both with API credentials, an SSH key and a node.
user1 = factory.make_User()
consumer1, token1 = user1.userprofile.create_authorisation_token()
key1 = factory.make_SSHKey(user1, get_ssh_key_string(1))
node1 = factory.make_Node(owner=user1)
user2 = factory.make_User()
consumer2, token2 = user2.userprofile.create_authorisation_token()
key2 = factory.make_SSHKey(user2, get_ssh_key_string(2))
node2 = factory.make_Node(owner=user2)
migrate()
# The SSH keys have been copied to the legacy user.
legacy_user = get_legacy_user()
legacy_users_ssh_keys = get_ssh_keys(legacy_user)
self.assertSetEqual({key1.key, key2.key}, set(legacy_users_ssh_keys))
# The API credentials have been moved to the legacy user.
legacy_users_nodes = Node.objects.filter(owner=legacy_user)
self.assertSetEqual({node1, node2}, set(legacy_users_nodes))
self.assertEqual(
(legacy_user, legacy_user, legacy_user, legacy_user),
(
reload_object(consumer1).user,
reload_object(token1).user,
reload_object(consumer2).user,
reload_object(token2).user,
),
)
def test_migrate_all_nodes_to_new_legacy_user_when_multiple_users(self):
factory.make_file_storage(owner=None)
user1 = factory.make_user()
node1 = factory.make_node(owner=user1)
user2 = factory.make_user()
node2 = factory.make_node(owner=user2)
migrate()
self.assertNotIn(reload_object(node1).owner, {user1, user2, None})
self.assertNotIn(reload_object(node2).owner, {user1, user2, None})
def test_migrate_all_nodes_to_bootstrap_owner_when_multiple_users(self):
user1 = factory.make_User()
node1 = factory.make_Node(owner=user1)
user2 = factory.make_User()
node2 = factory.make_Node(owner=user2)
make_provider_state_file(node1)
migrate()
self.assertEqual(
(user1, user1),
(reload_object(node1).owner, reload_object(node2).owner))
def test_migrate_all_files_to_single_user_when_only_one_user(self):
user = factory.make_user()
stored = factory.make_file_storage(owner=None)
migrate()
self.assertEqual(user, reload_object(stored).owner)
def test_migrate_runs_when_no_unowned_files_exist(self):
factory.make_file_storage(owner=factory.make_user())
migrate()
def test_migrate_runs_when_no_files_exist(self):
migrate()
