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())