Add ORAN Python SDK first draft

[it/dep.git] / smo-install / test / pythonsdk / unit-tests / test_a1sim.py

diff --git a/smo-install/test/pythonsdk/unit-tests/test_a1sim.py b/smo-install/test/pythonsdk/unit-tests/test_a1sim.py
new file mode 100644 (file)
index 0000000..77f83bf
--- /dev/null
+++ b/smo-install/test/pythonsdk/unit-tests/test_a1sim.py
@@ -0,0 +1,49 @@
+#!/usr/bin/env python3
+# SPDX-License-Identifier: Apache-2.0
+"""Test A1sim module."""
+
+from unittest import mock
+from oransdk.a1sim.a1sim import A1sim
+
+HEADER = {"Accept": "application/json", "Content-Type": "application/json"}
+BASE_URL = "http://localhost:8081"
+
+def test_initialization():
+    """Class initialization test."""
+    a1sim = A1sim()
+    assert isinstance(a1sim, A1sim)
+
+
+@mock.patch.object(A1sim, 'send_message')
+def test_check_version(mock_send_message):
+    """Test A1sim's class method."""
+    A1sim.check_version(BASE_URL)
+    mock_send_message.assert_called_once_with('GET',
+                                              'Get ric version',
+                                              (f"{BASE_URL}/counter/interface"))
+
+@mock.patch.object(A1sim, 'send_message')
+def test_check_status(mock_send_message):
+    """Test A1sim's class method."""
+    A1sim.check_status(BASE_URL)
+    mock_send_message.assert_called_once_with('GET',
+                                              'Get ric status',
+                                              (f"{BASE_URL}"))
+
+@mock.patch.object(A1sim, 'send_message')
+def test_get_policy_number(mock_send_message):
+    """Test A1sim's class method."""
+    A1sim.get_policy_number(BASE_URL)
+    mock_send_message.assert_called_once_with('GET',
+                                              'Get policy numbers for ric',
+                                              (f"{BASE_URL}/counter/num_instances"))
+
+@mock.patch.object(A1sim, 'send_message')
+def test_create_policy_type(mock_send_message):
+    """Test A1sim's class method."""
+    A1sim.create_policy_type(BASE_URL, "test_id", {})
+    mock_send_message.assert_called_once_with('PUT',
+                                              'Create Policy Type',
+                                              (f"{BASE_URL}/policytype?id=test_id"),
+                                              data={},
+                                              headers=HEADER)