def get_destination_user():
"""Return the user to which resources should be assigned."""
real_users = get_real_users()
if real_users.count() == 1:
return get_one(real_users)
else:
bootstrap_user = get_bootstrap_node_owner()
if bootstrap_user is None:
return get_legacy_user()
else:
return bootstrap_user
def test_returns_None_if_invalid_yaml(self):
invalid_content = '%'.encode('ascii')
factory.make_FileStorage(
filename=PROVIDER_STATE_FILENAME, content=invalid_content)
self.assertIsNone(get_bootstrap_node_owner())
def test_returns_None_if_node_does_not_exist(self):
node = factory.make_Node(owner=factory.make_User())
make_provider_state_file(node=node)
node.delete()
self.assertIsNone(get_bootstrap_node_owner())
def test_returns_owner_if_node_found(self):
node = factory.make_Node(owner=factory.make_User())
make_provider_state_file(node=node)
self.assertEqual(node.owner, get_bootstrap_node_owner())
def test_returns_None_if_no_provider_state_file(self):
self.assertIsNone(get_bootstrap_node_owner())
def test_returns_None_if_invalid_yaml(self):
invalid_content = '%'.encode('ascii')
factory.make_file_storage(
filename=PROVIDER_STATE_FILENAME, content=invalid_content)
self.assertIsNone(get_bootstrap_node_owner())
def test_returns_None_if_node_does_not_exist(self):
node = factory.make_node(owner=factory.make_user())
make_provider_state_file(node=node)
node.delete()
self.assertIsNone(get_bootstrap_node_owner())
def test_returns_owner_if_node_found(self):
node = factory.make_node(owner=factory.make_user())
make_provider_state_file(node=node)
self.assertEqual(node.owner, get_bootstrap_node_owner())
def test_returns_None_if_no_provider_state_file(self):
self.assertIsNone(get_bootstrap_node_owner())
