def setUp(self): self.maxDiff = None super(ServerActionsControllerTest, self).setUp() self.stubs = stubout.StubOutForTesting() fakes.stub_out_auth(self.stubs) self.stubs.Set(engine.db, 'instance_get', return_server_by_id) self.stubs.Set(engine.db, 'instance_get_by_uuid', return_server_by_uuid) self.stubs.Set(engine.db, 'instance_update', instance_update) fakes.stub_out_tank(self.stubs) fakes.stub_out_nw_api(self.stubs) fakes.stub_out_rate_limiting(self.stubs) self.snapshot = fakes.stub_out_compute_api_snapshot(self.stubs) service_class = 'engine.image.tank.TankImageService' self.service = utils.import_object(service_class) self.context = context.RequestContext(1, None) self.service.delete_all() self.sent_to_tank = {} fakes.stub_out_tank_add_image(self.stubs, self.sent_to_tank) self.flags(allow_instance_snapshots=True) self.uuid = FAKE_UUID self.url = '/v2/fake/servers/%s/action' % self.uuid self.controller = servers.Controller()
def setUp(self): super(ExtendedStatusTest, self).setUp() self.uuid = '70f6db34-de8d-4fbd-aafb-4065bdfa6114' self.url = '/v2/x7/servers/%s' % self.uuid fakes.stub_out_nw_api(self.stubs) self.flags(allow_admin_api=True) self.stubs.Set(compute.api.API, 'routing_get', fake_compute_get)
def _setup_mock_network_api(self): fakes.stub_out_nw_api(self.stubs)
def setUp(self): super(DiskConfigTestCase, self).setUp() self.flags(verbose=True) fakes.stub_out_nw_api(self.stubs) FAKE_INSTANCES = [ fakes.stub_instance(1, uuid=MANUAL_INSTANCE_UUID, auto_disk_config=False), fakes.stub_instance(2, uuid=AUTO_INSTANCE_UUID, auto_disk_config=True) ] def fake_instance_get(context, id_): for instance in FAKE_INSTANCES: if id_ == instance['id']: return instance self.stubs.Set(engine.db, 'instance_get', fake_instance_get) def fake_instance_get_by_uuid(context, uuid): for instance in FAKE_INSTANCES: if uuid == instance['uuid']: return instance self.stubs.Set(engine.db, 'instance_get_by_uuid', fake_instance_get_by_uuid) def fake_instance_get_all(context, *args, **kwargs): return FAKE_INSTANCES self.stubs.Set(engine.db, 'instance_get_all', fake_instance_get_all) self.stubs.Set(engine.db, 'instance_get_all_by_filters', fake_instance_get_all) def fake_instance_create(context, inst_, session=None): class FakeModel(dict): def save(self, session=None): pass inst = FakeModel(**inst_) inst['id'] = 1 inst['uuid'] = AUTO_INSTANCE_UUID inst['created_at'] = datetime.datetime(2010, 10, 10, 12, 0, 0) inst['updated_at'] = datetime.datetime(2010, 10, 10, 12, 0, 0) inst['progress'] = 0 inst['name'] = 'instance-1' # this is a property def fake_instance_get_for_create(context, id_, *args, **kwargs): return inst self.stubs.Set(engine.db, 'instance_get', fake_instance_get_for_create) self.stubs.Set(engine.db, 'instance_update', fake_instance_get_for_create) def fake_instance_get_all_for_create(context, *args, **kwargs): return [inst] self.stubs.Set(engine.db, 'instance_get_all', fake_instance_get_all_for_create) self.stubs.Set(engine.db, 'instance_get_all_by_filters', fake_instance_get_all_for_create) def fake_instance_add_security_group(context, instance_id, security_group_id): pass self.stubs.Set(engine.db, 'instance_add_security_group', fake_instance_add_security_group) return inst self.stubs.Set(engine.db, 'instance_create', fake_instance_create) app = v2.APIRouter() app = extensions.ExtensionMiddleware(app) app = wsgi.LazySerializationMiddleware(app) self.app = app
def setUp(self): super(BootFromVolumeTest, self).setUp() self.stubs.Set(engine.compute.API, 'create', fake_compute_api_create) fakes.stub_out_nw_api(self.stubs)