Add ORAN Python SDK first draft

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

diff --git a/smo-install/test/pythonsdk/unit-tests/test_enrichmentservice.py b/smo-install/test/pythonsdk/unit-tests/test_enrichmentservice.py
new file mode 100644 (file)
index 0000000..96fb9d3
--- /dev/null
+++ b/smo-install/test/pythonsdk/unit-tests/test_enrichmentservice.py
@@ -0,0 +1,125 @@
+#!/usr/bin/env python3
+# SPDX-License-Identifier: Apache-2.0
+"""Test Enrichment Service module."""
+
+from unittest import mock
+from oransdk.enrichmentservice.enrichmentservice import EnrichmentService
+
+HEADER = {"Content-Type": "application/json"}
+BASE_URL = "http://localhost:8083"
+
+def test_initialization():
+    """Class initialization test."""
+    ems = EnrichmentService()
+    assert isinstance(ems, EnrichmentService)
+
+
+@mock.patch.object(EnrichmentService, 'send_message')
+def test_check_status(mock_send_message):
+    """Test EnrichmentService's class method."""
+    EnrichmentService.check_status()
+    mock_send_message.assert_called_once_with('GET',
+                                              'Get EMS status',
+                                              (f"{BASE_URL}/status"))
+
+
+@mock.patch.object(EnrichmentService, 'send_message')
+def test_get_eitypes(mock_send_message):
+    """Test EnrichmentService's class method."""
+    EnrichmentService.get_eitypes()
+    mock_send_message.assert_called_once_with('GET',
+                                              'Get all the EiTypes',
+                                              (f"{BASE_URL}/data-producer/v1/info-types"),
+                                              headers=HEADER)
+
+
+@mock.patch.object(EnrichmentService, 'send_message')
+def test_get_eitype_individual(mock_send_message):
+    """Test EnrichmentService's class method."""
+    EnrichmentService.get_eitype_individual("test_eitype")
+    mock_send_message.assert_called_once_with('GET',
+                                              'Get individual EiType',
+                                              (f"{BASE_URL}/data-producer/v1/info-types/test_eitype"),
+                                              headers=HEADER)
+
+
+@mock.patch.object(EnrichmentService, 'send_message')
+def test_get_eiproducers(mock_send_message):
+    """Test EnrichmentService's class method."""
+    EnrichmentService.get_eiproducers()
+    mock_send_message.assert_called_once_with('GET',
+                                              'Get all the EiProducers',
+                                              (f"{BASE_URL}/data-producer/v1/info-producers"),
+                                              headers=HEADER)
+
+
+@mock.patch.object(EnrichmentService, 'send_message')
+def test_get_eiproducer_individual(mock_send_message):
+    """Test EnrichmentService's class method."""
+    EnrichmentService.get_eiproducer_individual("test_producer")
+    mock_send_message.assert_called_once_with('GET',
+                                              'Get individual EiProducer',
+                                              (f"{BASE_URL}/data-producer/v1/info-producers/test_producer"),
+                                              headers=HEADER)
+
+
+@mock.patch.object(EnrichmentService, 'send_message')
+def test_get_eiproducer_status(mock_send_message):
+    """Test EnrichmentService's class method."""
+    EnrichmentService.get_eiproducer_status("test_producer")
+    mock_send_message.assert_called_once_with('GET',
+                                              'Get the status of EiProducer',
+                                              (f"{BASE_URL}/data-producer/v1/info-producers/test_producer/status"),
+                                              headers=HEADER)
+
+
+@mock.patch.object(EnrichmentService, 'send_message')
+def test_get_eijobs(mock_send_message):
+    """Test EnrichmentService's class method."""
+    EnrichmentService.get_eijobs()
+    mock_send_message.assert_called_once_with('GET',
+                                              'Get all the EiJobs',
+                                              (f"{BASE_URL}/A1-EI/v1/eijobs"),
+                                              headers=HEADER)
+
+
+@mock.patch.object(EnrichmentService, 'send_message')
+def test_get_eijob_individual(mock_send_message):
+    """Test EnrichmentService's class method."""
+    EnrichmentService.get_eijob_individual("test_job")
+    mock_send_message.assert_called_once_with('GET',
+                                              'Get individual EiJob',
+                                              (f"{BASE_URL}/A1-EI/v1/eijobs/test_job"),
+                                              headers=HEADER)
+
+
+@mock.patch.object(EnrichmentService, 'send_message')
+def test_create_eitype(mock_send_message):
+    """Test EnrichmentService's class method."""
+    EnrichmentService.create_eitype("type_name", {})
+    mock_send_message.assert_called_once_with('PUT',
+                                              'Create EiType',
+                                              (f"{BASE_URL}/data-producer/v1/info-types/type_name"),
+                                              data={},
+                                              headers=HEADER)
+
+
+@mock.patch.object(EnrichmentService, 'send_message')
+def test_create_eiproducer(mock_send_message):
+    """Test EnrichmentService's class method."""
+    EnrichmentService.create_eiproducer("producer", {})
+    mock_send_message.assert_called_once_with('PUT',
+                                              'Create EiProducer',
+                                              (f"{BASE_URL}/data-producer/v1/info-producers/producer"),
+                                              data={},
+                                              headers=HEADER)
+
+@mock.patch.object(EnrichmentService, 'send_message')
+def test_create_eijob(mock_send_message):
+    """Test EnrichmentService's class method."""
+    EnrichmentService.create_eijob("job", {})
+    mock_send_message.assert_called_once_with('PUT',
+                                              'Create EiJob',
+                                              (f"{BASE_URL}/A1-EI/v1/eijobs/job"),
+                                              data={},
+                                              headers=HEADER)