2 # SPDX-License-Identifier: Apache-2.0
3 """Test A1sim module."""
5 from unittest import mock
6 from oransdk.a1sim.a1sim import A1sim
8 HEADER = {"Accept": "application/json", "Content-Type": "application/json"}
9 BASE_URL = "http://localhost:8081"
11 def test_initialization():
12 """Class initialization test."""
14 assert isinstance(a1sim, A1sim)
17 @mock.patch.object(A1sim, 'send_message')
18 def test_check_version(mock_send_message):
19 """Test A1sim's class method."""
20 A1sim.check_version(BASE_URL)
21 mock_send_message.assert_called_once_with('GET',
23 (f"{BASE_URL}/counter/interface"))
25 @mock.patch.object(A1sim, 'send_message')
26 def test_check_status(mock_send_message):
27 """Test A1sim's class method."""
28 A1sim.check_status(BASE_URL)
29 mock_send_message.assert_called_once_with('GET',
33 @mock.patch.object(A1sim, 'send_message')
34 def test_get_policy_number(mock_send_message):
35 """Test A1sim's class method."""
36 A1sim.get_policy_number(BASE_URL)
37 mock_send_message.assert_called_once_with('GET',
38 'Get policy numbers for ric',
39 (f"{BASE_URL}/counter/num_instances"))
41 @mock.patch.object(A1sim, 'send_message')
42 def test_create_policy_type(mock_send_message):
43 """Test A1sim's class method."""
44 A1sim.create_policy_type(BASE_URL, "test_id", {})
45 mock_send_message.assert_called_once_with('PUT',
47 (f"{BASE_URL}/policytype?id=test_id"),