Exemplo n.º 1
0
 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)
Exemplo n.º 2
0
 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)
Exemplo n.º 3
0
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)
Exemplo n.º 4
0
 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)
Exemplo n.º 5
0
 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)
Exemplo n.º 6
0
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)