summaryrefslogtreecommitdiffstats
path: root/charms/trusty/cassandra/tests/test_definitions.py
diff options
context:
space:
mode:
authorStuart Mackie <wsmackie@juniper.net>2016-10-07 12:24:58 -0700
committerStuart Mackie <wsmackie@juniper.net>2016-10-07 12:24:58 -0700
commit4faa7f927149a5c4ef7a03523f7bc14523cb9baa (patch)
tree0be55aa0809cc395e45baeae63db660b4e72fe83 /charms/trusty/cassandra/tests/test_definitions.py
parent82f1a7eb5535b30a95b1e71ff18c315d40d1e6f0 (diff)
Charms for Contrail 3.1 with Mitaka
Change-Id: Id37f3b9743d1974e31fcd7cd9c54be41bb0c47fb Signed-off-by: Stuart Mackie <wsmackie@juniper.net>
Diffstat (limited to 'charms/trusty/cassandra/tests/test_definitions.py')
-rwxr-xr-xcharms/trusty/cassandra/tests/test_definitions.py104
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)