def test_show_checkpoint_not_allowed(self, mock_provider): mock_provider.return_value = fakes.FakeProvider() context = mock.MagicMock(project_id='fake_project_id_1', is_admin=False) self.assertRaises(oslo_messaging.ExpectedException, self.pro_manager.show_checkpoint, context, 'provider1', 'fake_checkpoint')
def test_show_checkpoint_not_found(self, mock_provider, mock_cp_collection_get): mock_provider.return_value = fakes.FakeProvider() context = mock.MagicMock() mock_cp_collection_get.side_effect = exception.CheckpointNotFound() self.assertRaises(oslo_messaging.ExpectedException, self.pro_manager.show_checkpoint, context, 'provider1', 'non_existent_checkpoint')
def test_checkpoint_state_reset_with_wrong_checkpoint_state( self, mock_provider): fake_provider = fakes.FakeProvider() fake_checkpoint = fakes.FakeCheckpoint() fake_checkpoint.status = 'deleting' fake_provider.get_checkpoint = mock.MagicMock( return_value=fake_checkpoint) mock_provider.return_value = fake_provider context = mock.MagicMock(project_id='fake_project_id', is_admin=True) self.assertRaises(oslo_messaging.ExpectedException, self.pro_manager.reset_state, context, 'fake_project_id', 'fake_checkpoint_id', 'error')
def test_checkpoint_state_reset_with_access_not_allowed( self, mock_provider): fake_provider = fakes.FakeProvider() fake_checkpoint = fakes.FakeCheckpoint() fake_provider.get_checkpoint = mock.MagicMock( return_value=fake_checkpoint) mock_provider.return_value = fake_provider context = mock.MagicMock(project_id='fake_project_id_01', is_admin=False) self.assertRaises(oslo_messaging.ExpectedException, self.pro_manager.reset_state, context, 'fake_project_id', 'fake_checkpoint_id', 'error')
def test_checkpoint_state_reset(self, mock_provider): fake_provider = fakes.FakeProvider() fake_checkpoint = fakes.FakeCheckpoint() fake_checkpoint.commit = mock.MagicMock() fake_provider.get_checkpoint = mock.MagicMock( return_value=fake_checkpoint) mock_provider.return_value = fake_provider context = mock.MagicMock(project_id='fake_project_id', is_admin=True) self.pro_manager.reset_state(context, 'provider1', 'fake_checkpoint', 'error') self.assertEqual(fake_checkpoint.status, 'error') self.assertEqual(True, fake_checkpoint.commit.called)
def test_restore_with_project_id_not_same(self, mock_provider): mock_provider.return_value = fakes.FakeProvider() context = mock.MagicMock(project_id='fake_project_id_1', is_admin=False) fake_restore = { 'checkpoint_id': 'fake_checkpoint', 'provider_id': 'fake_provider_id', 'parameters': None } self.assertRaises(oslo_messaging.ExpectedException, self.pro_manager.restore, context, fake_restore, None)
def setUp(self): super(ResourceFlowTest, self).setUp() self.resource_graph = { parent: [child], child: [grandchild], grandchild: [], } self.provider = fakes.FakeProvider() self.test_graph = graph.build_graph([parent], self.resource_graph.__getitem__) self.taskflow_engine = TaskFlowEngine()
def test_list_checkpoints_with_all_tenants(self, mock_provider): fake_provider = fakes.FakeProvider() fake_provider.list_checkpoints = mock.MagicMock() mock_provider.return_value = fake_provider context = mock.MagicMock(project_id='fake_project_id') self.pro_manager.list_checkpoints(context, 'provider1', filters={}, all_tenants=True) fake_provider.list_checkpoints.assert_called_once_with( 'fake_project_id', 'provider1', limit=None, marker=None, plan_id=None, start_date=None, end_date=None, sort_dir=None, context=context, all_tenants=True)
def test_show_checkpoint(self, mock_provider): mock_provider.return_value = fakes.FakeProvider() context = mock.MagicMock() cp = self.pro_manager.show_checkpoint(context, 'provider1', 'fake_checkpoint') self.assertEqual(cp['id'], 'fake_checkpoint')
def test_protect(self, mock_provider): mock_provider.return_value = fakes.FakeProvider() self.pro_manager.protect(None, fakes.fake_protection_plan())
def test_protect(self, mock_provider, mock_operation_log_create, mock_operation_log_update): mock_provider.return_value = fakes.FakeProvider() self.pro_manager.protect(None, fakes.fake_protection_plan())