def __init__(self):
"""
Initialize the various classes required by the tests and setup.
"""
self._s1_util = S1ApUtil()
self._enBConfig()
subscriber_client = SubscriberDbGrpc()
mobility_client = MobilityServiceGrpc()
magmad_client = MagmadServiceGrpc()
self._sub_util = SubscriberUtil(subscriber_client)
# Remove existing subscribers to start
self._sub_util.cleanup()
self._mobility_util = MobilityUtil(mobility_client)
self._mobility_util.cleanup()
self._magmad_util = MagmadUtil(magmad_client)
# gateway tests don't require restart, just wait for healthy now
self._gateway_services = GatewayServicesUtil()
self.wait_gateway_healthy = True
if not self.wait_gateway_healthy:
self.init_s1ap_tester()
self._configuredUes = []
self._ue_idx = 0 # Index of UEs already used in test
self._trf_util = TrafficUtil()
class TestS1SetupFailureIncorrectPlmn(unittest.TestCase):
""" S1 Setup with incorrect plmn ID """
s1ap_utils._s1_util = S1ApUtil()
print("************************* Enb tester config")
req = s1ap_types.FwNbConfigReq_t()
req.cellId_pr.pres = True
req.cellId_pr.cell_id = 10
req.plmnId_pr.pres = True
req.plmnId_pr.plmn_id[0] = 3
req.plmnId_pr.plmn_id[1] = 3
req.plmnId_pr.plmn_id[2] = 3
req.plmnId_pr.plmn_id[3] = 3
req.plmnId_pr.plmn_id[4] = 3
req.plmnId_pr.plmn_id[5] = 3
assert (s1ap_utils._s1_util.issue_cmd(s1ap_types.tfwCmd.ENB_CONFIG,
req) == 0)
response = s1ap_utils._s1_util.get_response()
assert (response.msg_type == s1ap_types.tfwCmd.ENB_CONFIG_CONFIRM.value)
res = response.cast(s1ap_types.FwNbConfigCfm_t)
assert (res.status == s1ap_types.CfgStatus.CFG_DONE.value)
assert (s1ap_utils._s1_util.issue_cmd(s1ap_types.tfwCmd.ENB_S1_SETUP_REQ,
req) == 0)
response = s1ap_utils._s1_util.get_response()
assert (response.msg_type == s1ap_types.tfwCmd.ENB_S1_SETUP_RESP.value)
def __init__(
self,
stateless_mode=MagmadUtil.stateless_cmds.ENABLE,
apn_correction=MagmadUtil.apn_correction_cmds.DISABLE,
health_service=MagmadUtil.health_service_cmds.DISABLE,
federated_mode=False,
ip_version=4,
):
"""
Initialize the various classes required by the tests and setup.
"""
t = time.localtime()
current_time = time.strftime("%H:%M:%S", t)
if TestWrapper.TEST_CASE_EXECUTION_COUNT != 0:
print("\n**Running the test case again to identify flaky behavior")
TestWrapper.TEST_CASE_EXECUTION_COUNT += 1
print(
"\nTest Case Execution Count:",
TestWrapper.TEST_CASE_EXECUTION_COUNT,
"[Start time: " + str(current_time) + "]",
)
self._s1_util = S1ApUtil()
self._enBConfig(ip_version)
federated_mode = (os.environ.get("FEDERATED_MODE") == "True")
print(
f"\n *** Running the test in {'Non-' if not federated_mode else ''}"
"Federated Mode\n", )
if self._test_oai_upstream:
subscriber_client = SubscriberDbCassandra()
self.wait_gateway_healthy = False
elif federated_mode:
subscriber_client = HSSGrpc()
self.wait_gateway_healthy = True
else:
subscriber_client = SubscriberDbGrpc()
self.wait_gateway_healthy = True
mobility_client = MobilityServiceGrpc()
magmad_client = MagmadServiceGrpc()
self._sub_util = SubscriberUtil(subscriber_client)
# Remove existing subscribers to start
self._sub_util.cleanup()
self._mobility_util = MobilityUtil(mobility_client)
self._mobility_util.cleanup()
self._magmad_util = MagmadUtil(magmad_client)
self._magmad_util.config_stateless(stateless_mode)
self._magmad_util.config_apn_correction(apn_correction)
self._magmad_util.config_health_service(health_service)
# gateway tests don't require restart, just wait for healthy now
self._gateway_services = GatewayServicesUtil()
if not self.wait_gateway_healthy:
self.init_s1ap_tester()
self._configuredUes = []
self._ue_idx = 0 # Index of UEs already used in test
self._trf_util = TrafficUtil()
def setUp(self):
"""Initialize before test case execution"""
if s1ap_wrapper.TestWrapper.TEST_CASE_EXECUTION_COUNT != 0:
print("\n**Running the test case again to identify flaky behavior")
s1ap_wrapper.TestWrapper.TEST_CASE_EXECUTION_COUNT += 1
print(
"Test Case Execution Count:",
s1ap_wrapper.TestWrapper.TEST_CASE_EXECUTION_COUNT,
)
self._s1_util = S1ApUtil()
def __init__(
self,
stateless_mode=MagmadUtil.stateless_cmds.ENABLE,
apn_correction=MagmadUtil.apn_correction_cmds.DISABLE,
health_service=MagmadUtil.health_service_cmds.DISABLE,
):
"""
Initialize the various classes required by the tests and setup.
"""
if not TestWrapper.IS_TEST_RUNNING_FIRST_TIME:
print("\n**Running the test case again to identify flaky behavior")
TestWrapper.IS_TEST_RUNNING_FIRST_TIME = False
t = time.localtime()
current_time = time.strftime("%H:%M:%S", t)
print("Start time", current_time)
self._s1_util = S1ApUtil()
self._enBConfig()
if self._test_oai_upstream:
subscriber_client = SubscriberDbCassandra()
self.wait_gateway_healthy = False
else:
subscriber_client = SubscriberDbGrpc()
self.wait_gateway_healthy = True
mobility_client = MobilityServiceGrpc()
magmad_client = MagmadServiceGrpc()
self._sub_util = SubscriberUtil(subscriber_client)
# Remove existing subscribers to start
self._sub_util.cleanup()
self._mobility_util = MobilityUtil(mobility_client)
self._mobility_util.cleanup()
self._magmad_util = MagmadUtil(magmad_client)
self._magmad_util.config_stateless(stateless_mode)
self._magmad_util.config_apn_correction(apn_correction)
self._magmad_util.config_health_service(health_service)
# gateway tests don't require restart, just wait for healthy now
self._gateway_services = GatewayServicesUtil()
if not self.wait_gateway_healthy:
self.init_s1ap_tester()
self._configuredUes = []
self._ue_idx = 0 # Index of UEs already used in test
self._trf_util = TrafficUtil()
def __init__(self):
"""
Initialize the various classes required by the tests and setup.
"""
self._s1_util = S1ApUtil()
self._enBConfig()
# if self._test_cloud:
# self._cloud_manager = CloudManager()
# self._cloud_manager.clean_up()
# subscriber_client = SubscriberDbRest(self._cloud_manager)
# mobility_client = MobilityServiceRest(self._cloud_manager,
# self._gateway_services)
# self._sub_util = SubscriberUtil(subscriber_client)
# self._mobility_util = MobilityUtil(mobility_client)
# self._gateway_services = GatewayServicesUtil()
# self.wait_gateway_healthy = False
if self._test_oai_upstream:
subscriber_client = SubscriberDbMySQL()
mobility_client = MobilityServiceOAI()
self._sub_util = SubscriberUtil(subscriber_client)
self._mobility_util = MobilityUtil(mobility_client)
self.wait_gateway_healthy = False
else:
subscriber_client = SubscriberDbGrpc()
mobility_client = MobilityServiceGrpc()
self._sub_util = SubscriberUtil(subscriber_client)
# Remove existing subscribers to start
self._sub_util.clean_up()
self._mobility_util = MobilityUtil(mobility_client)
# gateway tests don't require restart, just wait for healthy now
self._gateway_services = GatewayServicesUtil()
self.wait_gateway_healthy = True
if not self.wait_gateway_healthy:
self.init_s1ap_tester()
self._configuredUes = []
self._ue_idx = 0 # Index of UEs already used in test
self._trf_util = TrafficUtil()
class TestS1SetupFailureIncorrectTac(unittest.TestCase):
"""S1 Setup Request with incorrect TAC value """
s1ap_utils._s1_util = S1ApUtil()
print("************************* Enb tester config")
req = s1ap_types.FwNbConfigReq_t()
req.cellId_pr.pres = True
req.cellId_pr.cell_id = 10
req.tac_pr.pres = True
req.tac_pr.tac = 0
assert (s1ap_utils._s1_util.issue_cmd(s1ap_types.tfwCmd.ENB_CONFIG,
req) == 0)
response = s1ap_utils._s1_util.get_response()
assert (response.msg_type == s1ap_types.tfwCmd.ENB_CONFIG_CONFIRM.value)
res = response.cast(s1ap_types.FwNbConfigCfm_t)
assert (res.status == s1ap_types.CfgStatus.CFG_DONE.value)
req = None
assert (s1ap_utils._s1_util.issue_cmd(s1ap_types.tfwCmd.ENB_S1_SETUP_REQ,
req) == 0)
response = s1ap_utils._s1_util.get_response()
assert (response.msg_type == s1ap_types.tfwCmd.ENB_S1_SETUP_RESP.value)
def __init__(
self,
stateless_mode=MagmadUtil.stateless_cmds.DISABLE,
apn_correction=MagmadUtil.apn_correction_cmds.DISABLE,
):
"""
Initialize the various classes required by the tests and setup.
"""
self._s1_util = S1ApUtil()
self._enBConfig()
if self._test_oai_upstream:
subscriber_client = SubscriberDbCassandra()
self.wait_gateway_healthy = False
else:
subscriber_client = SubscriberDbGrpc()
self.wait_gateway_healthy = True
mobility_client = MobilityServiceGrpc()
magmad_client = MagmadServiceGrpc()
self._sub_util = SubscriberUtil(subscriber_client)
# Remove existing subscribers to start
self._sub_util.cleanup()
self._mobility_util = MobilityUtil(mobility_client)
self._mobility_util.cleanup()
self._magmad_util = MagmadUtil(magmad_client)
self._magmad_util.config_stateless(stateless_mode)
self._magmad_util.config_apn_correction(apn_correction)
# gateway tests don't require restart, just wait for healthy now
self._gateway_services = GatewayServicesUtil()
if not self.wait_gateway_healthy:
self.init_s1ap_tester()
self._configuredUes = []
self._ue_idx = 0 # Index of UEs already used in test
self._trf_util = TrafficUtil()
def setUp(self):
self._s1_util = S1ApUtil()
def setUp(self):
"""Initialize before test case execution"""
if not s1ap_wrapper.TestWrapper.IS_TEST_RUNNING_FIRST_TIME:
print("\n**Running the test case again to identify flaky behavior")
s1ap_wrapper.TestWrapper.IS_TEST_RUNNING_FIRST_TIME = False
self._s1_util = S1ApUtil()
