X-Git-Url: https://osm.etsi.org/gitweb/?a=blobdiff_plain;ds=sidebyside;f=osm_mon%2Ftests%2Funit%2Fcollector%2Futils%2Ftest_openstack.py;h=989c48668d87f59d0741fc6657f0850724619a9b;hb=0597c071b77a846624fbf4dc03c197029c1c1e59;hp=bf370767ead7b4cf91df4dd87b3617b3b75af41a;hpb=9ff497c1c8e3323cbf0c3a5414901ffd29b2bf18;p=osm%2FMON.git

diff --git a/osm_mon/tests/unit/collector/utils/test_openstack.py b/osm_mon/tests/unit/collector/utils/test_openstack.py
index bf37076..989c486 100644
--- a/osm_mon/tests/unit/collector/utils/test_openstack.py
+++ b/osm_mon/tests/unit/collector/utils/test_openstack.py
@@ -23,6 +23,7 @@
 from unittest import TestCase, mock
 
 from osm_mon.collector.utils.openstack import OpenstackUtils
+from osm_mon.core.exceptions import CertificateNotCreated
 
 
 @mock.patch("osm_mon.collector.utils.openstack.session")
@@ -70,3 +71,84 @@ class OpenstackUtilsTest(TestCase):
         mock_session.Session.assert_called_once_with(
             auth=mock.ANY, verify=True, timeout=10
         )
+
+    @mock.patch("osm_mon.collector.utils.openstack.OpenstackUtils._create_file_cert")
+    def test_session_with_ca_cert_content(self, mock_create_file_cert, mock_session):
+        creds = {
+            "_id": "1234",
+            "config": {"ca_cert_content": "test"},
+            "vim_url": "url",
+            "vim_user": "user",
+            "vim_password": "password",
+            "vim_tenant_name": "tenant_name",
+        }
+        mock_create_file_cert.return_value = {"ca_cert": "testfile"}
+        OpenstackUtils.get_session(creds)
+        mock_session.Session.assert_called_once_with(
+            auth=mock.ANY, verify="testfile", timeout=10
+        )
+
+    @mock.patch("osm_mon.collector.utils.openstack.makedirs", return_value="")
+    @mock.patch("osm_mon.collector.utils.openstack.path")
+    def test_create_file_cert(self, mock_path, mock_makedirs, mock_session):
+        vim_config = {"ca_cert_content": "test"}
+        target_id = "1234"
+        mock_path.isdir.return_value = False
+
+        with mock.patch("builtins.open", mock.mock_open()) as mocked_file:
+            OpenstackUtils._create_file_cert(vim_config, target_id)
+            mock_makedirs.assert_called_once_with("/app/osm_mon/certs/1234")
+            mocked_file.assert_called_once_with(
+                f"/app/osm_mon/certs/{target_id}/ca_cert", "w"
+            )
+            assert vim_config["ca_cert"] == f"/app/osm_mon/certs/{target_id}/ca_cert"
+
+    @mock.patch("osm_mon.collector.utils.openstack.makedirs")
+    @mock.patch("osm_mon.collector.utils.openstack.path")
+    def test_create_file_cert_exists(self, mock_path, mock_makedirs, mock_session):
+        vim_config = {"ca_cert_content": "test"}
+        target_id = "1234"
+        mock_path.isdir.return_value = True
+
+        with mock.patch("builtins.open", mock.mock_open()) as mocked_file:
+            OpenstackUtils._create_file_cert(vim_config, target_id)
+            mock_makedirs.assert_not_called()
+            mocked_file.assert_called_once_with(
+                f"/app/osm_mon/certs/{target_id}/ca_cert", "w"
+            )
+            assert vim_config["ca_cert"] == f"/app/osm_mon/certs/{target_id}/ca_cert"
+
+    @mock.patch("osm_mon.collector.utils.openstack.makedirs", side_effect=Exception)
+    @mock.patch("osm_mon.collector.utils.openstack.path")
+    def test_create_file_cert_makedirs_except(
+        self, mock_path, mock_makedirs, mock_session
+    ):
+        vim_config = {"ca_cert_content": "test"}
+        target_id = "1234"
+        mock_path.isdir.return_value = False
+
+        with mock.patch("builtins.open", mock.mock_open()) as mocked_file:
+            with self.assertRaises(CertificateNotCreated):
+                OpenstackUtils._create_file_cert(vim_config, target_id)
+            mock_makedirs.assert_called_once_with("/app/osm_mon/certs/1234")
+            mocked_file.assert_not_called()
+            assert vim_config["ca_cert_content"] == "test"
+
+    @mock.patch("osm_mon.collector.utils.openstack.makedirs", return_value="")
+    @mock.patch("osm_mon.collector.utils.openstack.path")
+    def test_create_file_cert_open_excepts(
+        self, mock_path, mock_makedirs, mock_session
+    ):
+        vim_config = {"ca_cert_content": "test"}
+        target_id = "1234"
+        mock_path.isdir.return_value = False
+
+        with mock.patch("builtins.open", mock.mock_open()) as mocked_file:
+            mocked_file.side_effect = Exception
+            with self.assertRaises(CertificateNotCreated):
+                OpenstackUtils._create_file_cert(vim_config, target_id)
+            mock_makedirs.assert_called_once_with("/app/osm_mon/certs/1234")
+            mocked_file.assert_called_once_with(
+                f"/app/osm_mon/certs/{target_id}/ca_cert", "w"
+            )
+            assert vim_config["ca_cert_content"] == "test"