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())
