2 # SPDX-License-Identifier: Apache-2.0
3 """Test Enrichment Service module."""
5 from unittest import mock
6 from oransdk.enrichmentservice.enrichmentservice import EnrichmentService
8 HEADER = {"Content-Type": "application/json"}
9 BASE_URL = "http://localhost:8083"
11 def test_initialization():
12 """Class initialization test."""
13 ems = EnrichmentService()
14 assert isinstance(ems, EnrichmentService)
17 @mock.patch.object(EnrichmentService, 'send_message')
18 def test_check_status(mock_send_message):
19 """Test EnrichmentService's class method."""
20 EnrichmentService.check_status()
21 mock_send_message.assert_called_once_with('GET',
23 (f"{BASE_URL}/status"))
26 @mock.patch.object(EnrichmentService, 'send_message')
27 def test_get_eitypes(mock_send_message):
28 """Test EnrichmentService's class method."""
29 EnrichmentService.get_eitypes()
30 mock_send_message.assert_called_once_with('GET',
31 'Get all the EiTypes',
32 (f"{BASE_URL}/data-producer/v1/info-types"),
36 @mock.patch.object(EnrichmentService, 'send_message')
37 def test_get_eitype_individual(mock_send_message):
38 """Test EnrichmentService's class method."""
39 EnrichmentService.get_eitype_individual("test_eitype")
40 mock_send_message.assert_called_once_with('GET',
41 'Get individual EiType',
42 (f"{BASE_URL}/data-producer/v1/info-types/test_eitype"),
46 @mock.patch.object(EnrichmentService, 'send_message')
47 def test_get_eiproducers(mock_send_message):
48 """Test EnrichmentService's class method."""
49 EnrichmentService.get_eiproducers()
50 mock_send_message.assert_called_once_with('GET',
51 'Get all the EiProducers',
52 (f"{BASE_URL}/data-producer/v1/info-producers"),
56 @mock.patch.object(EnrichmentService, 'send_message')
57 def test_get_eiproducer_individual(mock_send_message):
58 """Test EnrichmentService's class method."""
59 EnrichmentService.get_eiproducer_individual("test_producer")
60 mock_send_message.assert_called_once_with('GET',
61 'Get individual EiProducer',
62 (f"{BASE_URL}/data-producer/v1/info-producers/test_producer"),
66 @mock.patch.object(EnrichmentService, 'send_message')
67 def test_get_eiproducer_status(mock_send_message):
68 """Test EnrichmentService's class method."""
69 EnrichmentService.get_eiproducer_status("test_producer")
70 mock_send_message.assert_called_once_with('GET',
71 'Get the status of EiProducer',
72 (f"{BASE_URL}/data-producer/v1/info-producers/test_producer/status"),
76 @mock.patch.object(EnrichmentService, 'send_message')
77 def test_get_eijobs(mock_send_message):
78 """Test EnrichmentService's class method."""
79 EnrichmentService.get_eijobs()
80 mock_send_message.assert_called_once_with('GET',
82 (f"{BASE_URL}/A1-EI/v1/eijobs"),
86 @mock.patch.object(EnrichmentService, 'send_message')
87 def test_get_eijob_individual(mock_send_message):
88 """Test EnrichmentService's class method."""
89 EnrichmentService.get_eijob_individual("test_job")
90 mock_send_message.assert_called_once_with('GET',
91 'Get individual EiJob',
92 (f"{BASE_URL}/A1-EI/v1/eijobs/test_job"),
96 @mock.patch.object(EnrichmentService, 'send_message')
97 def test_create_eitype(mock_send_message):
98 """Test EnrichmentService's class method."""
99 EnrichmentService.create_eitype("type_name", {})
100 mock_send_message.assert_called_once_with('PUT',
102 (f"{BASE_URL}/data-producer/v1/info-types/type_name"),
107 @mock.patch.object(EnrichmentService, 'send_message')
108 def test_create_eiproducer(mock_send_message):
109 """Test EnrichmentService's class method."""
110 EnrichmentService.create_eiproducer("producer", {})
111 mock_send_message.assert_called_once_with('PUT',
113 (f"{BASE_URL}/data-producer/v1/info-producers/producer"),
117 @mock.patch.object(EnrichmentService, 'send_message')
118 def test_create_eijob(mock_send_message):
119 """Test EnrichmentService's class method."""
120 EnrichmentService.create_eijob("job", {})
121 mock_send_message.assert_called_once_with('PUT',
123 (f"{BASE_URL}/A1-EI/v1/eijobs/job"),