mirror of
https://github.com/Telecominfraproject/wlan-lanforge-scripts.git
synced 2025-11-01 19:28:00 +00:00
test_l3_longevity.py - adding multi cast
This commit is contained in:
@@ -17,19 +17,38 @@ import realm
|
||||
import time
|
||||
import datetime
|
||||
|
||||
#The valid ways to have multiple TX would be if they were on different ports.
|
||||
# I don't think that we need to add that complexity for now.
|
||||
|
||||
#In fact, it occurs to me that a multicast TX profile might actually already
|
||||
# know the right amount of information from which to generate it's RX endpoints
|
||||
|
||||
#So if we create a TX endpoint, it has a multicast ip and port.
|
||||
# Those two pieces of information are also useful for creating RXes.
|
||||
# I suggest one profile class.
|
||||
|
||||
|
||||
#multicast_profile.createTx(port) and multicast_profile.createRX(port_list)
|
||||
|
||||
#the ports have no knowledge of the protocols they will interact with,
|
||||
#the ports are layer-1 devices with some layer2 and layer3 features (mac addresses, IP addresses)
|
||||
#so anything you can legally assign an IP to can take a Layer3 connection or better
|
||||
|
||||
|
||||
class L3VariableTimeLongevity(LFCliBase):
|
||||
def __init__(self, host, port, endp_type, side_b, radios, radio_name_list, number_of_stations_per_radio_list,
|
||||
def __init__(self, host, port, endp_type, is_multicast, side_b, radios, radio_name_list, number_of_stations_per_radio_list,
|
||||
ssid_list, ssid_password_list, security, station_lists, name_prefix, resource=1,
|
||||
side_a_min_rate=56, side_a_max_rate=0,
|
||||
side_b_min_rate=56, side_b_max_rate=0,
|
||||
side_a_min_rate=56000, side_a_max_rate=0,
|
||||
side_b_min_rate=56000, side_b_max_rate=0,
|
||||
number_template="00", test_duration="256s",
|
||||
_debug_on=False,
|
||||
_debug_on=True,
|
||||
_exit_on_error=False,
|
||||
_exit_on_fail=False):
|
||||
super().__init__(host, port, _debug=_debug_on, _halt_on_error=_exit_on_error, _exit_on_fail=_exit_on_fail)
|
||||
self.host = host
|
||||
self.port = port
|
||||
self.endp_type = endp_type
|
||||
self.is_multicast = is_multicast
|
||||
self.side_b = side_b
|
||||
self.ssid_list = ssid_list
|
||||
self.ssid_password_list = ssid_password_list
|
||||
@@ -45,8 +64,9 @@ class L3VariableTimeLongevity(LFCliBase):
|
||||
self.number_of_stations_per_radio_list = number_of_stations_per_radio_list
|
||||
self.local_realm = realm.Realm(lfclient_host=self.host, lfclient_port=self.port)
|
||||
self.cx_profile = self.local_realm.new_l3_cx_profile()
|
||||
self.multicast_profile = self.local_realm.new_multicast_profile()
|
||||
self.station_profiles = []
|
||||
|
||||
|
||||
index = 0
|
||||
for radio in radios:
|
||||
self.station_profile = self.local_realm.new_station_profile()
|
||||
@@ -59,13 +79,25 @@ class L3VariableTimeLongevity(LFCliBase):
|
||||
self.station_profiles.append(self.station_profile)
|
||||
index += 1
|
||||
|
||||
self.cx_profile.host = self.host
|
||||
self.cx_profile.port = self.port
|
||||
self.cx_profile.name_prefix = self.name_prefix
|
||||
self.cx_profile.side_a_min_bps = side_a_min_rate
|
||||
self.cx_profile.side_a_max_bps = side_a_max_rate
|
||||
self.cx_profile.side_b_min_bps = side_b_min_rate
|
||||
self.cx_profile.side_b_max_bps = side_b_max_rate
|
||||
if is_multicast:
|
||||
self.multicast_profile.host = self.host
|
||||
self.cx_profile.host = self.host
|
||||
self.cx_profile.port = self.port
|
||||
self.cx_profile.name_prefix = self.name_prefix
|
||||
self.cx_profile.side_a_min_bps = 0
|
||||
self.cx_profile.side_a_max_bps = 0
|
||||
self.cx_profile.side_b_min_bps = side_b_min_rate
|
||||
self.cx_profile.side_b_max_bps = side_b_max_rate
|
||||
|
||||
|
||||
else:
|
||||
self.cx_profile.host = self.host
|
||||
self.cx_profile.port = self.port
|
||||
self.cx_profile.name_prefix = self.name_prefix
|
||||
self.cx_profile.side_a_min_bps = side_a_min_rate
|
||||
self.cx_profile.side_a_max_bps = side_a_max_rate
|
||||
self.cx_profile.side_b_min_bps = side_b_min_rate
|
||||
self.cx_profile.side_b_max_bps = side_b_max_rate
|
||||
|
||||
def __get_rx_values(self):
|
||||
cx_list = self.json_get("endp?fields=name,rx+bytes", debug_=True)
|
||||
@@ -86,6 +118,8 @@ class L3VariableTimeLongevity(LFCliBase):
|
||||
expected_passes += 1
|
||||
if new_list[item] > old_list[item]:
|
||||
passes += 1
|
||||
print(item, new_list[item], old_list[item], passes, expected_passes)
|
||||
|
||||
if passes == expected_passes:
|
||||
return True
|
||||
else:
|
||||
@@ -106,10 +140,36 @@ class L3VariableTimeLongevity(LFCliBase):
|
||||
temp_stations_list.append(self.side_b)
|
||||
self.local_realm.wait_for_ip(self.resource, temp_station_list)
|
||||
|
||||
|
||||
temp_station_list = []
|
||||
if self.is_multicast:
|
||||
for station_list in self.station_lists:
|
||||
for station in range(len(station_list)):
|
||||
temp_station_list.append(str(self.resource) + "." + station_list[station])
|
||||
|
||||
# start recievers
|
||||
self.multicast_profile.start_mc_rx(side_a=temp_station_list)
|
||||
|
||||
# right now the multicast is hard coded, need to pass in station
|
||||
self.multicast_profile.start_mc_tx(side_b=self.side_b)
|
||||
|
||||
else:
|
||||
self.cx_profile.start_cx()
|
||||
|
||||
cur_time = datetime.datetime.now()
|
||||
old_cx_rx_values = self.__get_rx_values()
|
||||
old_rx_values = self.__get_rx_values()
|
||||
filtered_old_rx_values = []
|
||||
if self.is_multicast:
|
||||
for rx_value in old_rx_values:
|
||||
for station in self.station_lists:
|
||||
if rx_value in station:
|
||||
filtered_old_rx_values += rx_value
|
||||
else:
|
||||
filtered_old_rx_values = old_rx_values
|
||||
|
||||
|
||||
end_time = self.local_realm.parse_time(self.test_duration) + cur_time
|
||||
self.cx_profile.start_cx()
|
||||
|
||||
passes = 0
|
||||
expected_passes = 0
|
||||
while cur_time < end_time:
|
||||
@@ -117,15 +177,24 @@ class L3VariableTimeLongevity(LFCliBase):
|
||||
while cur_time < interval_time:
|
||||
cur_time = datetime.datetime.now()
|
||||
time.sleep(1)
|
||||
|
||||
new_rx_values = self.__get_rx_values()
|
||||
filtered_new_rx_values = []
|
||||
if self.is_multicast:
|
||||
for rx_value in new_rx_values:
|
||||
for station in self.station_lists:
|
||||
if rx_value in station:
|
||||
filtered_new_rx_values += rx_value
|
||||
else:
|
||||
filtered_new_rx_values = new_rx_values
|
||||
|
||||
new_cx_rx_values = self.__get_rx_values()
|
||||
expected_passes += 1
|
||||
if self.__compare_vals(old_cx_rx_values, new_cx_rx_values):
|
||||
if self.__compare_vals(filtered_old_rx_values, filtered_new_rx_values):
|
||||
passes += 1
|
||||
else:
|
||||
self._fail("FAIL: Not all stations increased traffic", print_fail)
|
||||
break
|
||||
old_cx_rx_values = new_cx_rx_values
|
||||
old_rx_values = new_rx_values
|
||||
cur_time = datetime.datetime.now()
|
||||
|
||||
if passes == expected_passes:
|
||||
@@ -136,17 +205,52 @@ class L3VariableTimeLongevity(LFCliBase):
|
||||
for station_list in self.station_lists:
|
||||
for station_name in station_list:
|
||||
data = LFUtils.portDownRequest(1, station_name)
|
||||
url = "json-cli/set_port"
|
||||
url = "cli-json/set_port"
|
||||
self.json_post(url, data)
|
||||
|
||||
def cleanup(self, resource):
|
||||
resource = 1
|
||||
remove_all_endpoints = True
|
||||
self.local_realm.remove_all_cxs(remove_all_endpoints)
|
||||
self.local_realm.remove_all_stations(resource)
|
||||
data = {
|
||||
"name":"BLANK",
|
||||
"action":"overwrite"
|
||||
}
|
||||
url = "cli-json/load"
|
||||
self.json_post(url, data)
|
||||
#remove_all_endpoints = True
|
||||
#self.local_realm.remove_all_cxs(remove_all_endpoints)
|
||||
#self.local_realm.remove_all_stations(resource)
|
||||
|
||||
|
||||
def build(self):
|
||||
# refactor in LFUtils.port_zero_request()
|
||||
resource = 1
|
||||
data ={
|
||||
'shelf':1,
|
||||
'resource':1,
|
||||
'port':'eth1',
|
||||
'ip_addr':'0.0.0.0',
|
||||
'netmask':'0.0.0.0',
|
||||
'gateway':'0.0.0.0',
|
||||
'current_flags':0,
|
||||
'interest':402653212
|
||||
}
|
||||
|
||||
url = "cli-json/set_port"
|
||||
self.json_post(url, data)
|
||||
|
||||
# refactor into LFUtils
|
||||
data ={
|
||||
"shelf":1,
|
||||
"resource": resource,
|
||||
"port":"br0",
|
||||
"network_devs":"eth1",
|
||||
"br_flags":1
|
||||
}
|
||||
url = "cli-json/add_br"
|
||||
self.json_post(url, data)
|
||||
|
||||
#refactor later
|
||||
|
||||
try:
|
||||
data = LFUtils.port_dhcp_up_request(resource, self.side_b)
|
||||
self.json_post("/cli-json/set_port", data)
|
||||
@@ -154,32 +258,38 @@ class L3VariableTimeLongevity(LFCliBase):
|
||||
print("LFUtils.port_dhcp_up_request didn't complete ")
|
||||
print("or the json_post failed either way {} did not set up dhcp so test may no pass ".format(self.side_b))
|
||||
|
||||
#exit(1)
|
||||
|
||||
resource = 1
|
||||
index = 0
|
||||
for station_profile in self.station_profiles:
|
||||
station_profile.use_security(station_profile.security, station_profile.ssid, station_profile.ssid_pass)
|
||||
station_profile.set_number_template(station_profile.number_template)
|
||||
print("Creating stations")
|
||||
station_profile.set_command_flag("add_sta", "create_admin_down", 1)
|
||||
station_profile.set_command_param("set_port", "report_timer", 1500)
|
||||
station_profile.set_command_flag("set_port", "rpt_timer", 1)
|
||||
#station_profile.set_command_flag("add_sta", "create_admin_down", 1)
|
||||
#station_profile.set_command_param("set_port", "report_timer", 1500)
|
||||
#station_profile.set_command_flag("set_port", "rpt_timer", 1)
|
||||
temp_station_list = []
|
||||
|
||||
index = 0
|
||||
for station_list in self.station_lists:
|
||||
for station in range(len(station_list)):
|
||||
temp_station_list.append(str(self.resource) + "." + station_list[station])
|
||||
station_profile.create(resource=1, radio=self.radio_list[index], sta_names_=station_list, debug=False )
|
||||
station_profile.create(resource=1, radio=self.radio_list[index], sta_names_=station_list, debug=True )
|
||||
index += 1
|
||||
self.cx_profile.create(endp_type=self.endp_type, side_a=temp_station_list, side_b='1.'+self.side_b, sleep_time=.5)
|
||||
if self.is_multicast:
|
||||
self.multicast_profile.create_mc_tx(self.side_b)
|
||||
self.multicast_profile.create_mc_rx(side_a=temp_station_list)
|
||||
else:
|
||||
self.cx_profile.create(endp_type=self.endp_type, side_a=temp_station_list, side_b='1.'+self.side_b, sleep_time=.5)
|
||||
self._pass("PASS: Stations build finished")
|
||||
|
||||
def valid_endp_type(endp_type):
|
||||
valid_endp_type=['lf_udp','lf_udp6','lf_tcp','lf_tcp6']
|
||||
valid_endp_type=['lf_udp','lf_udp6','lf_tcp','lf_tcp6','mc_udp','mc_udp6']
|
||||
if str(endp_type) in valid_endp_type:
|
||||
return endp_type
|
||||
else:
|
||||
print('invalid endp_type. Valid types lf_udp, lf_udp6, lf_tcp, lf_tcp6')
|
||||
print('invalid endp_type. Valid types lf_udp, lf_udp6, lf_tcp, lf_tcp6, mc_udp, mc_udp6')
|
||||
exit(1)
|
||||
|
||||
def main():
|
||||
@@ -200,33 +310,38 @@ def main():
|
||||
''',
|
||||
|
||||
description='''\
|
||||
test_l3_longevity.py:
|
||||
--------------------
|
||||
Basic Idea: create stations, create traffic between upstream port and stations, run traffic.
|
||||
The traffic on the stations will be checked once per minute to verify that traffic is transmitted
|
||||
and recieved.
|
||||
test_l3_longevity.py:
|
||||
--------------------
|
||||
Basic Idea: create stations, create traffic between upstream port and stations, run traffic.
|
||||
The traffic on the stations will be checked once per minute to verify that traffic is transmitted
|
||||
and recieved.
|
||||
|
||||
Test will exit on failure of not recieving traffice for one minute on any station.
|
||||
Test will exit on failure of not recieving traffice for one minute on any station.
|
||||
|
||||
Scripts are executed from: ./lanforge/py-scripts
|
||||
Scripts are executed from: ./lanforge/py-scripts
|
||||
|
||||
Generic command layout:
|
||||
python .\\test_l3_longevity.py --test_duration <duration> --endp_type <traffic type> --upstream_port <port> --radio <radio 0> <stations> <ssid> <ssid password>
|
||||
Stations start counting form zero, thus stations count from zero - number of las
|
||||
|
||||
Note: multiple --radio switches may be entered up to the number of radios available:
|
||||
--radio <radio 0> <stations> <ssid> <ssid password> --radio <radio 01> <stations> <ssid> <ssid password>
|
||||
Generic command layout:
|
||||
python .\\test_l3_longevity.py --test_duration <duration> --endp_type <traffic type> --upstream_port <port> --radio <radio 0> <stations> <ssid> <ssid password>
|
||||
|
||||
<duration>: number followed by one of the following
|
||||
d - days
|
||||
h - hours
|
||||
m - minutes
|
||||
s - seconds
|
||||
Note: multiple --radio switches may be entered up to the number of radios available:
|
||||
--radio <radio 0> <stations> <ssid> <ssid password> --radio <radio 01> <number of last station> <ssid> <ssid password>
|
||||
|
||||
<traffic type>:
|
||||
lf_udp : IPv4 UDP traffic
|
||||
lf_tcp : IPv4 TCP traffic
|
||||
lf_udp6 : IPv6 UDP traffic
|
||||
lf_tcp6 : IPv6 TCP traffic
|
||||
<duration>: number followed by one of the following
|
||||
d - days
|
||||
h - hours
|
||||
m - minutes
|
||||
s - seconds
|
||||
|
||||
<traffic type>:
|
||||
lf_udp : IPv4 UDP traffic
|
||||
lf_tcp : IPv4 TCP traffic
|
||||
lf_udp6 : IPv6 UDP traffic
|
||||
lf_tcp6 : IPv6 TCP traffic
|
||||
|
||||
mc_udp : IPv4 multi cast UDP traffic
|
||||
mc_udp6 : IPv6 multi cast UDP traffic
|
||||
|
||||
Example:
|
||||
1. Test duration 4 minutes
|
||||
@@ -247,8 +362,8 @@ def main():
|
||||
parser.add_argument('-u', '--upstream_port', help='--upstream_port <cross connect upstream_port> example: --upstream_port eth1',default='eth1')
|
||||
|
||||
requiredNamed = parser.add_argument_group('required arguments')
|
||||
requiredNamed.add_argument('-r','--radio', action='append', nargs=4, metavar=('<wiphyX>', '<number of stations>','<ssid>','<ssid password>'),
|
||||
help ='--radio <number_of_wiphy> <number_of_stations> <ssid> <ssid password> ',required=True)
|
||||
requiredNamed.add_argument('-r','--radio', action='append', nargs=4, metavar=('<wiphyX>', '<number last station>','<ssid>','<ssid password>'),
|
||||
help ='--radio <number_of_wiphy> <number of last station> <ssid> <ssid password> ',required=True)
|
||||
args = parser.parse_args()
|
||||
|
||||
if args.test_duration:
|
||||
@@ -263,6 +378,8 @@ def main():
|
||||
if args.radio:
|
||||
radios = args.radio
|
||||
|
||||
is_multicast = False
|
||||
|
||||
radio_offset = 0
|
||||
number_of_stations_offset = 1
|
||||
ssid_offset = 2
|
||||
@@ -275,6 +392,9 @@ def main():
|
||||
ssid_list = []
|
||||
ssid_password_list = []
|
||||
|
||||
if endp_type in ['mc_udp','mc_udp6']:
|
||||
is_multicast = True
|
||||
|
||||
index = 0
|
||||
for radio in radios:
|
||||
radio_name = radio[radio_offset]
|
||||
@@ -303,6 +423,7 @@ def main():
|
||||
station_lists= station_lists,
|
||||
name_prefix="var_time",
|
||||
endp_type=endp_type,
|
||||
is_multicast=is_multicast,
|
||||
side_b=side_b,
|
||||
radios=radios,
|
||||
radio_name_list=radio_name_list,
|
||||
|
||||
Reference in New Issue
Block a user