X-Git-Url: https://osm.etsi.org/gitweb/?p=osm%2FN2VC.git;a=blobdiff_plain;f=n2vc%2Ftests%2Funit%2Ftest_k8s_juju_conn.py;h=f4543801bbb33c3e4d3ecf994e658ddd7ee2a5a2;hp=b182d8c01fac1c2578d3fdcc608fe83ec2dd878b;hb=f8c25c9ceeeca8ac91caa62faa402a8b6dab02e1;hpb=a71d4a04c1e8ad3ffe1a129024e6dbc14d6d3bd5

diff --git a/n2vc/tests/unit/test_k8s_juju_conn.py b/n2vc/tests/unit/test_k8s_juju_conn.py
index b182d8c..f454380 100644
--- a/n2vc/tests/unit/test_k8s_juju_conn.py
+++ b/n2vc/tests/unit/test_k8s_juju_conn.py
@@ -789,3 +789,31 @@ class GetCredentialsTest(K8sJujuConnTestCase):
         self.k8s_juju_conn.db.get_one.assert_called_once()
         self.k8s_juju_conn.db.encrypt_decrypt_fields.assert_called_once()
         mock_safe_dump.assert_called_once()
+
+
+class UpdateVcaStatusTest(K8sJujuConnTestCase):
+    def setUp(self):
+        super(UpdateVcaStatusTest, self).setUp()
+        self.vcaStatus = {"model": {"applications": {"app": {"actions": {}}}}}
+        self.kdu_name = "kdu_name"
+        self.kdu_instance = "{}-{}".format(self.kdu_name, "id")
+        self.k8s_juju_conn.libjuju.get_executed_actions = AsyncMock()
+        self.k8s_juju_conn.libjuju.get_actions = AsyncMock()
+        self.k8s_juju_conn.libjuju.get_application_configs = AsyncMock()
+
+    def test_success(self):
+        self.loop.run_until_complete(self.k8s_juju_conn.update_vca_status(
+            self.vcaStatus, self.kdu_instance))
+        self.k8s_juju_conn.libjuju.get_executed_actions.assert_called_once()
+        self.k8s_juju_conn.libjuju.get_actions.assert_called_once()
+        self.k8s_juju_conn.libjuju.get_application_configs.assert_called_once()
+
+    def test_exception(self):
+        self.k8s_juju_conn.libjuju.get_model.return_value = None
+        self.k8s_juju_conn.libjuju.get_executed_actions.side_effect = Exception()
+        with self.assertRaises(Exception):
+            self.loop.run_until_complete(self.k8s_juju_conn.update_vca_status(
+                self.vcaStatus, self.kdu_instance))
+            self.k8s_juju_conn.libjuju.get_executed_actions.assert_not_called()
+            self.k8s_juju_conn.libjuju.get_actions.assert_not_called_once()
+            self.k8s_juju_conn.libjuju.get_application_configs.assert_not_called_once()