From 4d31daa296d98447a93d65d8faa6e095be5ddf3c Mon Sep 17 00:00:00 2001
From: Rahul Rahul <rahul.rahul@intel.com>
Date: Fri, 13 May 2016 14:18:20 -0700
Subject: [PATCH] autotests: connectDisconnect test

Fixed path to utility.py
---
 autotests/test1AP/connectDisconnectTest.py | 96 ++++++++++++++++++++++
 1 file changed, 96 insertions(+)
 create mode 100755 autotests/test1AP/connectDisconnectTest.py

diff --git a/autotests/test1AP/connectDisconnectTest.py b/autotests/test1AP/connectDisconnectTest.py
new file mode 100755
index 00000000..a793cc66
--- /dev/null
+++ b/autotests/test1AP/connectDisconnectTest.py
@@ -0,0 +1,96 @@
+#!/usr/bin/python3
+
+#built-in python libraries
+import unittest
+from gi.repository import GLib
+import dbus
+import time
+import logging
+import os
+from random import randrange
+from subprocess import Popen
+import sys
+sys.path.append('../utility') #needed to import all the utlilty modules
+import utility
+
+def defineAgentVars():
+    global manager, pathAgent
+    bus = dbus.SystemBus()
+    pathAgent = "/connectToNetwork/agent/" + str(randrange(100))
+    manager = dbus.Interface(bus.get_object('net.connman.iwd', "/"),
+                          'net.connman.iwd.Manager')
+
+def getManager():
+    return manager
+
+def getPathAgent():
+    return pathAgent
+
+class TestConnectDisconnect(unittest.TestCase):
+    def doConnectDisconnect(self):
+        deviceList = utility.getDeviceList(bus)
+        networkList = utility.getNetworkList(deviceList, bus)
+        utility.printNetworkInfo(networkList)
+        networkToConnect = utility.getNetworkToConnectTo(networkList)
+
+        # check if networkToConnect is not null. If yes, restart program
+        # so that the network list is updated. Alternatively, we can scan
+        # for networks.
+        if (networkToConnect == ""):
+            logger.debug("RESTART PROGRAM")
+            os.execl(sys.executable, sys.executable, * sys.argv)
+
+        self.assertNotEqual(networkToConnect, "")
+        # start simpleAgent
+        proc = Popen([sys.executable, './simpleAgent.py'])
+        time.sleep(2)
+        network = dbus.Interface(bus.get_object("net.connman.iwd",
+                                            networkToConnect),
+                                            "net.connman.iwd.Network")
+        utility.connect(networkToConnect, self, mainloop, bus)
+
+        logger.info("Currently connected to: %s",
+                    utility.getCurrentlyConnectedNetworkName())
+        self.assertEqual(utility.getCurrentlyConnectedNetworkName(), "IntelWIFI")
+
+        # retrieve the deviceId form networkToConnect. This will be used
+        # for checking if we are disconnecting from the right device later.
+        deviceIdIndex = networkToConnect.rfind("/", 0,
+                                               len(networkToConnect))
+        deviceIdOfConnectedNetwork = networkToConnect[0:deviceIdIndex]
+        logger.debug("device id of connected network %s",
+                     deviceIdOfConnectedNetwork)
+
+        # wait 2 seconds before initiating disconnect
+        time.sleep(2)
+        deviceIdToDisconnect = utility.getCurrentlyConnectedDevice()
+        logger.info("Disconnecting from: %s", deviceIdToDisconnect)
+        self.assertEqual(deviceIdToDisconnect, deviceIdOfConnectedNetwork)
+        utility.disconnect(deviceIdToDisconnect, mainloop, bus)
+        #terminate proc
+        proc.terminate()
+
+    # connect to network A. Wait for 2 seconds. Disconnect.
+    def test_connectDisconnect(self):
+        logger.info(sys._getframe().f_code.co_name)
+        while (True):
+            if bus.name_has_owner('net.connman.iwd') == True:
+                break
+        self.doConnectDisconnect()
+        #watch doesn't seem to work. So used name_has_owner
+        #watch = bus.watch_name_owner("net.connman.iwd",self.doConnectDisconnect)
+
+    @classmethod
+    def setUpClass(cls):
+        global logger, bus, mainloop
+        utility.initLogger()
+        logger = logging.getLogger(__name__)
+        bus = dbus.SystemBus()
+        mainloop = GLib.MainLoop()
+
+    @classmethod
+    def tearDownClass(cls):
+        mainloop.quit()
+
+if __name__ == '__main__':
+    unittest.main(exit=True)