smo-install/test/pythonsdk/unit-tests/test_a1sim.py

   1 #!/usr/bin/env python3
   2 # SPDX-License-Identifier: Apache-2.0
   3 """Test A1sim module."""
   4
   5 from unittest import mock
   6 from oransdk.a1sim.a1sim import A1sim
   7
   8 HEADER = {"Accept": "application/json", "Content-Type": "application/json"}
   9 BASE_URL = "http://localhost:8081"
  10
  11 def test_initialization():
  12     """Class initialization test."""
  13     a1sim = A1sim()
  14     assert isinstance(a1sim, A1sim)
  15
  16
  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',
  22                                               'Get ric version',
  23                                               (f"{BASE_URL}/counter/interface"))
  24
  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',
  30                                               'Get ric status',
  31                                               (f"{BASE_URL}"))
  32
  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"))
  40
  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',
  46                                               'Create Policy Type',
  47                                               (f"{BASE_URL}/policytype?id=test_id"),
  48                                               data={},
  49                                               headers=HEADER)