From 3aedbacc5f56b96fd19e5db9090bbd9db43c154e Mon Sep 17 00:00:00 2001
From: Hui Xiang <hui.xiang@canonical.com>
Date: Fri, 16 Jan 2015 14:47:50 +0800
Subject: [PATCH] Add more unit tests.

---
 hooks/quantum_hooks.py           |  2 +-
 unit_tests/test_quantum_hooks.py | 25 ++++++++++++++++++++++++-
 2 files changed, 25 insertions(+), 2 deletions(-)

diff --git a/hooks/quantum_hooks.py b/hooks/quantum_hooks.py
index 8d32636b..19409b17 100755
--- a/hooks/quantum_hooks.py
+++ b/hooks/quantum_hooks.py
@@ -292,7 +292,7 @@ def ha_relation_joined():
 @hooks.hook('ha-relation-departed')
 def ha_relation_destroyed():
     # If e.g. we want to upgrade to Juno and use native Neutron HA support then
-    #  we need to un-corosync-cluster to enable the transition.
+    # we need to un-corosync-cluster to enable the transition.
     if config('ha-legacy-mode'):
         delete_legacy_resources()
         remove_legacy_ha_files()
diff --git a/unit_tests/test_quantum_hooks.py b/unit_tests/test_quantum_hooks.py
index b814d2c7..56c3bd28 100644
--- a/unit_tests/test_quantum_hooks.py
+++ b/unit_tests/test_quantum_hooks.py
@@ -44,7 +44,12 @@ TO_PATCH = [
     'create_sysctl',
     'update_nrpe_config',
     'update_legacy_ha_files',
-    'add_hostname_to_hosts'
+    'add_hostname_to_hosts',
+    'install_legacy_ha_files',
+    'cache_env_data',
+    'get_hacluster_config',
+    'delete_legacy_resources',
+    'remove_legacy_ha_files'
 ]
 
 
@@ -263,3 +268,21 @@ class TestQuantumHooks(CharmTestCase):
     def test_stop(self):
         self._call_hook('stop')
         self.assertTrue(self.stop_services.called)
+
+    def test_ha_relation_joined(self):
+        self.test_config.set('ha-legacy-mode', True)
+        self._call_hook('ha_relation_joined')
+        self.assertTrue(self.cache_env_data.called)
+        self.assertTrue(self.get_hacluster_config.called)
+        self.assertTrue(self.install_legacy_ha_files.called)
+
+    def test_ha_relation_departed(self):
+        self.test_config.set('ha-legacy-mode', True)
+        self._call_hook('ha-relation-departed')
+        self.assertTrue(self.remove_legacy_ha_files.called)
+        self.assertTrue(self.delete_legacy_resources.called)
+
+    def test_quantum_network_service_relation_changed(self):
+        self.test_config.set('ha-legacy-mode', True)
+        self._call_hook('quantum-network-service-relation-changed')
+        self.assertTrue(self.cache_env_data.called)