← Back to branch summary

~hopem/charms/trusty/glance/lp1518975

« back to all changes in this revision

Viewing changes to unit_tests/test_glance_utils.py

  • Committer: David Ames
  • Date: 2016-01-19 16:48:53 UTC
  • mfrom: (158.2.3 glance)
  • Revision ID: david.ames@canonical.com-20160119164853-1mhlu5vfkcrxyr3s
[tinwood,r=thedac] Fixes Bug#1526511  change pause/resume actions use (new) assess_status()

Show diffs side-by-side

added added

removed removed

Lines of Context:
unit_tests/test_glance_utils.py
293
293
            call('glance-registry'),
294
294
        ]
295
295
        self.assertEquals(service_restart.call_args_list, expected)
 
296
 
 
297
    @patch.object(utils, 'HookData')
 
298
    @patch.object(utils, 'kv')
 
299
    def test_is_paused(self, kv, HookData):
 
300
        """test_is_paused: Test is_paused() returns value
 
301
        from kv('unit-paused')"""
 
302
        HookData()().return_value = True
 
303
        kv().get.return_value = True
 
304
        self.assertEqual(utils.is_paused(), True)
 
305
        kv().get.assert_called_with('unit-paused')
 
306
        kv().get.return_value = False
 
307
        self.assertEqual(utils.is_paused(), False)
 
308
 
 
309
    @patch.object(utils, 'is_paused')
 
310
    @patch.object(utils, 'status_set')
 
311
    def test_assess_status(self, status_set, is_paused):
 
312
        """test_assess_status: verify that it does pick the right status"""
 
313
        # check that paused status does the right thing
 
314
        is_paused.return_value = True
 
315
        utils.assess_status(None)
 
316
        status_set.assert_called_with(
 
317
            "maintenance",
 
318
            "Paused. Use 'resume' action to resume normal service.")
 
319
 
 
320
        # if it isn't paused, the assess_status() calls
 
321
        # set_os_workload_status()
 
322
        is_paused.return_value = False
 
323
        with patch.object(utils, 'set_os_workload_status') \
 
324
                as set_os_workload_status:
 
325
            utils.assess_status("TEST CONFIG")
 
326
            set_os_workload_status.assert_called_with(
 
327
                "TEST CONFIG",
 
328
                utils.REQUIRED_INTERFACES,
 
329
                charm_func=utils.check_optional_relations)