Added check for security type

This commit is contained in:
Logan Lipke
2020-07-13 12:52:55 -07:00
parent 8453454877
commit 4f057c3f9b
6 changed files with 12 additions and 6 deletions

View File

@@ -43,7 +43,8 @@ class IPv4Test(LFCliBase):
def build(self):
# Build stations
self.station_profile.use_wpa2(True, self.ssid, self.password)
if self.security == "wpa2":
self.station_profile.use_wpa2(True, self.ssid, self.password)
self.station_profile.set_number_template(self.number_template)
print("Creating stations")
self.station_profile.set_command_flag("add_sta", "create_admin_down", 1)

View File

@@ -83,7 +83,8 @@ class IPV4L4(LFCliBase):
def build(self):
# Build stations
self.station_profile.use_wpa2(True, self.ssid, self.password)
if self.security == "wpa2":
self.station_profile.use_wpa2(True, self.ssid, self.password)
self.station_profile.set_number_template(self.number_template)
print("Creating stations")
self.station_profile.set_command_flag("add_sta", "create_admin_down", 1)

View File

@@ -79,7 +79,8 @@ class IPV4L4(LFCliBase):
def build(self):
# Build stations
self.station_profile.use_wpa2(True, self.ssid, self.password)
if self.security == "wpa2":
self.station_profile.use_wpa2(True, self.ssid, self.password)
self.station_profile.set_number_template(self.number_template)
print("Creating stations")
self.station_profile.set_command_flag("add_sta", "create_admin_down", 1)

View File

@@ -134,7 +134,8 @@ class IPV4VariableTime(LFCliBase):
debug=self.debug)
def build(self):
self.station_profile.use_wpa2(True, self.ssid, self.password)
if self.security == "wpa2":
self.station_profile.use_wpa2(True, self.ssid, self.password)
self.station_profile.set_number_template(self.number_template)
print("Creating stations")
self.station_profile.set_command_flag("add_sta", "create_admin_down", 1)

View File

@@ -44,7 +44,8 @@ class IPv6Test(LFCliBase):
self.station_profile.mode = 0
def build(self):
self.profile.use_wpa2(True, self.ssid, self.password)
if self.security == "wpa2":
self.station_profile.use_wpa2(True, self.ssid, self.password)
self.profile.set_number_template(self.prefix)
print("Creating stations")
self.profile.set_command_flag("add_sta", "create_admin_down", 1)

View File

@@ -136,7 +136,8 @@ class IPV6VariableTime(LFCliBase):
debug=self.debug)
def build(self):
self.station_profile.use_wpa2(True, self.ssid, self.password)
if self.security == "wpa2":
self.station_profile.use_wpa2(True, self.ssid, self.password)
self.station_profile.set_number_template(self.number_template)
print("Creating stations")
self.station_profile.set_command_flag("add_sta", "create_admin_down", 1)