def test_upgrade_charm(self, getgrnam): grp_mock = MagicMock() grp_mock.gr_gid = None getgrnam.return_value = grp_mock self.remove_old_packages.return_value = False hooks.upgrade_charm() self.remove_old_packages.assert_called_once_with() self.assertFalse(self.service_restart.called)
def test_upgrade_charm_purge(self, getgrnam): grp_mock = MagicMock() grp_mock.gr_gid = None getgrnam.return_value = grp_mock self.remove_old_packages.return_value = True self.services.return_value = ['nova-compute'] hooks.upgrade_charm() self.remove_old_packages.assert_called_once_with() self.service_restart.assert_called_once_with('nova-compute')