From 908dff807e6ade982a8386e43af9b94d36db4054 Mon Sep 17 00:00:00 2001
From: James Prestwood <prestwoj@gmail.com>
Date: Tue, 28 Sep 2021 15:36:07 -0700
Subject: [PATCH] auto-t: fix testSAE to continue if a test fails

---
 autotests/testSAE/connection_test.py | 27 ++++++++++++---------------
 1 file changed, 12 insertions(+), 15 deletions(-)

diff --git a/autotests/testSAE/connection_test.py b/autotests/testSAE/connection_test.py
index 88aa077e..75f5c12d 100644
--- a/autotests/testSAE/connection_test.py
+++ b/autotests/testSAE/connection_test.py
@@ -62,9 +62,7 @@ class Test(unittest.TestCase):
         self.hostapd.set_value('vendor_elements', '')
         self.hostapd.reload()
         self.hostapd.wait_for_event("AP-ENABLED")
-        wd = IWD(True)
-        self.validate_connection(wd)
-        wd.clear_storage()
+        self.validate_connection(self.wd)
 
     def test_SAE_force_group_19(self):
         self.hostapd.set_value('sae_pwe', '0')
@@ -74,9 +72,7 @@ class Test(unittest.TestCase):
         self.hostapd.set_value('vendor_elements', 'dd0cf4f5e8050500000000000000')
         self.hostapd.reload()
         self.hostapd.wait_for_event("AP-ENABLED")
-        wd = IWD(True)
-        self.validate_connection(wd)
-        wd.clear_storage()
+        self.validate_connection(self.wd)
 
     def test_SAE_Group20(self):
         self.hostapd.set_value('sae_pwe', '0')
@@ -84,9 +80,7 @@ class Test(unittest.TestCase):
         self.hostapd.set_value('vendor_elements', '')
         self.hostapd.reload()
         self.hostapd.wait_for_event("AP-ENABLED")
-        wd = IWD(True)
-        self.validate_connection(wd)
-        wd.clear_storage()
+        self.validate_connection(self.wd)
 
     def test_SAE_H2E(self):
         self.hostapd.set_value('sae_pwe', '1')
@@ -94,9 +88,7 @@ class Test(unittest.TestCase):
         self.hostapd.set_value('vendor_elements', '')
         self.hostapd.reload()
         self.hostapd.wait_for_event("AP-ENABLED")
-        wd = IWD(True)
-        self.validate_connection(wd)
-        wd.clear_storage()
+        self.validate_connection(self.wd)
 
     def test_SAE_H2E_Group20(self):
         self.hostapd.set_value('sae_pwe', '1')
@@ -104,9 +96,14 @@ class Test(unittest.TestCase):
         self.hostapd.set_value('vendor_elements', '')
         self.hostapd.reload()
         self.hostapd.wait_for_event("AP-ENABLED")
-        wd = IWD(True)
-        self.validate_connection(wd)
-        wd.clear_storage()
+        self.validate_connection(self.wd)
+
+    def setUp(self):
+        self.wd = IWD(True)
+
+    def tearDown(self):
+        self.wd.clear_storage()
+        self.wd = None
 
     @classmethod
     def setUpClass(cls):