def test_remove_role(self):
"""Test assigning Role to a Plan."""
self.assertThat('delete', tutils.IsMethodOn(self.api))
self.api.delete = mock.Mock(return_value=('resp', 'fake_plan_data'))
self.assertThat('resource_class', tutils.IsMethodOn(self.pm))
self.pm.resource_class = mock.Mock(return_value='fake_plan')
self.assertEqual(self.pm.remove_role('42', role_uuid='qwert12345'),
'fake_plan')
self.api.delete.assert_called_with('/v2/plans/42/roles/qwert12345')
self.pm.resource_class.assert_called_with(self.pm, 'fake_plan_data')
def test_delete(self):
"""Test deleting/removing an plan via DELETE."""
self.assertThat('_delete', tutils.IsMethodOn(self.pm))
self.pm._delete = mock.Mock(return_value=None)
self.assertEqual(self.pm.delete(42), None)
self.pm._delete.assert_called_with('/v2/plans/42')
def test_list(self):
"""Test retrieving a list of plans via GET."""
self.assertThat('_list', tutils.IsMethodOn(self.pm))
self.pm._list = mock.Mock(return_value=['fake_plan'])
self.assertEqual(self.pm.list(), ['fake_plan'])
self.pm._list.assert_called_with('/v2/plans')
def test_get_404(self):
"""Test a 404 response to a standard GET."""
self.assertThat('_get', tutils.IsMethodOn(self.pm))
self.pm._get = mock.Mock(return_value=None)
self.assertEqual(self.pm.get('fake_plan'), None)
self.pm._get.assert_called_with('/v2/plans/fake_plan')
def test_get(self):
"""Test a standard GET operation to read/retrieve the plan."""
self.assertThat('_get', tutils.IsMethodOn(self.pm))
self.pm._get = mock.Mock(return_value='fake_plan')
self.assertEqual(self.pm.get('fake_plan'), 'fake_plan')
self.pm._get.assert_called_with('/v2/plans/fake_plan')
def test_templates(self):
"""Test a GET operation to retrieve the plan's templates."""
self.assertThat('_get', tutils.IsMethodOn(self.pm))
self.pm._get = mock.MagicMock()
self.pm._get.return_value.to_dict.return_value = 'fake_templates_dict'
self.assertEqual(self.pm.templates('fake_plan'), 'fake_templates_dict')
self.pm._get.assert_called_with('/v2/plans/fake_plan/templates')
def test_add_role(self):
"""Test assigning Role to a Plan."""
self.assertThat('_post', tutils.IsMethodOn(self.pm))
self.pm._post = mock.Mock(return_value='dummy plan')
self.assertEqual(self.pm.add_role('42', role_uuid='qwert12345'),
'dummy plan')
self.pm._post.assert_called_with('/v2/plans/42/roles',
{'uuid': 'qwert12345'})
def test_create(self):
"""Test creating a new plan via POST."""
self.assertThat('_post', tutils.IsMethodOn(self.pm))
self.pm._post = mock.Mock(return_value=['fake_plan'])
self.assertEqual(self.pm.create(dummy='dummy plan data'),
['fake_plan'])
self.pm._post.assert_called_with('/v2/plans',
{'dummy': 'dummy plan data'})
def test_roles_subresource(self):
self.assertThat('_get', tutils.IsMethodOn(self.pm))
self.pm._get = mock.Mock(return_value=plans.Plan(
None, {'roles': [{
'name': 'foo_role'
}, {
'name': 'bar_role'
}]}))
test_roles = self.pm.get('42').roles
self.assertTrue(isinstance(test_roles, list))
self.assertTrue(isinstance(test_roles[0], roles.Role))
def test_patch(self):
"""Test patching a plan."""
self.assertThat('_patch', tutils.IsMethodOn(self.pm))
self.pm._patch = mock.Mock(return_value=['fake_plan'])
self.assertEqual(
self.pm.patch('42', [{
'name': 'dummy',
'value': 'dummy plan data'
}]), ['fake_plan'])
self.pm._patch.assert_called_with('/v2/plans/42',
[{
'name': 'dummy',
'value': 'dummy plan data'
}])