def test_upgrade_charm(self, configs, mock_update_image_location_policy):
     self.remove_old_packages.return_value = False
     self.filter_installed_packages.return_value = ['test']
     relations.upgrade_charm()
     self.apt_install.assert_called_with(['test'], fatal=True)
     self.assertTrue(configs.write_all.called)
     self.assertTrue(self.reinstall_paste_ini.called)
     self.assertTrue(mock_update_image_location_policy.called)
     self.assertTrue(self.remove_old_packages.called)
 def test_upgrade_charm(self, configs, mock_update_image_location_policy):
     self.remove_old_packages.return_value = False
     self.filter_installed_packages.return_value = ['test']
     relations.upgrade_charm()
     self.apt_install.assert_called_with(['test'], fatal=True)
     self.assertTrue(configs.write_all.called)
     self.assertTrue(self.reinstall_paste_ini.called)
     self.assertTrue(mock_update_image_location_policy.called)
     self.assertTrue(self.remove_old_packages.called)
 def test_upgrade_charm_purge(self, configs,
                              mock_update_image_location_policy):
     self.services.return_value = ['glance-api']
     self.remove_old_packages.return_value = True
     self.filter_installed_packages.return_value = ['test']
     relations.upgrade_charm()
     self.apt_install.assert_called_with(['test'], fatal=True)
     self.assertTrue(configs.write_all.called)
     self.reinstall_paste_ini.assert_called_once_with(force_reinstall=True)
     self.assertTrue(mock_update_image_location_policy.called)
     self.assertTrue(self.remove_old_packages.called)
     self.service_restart.assert_called_once_with('glance-api')
 def test_upgrade_charm_purge(self, configs,
                              mock_update_image_location_policy):
     self.services.return_value = ['glance-api']
     self.remove_old_packages.return_value = True
     self.filter_installed_packages.return_value = ['test']
     relations.upgrade_charm()
     self.apt_install.assert_called_with(['test'], fatal=True)
     self.assertTrue(configs.write_all.called)
     self.reinstall_paste_ini.assert_called_once_with(force_reinstall=True)
     self.assertTrue(mock_update_image_location_policy.called)
     self.assertTrue(self.remove_old_packages.called)
     self.service_restart.assert_called_once_with('glance-api')
 def test_upgrade_charm(self, cluster_changed):
     relations.upgrade_charm()
     cluster_changed.assert_called_with()