From c5eba1cc27750edcf30cd172bcfbcade613a5a18 Mon Sep 17 00:00:00 2001
From: James Prestwood <james.prestwood@linux.intel.com>
Date: Mon, 15 Jul 2019 09:12:58 -0700
Subject: [PATCH] auto-t: add test for Hotspot with roaming consortium

---
 autotests/testHotspot/roaming.conf    | 10 ++++
 autotests/testHotspot/roaming_test.py | 69 +++++++++++++++++++++++++++
 2 files changed, 79 insertions(+)
 create mode 100644 autotests/testHotspot/roaming.conf
 create mode 100644 autotests/testHotspot/roaming_test.py

diff --git a/autotests/testHotspot/roaming.conf b/autotests/testHotspot/roaming.conf
new file mode 100644
index 00000000..33391e28
--- /dev/null
+++ b/autotests/testHotspot/roaming.conf
@@ -0,0 +1,10 @@
+[Security]
+EAP-Method=TTLS
+EAP-Identity=anonymous@example.com
+EAP-TTLS-Phase2-Method=Tunneled-MSCHAPv2
+
+[Hotspot]
+RoamingConsortium=1020304050
+
+[Settings]
+Autoconnect=False
diff --git a/autotests/testHotspot/roaming_test.py b/autotests/testHotspot/roaming_test.py
new file mode 100644
index 00000000..7ce88551
--- /dev/null
+++ b/autotests/testHotspot/roaming_test.py
@@ -0,0 +1,69 @@
+#!/usr/bin/python3
+
+import unittest
+import sys
+import os
+
+sys.path.append('../util')
+import iwd
+from iwd import IWD
+from iwd import PSKAgent
+from iwd import NetworkType
+from hostapd import HostapdCLI
+import testutil
+
+class Test(unittest.TestCase):
+
+    def test_connection_success(self):
+        wd = IWD(True, '/tmp')
+
+        hapd = HostapdCLI(config='ssidHotspot.conf')
+
+        psk_agent = PSKAgent('abc', ('domain\\user', 'testpasswd'))
+        wd.register_psk_agent(psk_agent)
+
+        devices = wd.list_devices(1)
+        device = devices[0]
+
+        condition = 'not obj.scanning'
+        wd.wait_for_object_condition(device, condition)
+
+        device.scan()
+
+        condition = 'not obj.scanning'
+        wd.wait_for_object_condition(device, condition)
+
+        ordered_network = device.get_ordered_network('Hotspot')
+
+        self.assertEqual(ordered_network.type, NetworkType.eap)
+
+        condition = 'not obj.connected'
+        wd.wait_for_object_condition(ordered_network.network_object, condition)
+
+        ordered_network.network_object.connect()
+
+        condition = 'obj.connected'
+        wd.wait_for_object_condition(ordered_network.network_object, condition)
+
+        testutil.test_iface_operstate()
+        testutil.test_ifaces_connected(device.name, hapd.ifname)
+
+        device.disconnect()
+
+        condition = 'not obj.connected'
+        wd.wait_for_object_condition(ordered_network.network_object, condition)
+
+        wd.unregister_psk_agent(psk_agent)
+
+    @classmethod
+    def setUpClass(cls):
+        IWD.copy_to_hotspot('roaming.conf')
+        conf = '[General]\ndisable_anqp=1\n'
+        os.system('echo "%s" > /tmp/main.conf' % conf)
+
+    @classmethod
+    def tearDownClass(cls):
+        IWD.clear_storage()
+
+if __name__ == '__main__':
+    unittest.main(exit=True)