1 files changed, 104 insertions, 0 deletions
diff --git a/charms/trusty/cassandra/tests/test_definitions.py b/charms/trusty/cassandra/tests/test_definitions.py
new file mode 100755
index 0000000..98103c0
--- /dev/null
+++ b/charms/trusty/cassandra/tests/test_definitions.py
@@ -0,0 +1,104 @@
+#!.venv3/bin/python3
+
+# Copyright 2015 Canonical Ltd.
+#
+# This file is part of the Cassandra Charm for Juju.
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License version 3, as
+# published by the Free Software Foundation.
+#
+# This program is distributed in the hope that it will be useful, but
+# WITHOUT ANY WARRANTY; without even the implied warranties of
+# MERCHANTABILITY, SATISFACTORY QUALITY, or FITNESS FOR A PARTICULAR
+# PURPOSE.  See the GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+from itertools import chain
+import functools
+import unittest
+from unittest.mock import patch
+
+from charmhelpers.core import hookenv
+from charmhelpers.core.services import ServiceManager
+
+from tests.base import TestCaseBase
+
+import definitions
+
+
+patch = functools.partial(patch, autospec=True)
+
+
+class TestDefinitions(TestCaseBase):
+    def test_get_service_definitions(self):
+        # We can't really test this in unit tests, but at least we can
+        # ensure the basic data structure is returned and accepted.
+        defs = definitions.get_service_definitions()
+        self.assertIsInstance(defs, list)
+        for d in defs:
+            with self.subTest(d=d):
+                self.assertIsInstance(d, dict)
+
+    def test_get_service_definitions_open_ports(self):
+        config = hookenv.config()
+        defs = definitions.get_service_definitions()
+        expected_ports = set([config['rpc_port'],
+                              config['native_transport_port'],
+                              config['storage_port'],
+                              config['ssl_storage_port']])
+        opened_ports = set(chain(*(d.get('ports', []) for d in defs)))
+        self.assertSetEqual(opened_ports, expected_ports)
+
+    def test_get_service_manager(self):
+        self.assertIsInstance(definitions.get_service_manager(),
+                              ServiceManager)
+
+    @patch('helpers.get_unit_superusers')
+    @patch('helpers.is_decommissioned')
+    @patch('helpers.is_cassandra_running')
+    def test_requires_live_node(self, is_running, is_decommissioned, get_sup):
+        is_decommissioned.return_value = False  # Is not decommissioned.
+        is_running.return_value = True  # Is running.
+        get_sup.return_value = set([hookenv.local_unit()])  # Creds exist.
+
+        self.assertTrue(bool(definitions.RequiresLiveNode()))
+
+    @patch('helpers.get_unit_superusers')
+    @patch('helpers.is_decommissioned')
+    @patch('helpers.is_cassandra_running')
+    def test_requires_live_node_decommissioned(self, is_running,
+                                               is_decommissioned, get_sup):
+        is_decommissioned.return_value = True  # Is decommissioned.
+        is_running.return_value = True  # Is running.
+        get_sup.return_value = set([hookenv.local_unit()])  # Creds exist.
+
+        self.assertFalse(bool(definitions.RequiresLiveNode()))
+
+    @patch('helpers.get_unit_superusers')
+    @patch('helpers.is_decommissioned')
+    @patch('helpers.is_cassandra_running')
+    def test_requires_live_node_down(self, is_running,
+                                     is_decommissioned, get_sup):
+        is_decommissioned.return_value = False  # Is not decommissioned.
+        is_running.return_value = False  # Is not running.
+        get_sup.return_value = set([hookenv.local_unit()])  # Creds exist.
+
+        self.assertFalse(bool(definitions.RequiresLiveNode()))
+
+    @patch('helpers.get_unit_superusers')
+    @patch('helpers.is_decommissioned')
+    @patch('helpers.is_cassandra_running')
+    def test_requires_live_node_creds(self, is_running,
+                                      is_decommissioned, get_sup):
+        is_decommissioned.return_value = False  # Is not decommissioned.
+        is_running.return_value = True  # Is running.
+        get_sup.return_value = set()  # Creds do not exist.
+
+        self.assertFalse(bool(definitions.RequiresLiveNode()))
+
+
+if __name__ == '__main__':
+    unittest.main(verbosity=2)