From 4972bbe4ff284dae6404967dbe3c3d7250f16038 Mon Sep 17 00:00:00 2001
From: Marius <moprin@cloudbasesolutions.com>
Date: Thu, 11 Mar 2021 13:15:25 +0200
Subject: [PATCH] Add lib unit_tests

---
 unit_tests/test_lib_charm_openstack_magnum.py | 81 +++++++++++++++++++
 1 file changed, 81 insertions(+)
 create mode 100644 unit_tests/test_lib_charm_openstack_magnum.py

diff --git a/unit_tests/test_lib_charm_openstack_magnum.py b/unit_tests/test_lib_charm_openstack_magnum.py
new file mode 100644
index 0000000..e33e554
--- /dev/null
+++ b/unit_tests/test_lib_charm_openstack_magnum.py
@@ -0,0 +1,81 @@
+# Copyright 2021 Canonical Ltd
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#  http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import mock
+
+import charms_openstack.test_utils as test_utils
+
+import charm.openstack.magnum.magnum as magnum
+
+
+class Helper(test_utils.PatchHelper):
+
+    def setUp(self):
+        super().setUp()
+        self.patch_release(magnum.MagnumCharm.release)
+
+
+class TestMagnumCharmConfigProperties(Helper):
+
+    def test_magnum_password(self):
+        cls = mock.MagicMock()
+        self.patch_object(magnum, 'leadership')
+        magnum.magnum_password(cls)
+        self.leadership.leader_get.assert_called_with('magnum_password')
+
+    def test_ca_file_path(self):
+        cls = mock.MagicMock()
+        self.patch('os.path.exists', 'exists')
+        self.patch('os.path.join', 'ca_cert_file')
+        self.ca_cert_file.return_value = '/certs/magnum.crt'
+        ca_file = magnum.ca_file_path(cls)
+        self.assertEqual('/certs/magnum.crt', ca_file)
+
+
+class TestMagnumCharm(Helper):
+
+    def setUp(self):
+        super().setUp()
+        self.target = magnum.MagnumCharm()
+
+    def test_get_amqp_credentials(self):
+        cfg_data = {
+            "rabbit-user": "magnum",
+            "rabbit-vhost": "openstack",
+        }
+        self.target.config = cfg_data
+        result = self.target.get_amqp_credentials()
+        self.assertEqual(result, ('magnum', 'openstack'))
+
+    def test_get_database_setup(self):
+        cfg_data = {
+            "database-user": "magnum",
+            "database": "magnum",
+        }
+        self.target.config = cfg_data
+        result = self.target.get_database_setup()
+        self.assertEqual(result, [{'database': 'magnum',
+                                   'username': 'magnum'}])
+
+    def test_local_address(self):
+        configuration_class = mock.MagicMock()
+        self.target.configuration_class = configuration_class
+        self.assertEqual(self.target.local_address,
+                         configuration_class().local_address)
+
+    def test_local_unit_name(self):
+        configuration_class = mock.MagicMock()
+        self.target.configuration_class = configuration_class
+        self.assertEqual(self.target.local_unit_name,
+                         configuration_class().local_unit_name)