Robert Schmidt
0da2f27fa7
Add some basic unit tests to verify that Iperf/Ping work. Harmonize the logging by only print in the "main" iperf/ping function, to avoid that if we test with multiple UEs, the output intermixes. Also, do one common mkdir before starting multiple pings/iperfs. Finally, harmonize the paths so that Iperf and Ping log their files in the same place.
56 lines
1.6 KiB
Python
56 lines
1.6 KiB
Python
import sys
|
|
import logging
|
|
logging.basicConfig(
|
|
level=logging.DEBUG,
|
|
stream=sys.stdout,
|
|
format="[%(asctime)s] %(levelname)8s: %(message)s"
|
|
)
|
|
import os
|
|
os.system(f'rm -rf cmake_targets')
|
|
os.system(f'mkdir -p cmake_targets/log')
|
|
import unittest
|
|
|
|
sys.path.append('./') # to find OAI imports below
|
|
import cls_oai_html
|
|
import cls_oaicitest
|
|
import cls_containerize
|
|
import ran
|
|
import epc
|
|
|
|
class TestPingIperf(unittest.TestCase):
|
|
def setUp(self):
|
|
self.html = cls_oai_html.HTMLManagement()
|
|
self.html.testCaseId = "000000"
|
|
self.ci = cls_oaicitest.OaiCiTest()
|
|
self.ci.ue_ids = ["test"]
|
|
self.ci.nodes = ["localhost"]
|
|
self.cont = cls_containerize.Containerize()
|
|
self.epc = epc.EPCManagement()
|
|
self.epc.IPAddress = "localhost"
|
|
self.epc.UserName = None
|
|
self.epc.Password = None
|
|
self.epc.SourceCodePath = os.getcwd()
|
|
self.ran = ran.RANManagement()
|
|
|
|
def test_ping(self):
|
|
self.ci.ping_args = "-c3 127.0.0.1"
|
|
self.ci.ping_packetloss_threshold = "0"
|
|
# TODO Should need nothing but options and UE(s) to use
|
|
self.ci.Ping(self.html, self.ran, self.epc, self.cont)
|
|
|
|
def test_iperf(self):
|
|
# note: needs to be five seconds because Iperf() adds -O 3, so if it is
|
|
# too short, the server is terminated before the client loaded
|
|
# everything
|
|
self.ci.iperf_args = "-u -t 5 -b 1M -R"
|
|
self.ci.svr_id = "test"
|
|
self.ci.svr_node = "localhost"
|
|
self.ci.iperf_packetloss_threshold = "0"
|
|
self.ci.iperf_bitrate_threshold = "0"
|
|
self.ci.iperf_profile = "balanced"
|
|
# TODO Should need nothing but options and UE(s) to use
|
|
self.ci.Iperf(self.html, self.ran, self.epc, self.cont)
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|