def setUp(self): super(FreshInstanceTasksTest, self).setUp() when(taskmanager_models.FreshInstanceTasks).id().thenReturn( "instance_id") when(taskmanager_models.FreshInstanceTasks).tenant_id().thenReturn( "tenant_id") when(taskmanager_models.FreshInstanceTasks).hostname().thenReturn( "hostname") when(taskmanager_models.FreshInstanceTasks).name().thenReturn('name') taskmanager_models.FreshInstanceTasks.nova_client = fake_nova_client() taskmanager_models.CONF = mock() when(taskmanager_models.CONF).get(any()).thenReturn('') self.orig_ISS_find_by = InstanceServiceStatus.find_by self.orig_DBI_find_by = DBInstance.find_by self.userdata = "hello moto" self.guestconfig_content = "guest config" with NamedTemporaryFile(suffix=".cloudinit", delete=False) as f: self.cloudinit = f.name f.write(self.userdata) with NamedTemporaryFile(delete=False) as f: self.guestconfig = f.name f.write(self.guestconfig_content) self.freshinstancetasks = taskmanager_models.FreshInstanceTasks( None, None, None, None)
def setUp(self): super(FreshInstanceTasksTest, self).setUp() mock_instance = patch('trove.instance.models.FreshInstance') mock_instance.start() self.addCleanup(mock_instance.stop) mock_instance.id = Mock(return_value='instance_id') mock_instance.tenant_id = Mock(return_value="tenant_id") mock_instance.hostname = Mock(return_value="hostname") mock_instance.name = Mock(return_value='name') mock_instance.nova_client = Mock( return_value=fake_nova_client()) mock_datastore_v = patch( 'trove.datastore.models.DatastoreVersion') mock_datastore_v.start() self.addCleanup(mock_datastore_v.stop) mock_datastore = patch( 'trove.datastore.models.Datastore') mock_datastore.start() self.addCleanup(mock_datastore.stop) taskmanager_models.FreshInstanceTasks.nova_client = fake_nova_client() self.orig_ISS_find_by = InstanceServiceStatus.find_by self.orig_DBI_find_by = DBInstance.find_by self.userdata = "hello moto" self.guestconfig_content = "guest config" with NamedTemporaryFile(suffix=".cloudinit", delete=False) as f: self.cloudinit = f.name f.write(self.userdata) with NamedTemporaryFile(delete=False) as f: self.guestconfig = f.name f.write(self.guestconfig_content) self.freshinstancetasks = taskmanager_models.FreshInstanceTasks( None, Mock(), None, None)
def setUp(self): super(FreshInstanceTasksTest, self).setUp() mock_instance = patch('trove.instance.models.FreshInstance') mock_instance.start() self.addCleanup(mock_instance.stop) mock_instance.id = Mock(return_value='instance_id') mock_instance.tenant_id = Mock(return_value="tenant_id") mock_instance.hostname = Mock(return_value="hostname") mock_instance.name = Mock(return_value='name') mock_instance.nova_client = Mock( return_value=fake_nova_client()) mock_datastore_v = patch( 'trove.datastore.models.DatastoreVersion') mock_datastore_v.start() self.addCleanup(mock_datastore_v.stop) mock_datastore = patch( 'trove.datastore.models.Datastore') mock_datastore.start() self.addCleanup(mock_datastore.stop) taskmanager_models.FreshInstanceTasks.nova_client = fake_nova_client() self.orig_ISS_find_by = InstanceServiceStatus.find_by self.orig_DBI_find_by = DBInstance.find_by self.userdata = "hello moto" self.guestconfig_content = "guest config" with NamedTemporaryFile(mode="w", suffix=".cloudinit", delete=False) as f: self.cloudinit = f.name f.write(self.userdata) with NamedTemporaryFile(mode="w", delete=False) as f: self.guestconfig = f.name f.write(self.guestconfig_content) self.freshinstancetasks = taskmanager_models.FreshInstanceTasks( None, Mock(), None, None) self.tm_sg_create_inst_patch = patch.object( trove.taskmanager.models.SecurityGroup, 'create_for_instance', Mock(return_value={'id': uuid.uuid4(), 'name': uuid.uuid4()})) self.tm_sg_create_inst_mock = self.tm_sg_create_inst_patch.start() self.addCleanup(self.tm_sg_create_inst_patch.stop) self.tm_sgr_create_sgr_patch = patch.object( trove.taskmanager.models.SecurityGroupRule, 'create_sec_group_rule') self.tm_sgr_create_sgr_mock = self.tm_sgr_create_sgr_patch.start() self.addCleanup(self.tm_sgr_create_sgr_patch.stop) self.task_models_conf_patch = patch('trove.taskmanager.models.CONF') self.task_models_conf_mock = self.task_models_conf_patch.start() self.addCleanup(self.task_models_conf_patch.stop) self.inst_models_conf_patch = patch('trove.instance.models.CONF') self.inst_models_conf_mock = self.inst_models_conf_patch.start() self.addCleanup(self.inst_models_conf_patch.stop)
def setUp(self): super(FreshInstanceTasksTest, self).setUp() when(taskmanager_models.FreshInstanceTasks).id().thenReturn( "instance_id") when(taskmanager_models.FreshInstanceTasks).hostname().thenReturn( "hostname") taskmanager_models.FreshInstanceTasks.nova_client = fake_nova_client() taskmanager_models.CONF = mock() self.userdata = "hello moto" with NamedTemporaryFile(suffix=".cloudinit", delete=False) as f: self.cloudinit = f.name f.write(self.userdata) self.freshinstancetasks = taskmanager_models.FreshInstanceTasks( None, None, None, None)