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,6 +43,7 @@ class IPv4Test(LFCliBase):
def build(self):
# Build stations
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")

View File

@@ -83,6 +83,7 @@ class IPV4L4(LFCliBase):
def build(self):
# Build stations
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")

View File

@@ -79,6 +79,7 @@ class IPV4L4(LFCliBase):
def build(self):
# Build stations
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")

View File

@@ -134,6 +134,7 @@ class IPV4VariableTime(LFCliBase):
debug=self.debug)
def build(self):
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")

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,6 +136,7 @@ class IPV6VariableTime(LFCliBase):
debug=self.debug)
def build(self):
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")