def test_aggregate_host_add_to_db(self): result = _create_aggregate(self.context, metadata=None) host = _get_fake_hosts(1)[0] aggregate_obj._host_add_to_db(self.context, result['id'], host) expected = aggregate_obj._aggregate_get_from_db(self.context, result['id']) self.assertEqual([_get_fake_hosts(1)[0]], expected.hosts)
def create_aggregate(context, db_id): fake_aggregate = _get_fake_aggregate(db_id, in_api=False, result=False) aggregate_obj._aggregate_create_in_db(context, fake_aggregate, metadata=_get_fake_metadata(db_id)) for host in _get_fake_hosts(db_id): aggregate_obj._host_add_to_db(context, fake_aggregate['id'], host)
def test_aggregate_host_re_add_to_db(self): result = _create_aggregate_with_hosts(self.context, metadata=None) host = _get_fake_hosts(1)[0] aggregate_obj._host_delete_from_db(self.context, result['id'], host) aggregate_obj._host_add_to_db(self.context, result['id'], host) expected = _aggregate_host_get_all(self.context, result['id']) self.assertEqual(len(expected), 2)
def create_aggregate(context, db_id, in_api=True): if in_api: fake_aggregate = _get_fake_aggregate(db_id, in_api=False, result=False) aggregate_obj._aggregate_create_in_db(context, fake_aggregate, metadata=_get_fake_metadata(db_id)) for host in _get_fake_hosts(db_id): aggregate_obj._host_add_to_db(context, fake_aggregate['id'], host) else: fake_aggregate = _get_fake_aggregate(db_id, in_api=False, result=False) db.aggregate_create(context, fake_aggregate, metadata=_get_fake_metadata(db_id)) for host in _get_fake_hosts(db_id): db.aggregate_host_add(context, fake_aggregate['id'], host)