Unit test updates

unit_tests/test_swift_storage_relations.py

@@ -41,6 +41,8 @@ TO_PATCH = [
     'register_configs',
     'update_nrpe_config',
     'get_ipv6_addr',
+    'assess_status',
+    'status_set',
 ]
 
 

unit_tests/test_swift_storage_utils.py

@@ -27,6 +27,7 @@ TO_PATCH = [
     'service_restart',
     '_save_script_rc',
     'lsb_release',
+    'status_set',
 ]
 
 
@@ -335,3 +336,14 @@ class SwiftStorageUtilsTests(CharmTestCase):
                     swift_utils.OBJECT_SVCS)
         for service in services:
             self.assertIn(call(service), self.service_restart.call_args_list)
+
+    @patch.object(swift_utils, 'relation_ids')
+    def test_assess_status(self, relation_ids):
+        relation_ids.return_value = []
+        swift_utils.assess_status()
+        self.status_set.assert_called_with('blocked',
+                                           'Missing relation: proxy')
+
+        relation_ids.return_value = ['swift-storage:1']
+        swift_utils.assess_status()
+        self.status_set.assert_called_with('active', 'Unit is ready')