Refactor class structure for unit tests

This commit is contained in:
James Page 2017-07-10 16:42:07 +01:00
parent 929f9a377d
commit cb0b925679
1 changed files with 2 additions and 5 deletions

View File

@ -67,10 +67,10 @@ class TestRegisteredHooks(test_utils.TestRegisteredHooks):
self.registered_hooks_test_helper(handlers, hook_set, defaults)
class TestHandlerBase(test_utils.PatchHelper):
class TestHandlers(test_utils.PatchHelper):
def setUp(self):
super(TestHandlerBase, self).setUp()
super(TestHandlers, self).setUp()
self.gnocchi_charm = mock.MagicMock()
self.patch_object(handlers.charm, 'provide_charm_instance',
new=mock.MagicMock())
@ -78,9 +78,6 @@ class TestHandlerBase(test_utils.PatchHelper):
self.gnocchi_charm
self.provide_charm_instance().__exit__.return_value = None
class TestHandlers(TestHandlerBase):
def test_render_stuff(self):
handlers.render_config('arg1', 'arg2')
self.gnocchi_charm.render_with_interfaces.assert_called_once_with(