diff options
| -rw-r--r-- | unit_tests/test_hooks.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/unit_tests/test_hooks.py b/unit_tests/test_hooks.py index edfbae9..c95e480 100644 --- a/unit_tests/test_hooks.py +++ b/unit_tests/test_hooks.py @@ -343,7 +343,7 @@ class MongoHooksTest(CharmTestCase): call1 = call('juju-local:27017', 'juju-remote:27017') mock_leave_replset.assert_has_calls([call1]) - + @patch('time.sleep') @patch.object(hooks, 'Connection') @patch.object(hooks, 'unit_get') @@ -352,10 +352,10 @@ class MongoHooksTest(CharmTestCase): def test_replica_set_relation_broken(self, mock_am_i_primary, mock_leave_replset, mock_unit_get, mock_Connection, mock_sleep): - + mock_am_i_primary.return_value = False hooks.replica_set_relation_broken() - self.assertEqual(1, mock_leave_replset.call_count) + self.assertEqual(1, mock_leave_replset.call_count) mock_am_i_primary.reset_mock() mock_leave_replset.reset_mock() mock_am_i_primary.return_value = True |