autotests: simpleAgent for twoNetworksTest

autotests/test2AP/simpleAgent.py

@@ -0,0 +1,82 @@
+#!/usr/bin/python3
+
+#built-in python libraries
+from gi.repository import GLib
+import dbus
+import dbus.service
+import dbus.mainloop.glib
+import logging
+import traceback
+import threading
+import time
+
+import sys
+sys.path.append('../utility') #needed to import all the utlilty modules
+import utility
+import twoNetworksTest
+
+class Agent(dbus.service.Object):
+    @dbus.service.method("net.connman.iwd.Agent",
+                            in_signature='', out_signature='')
+    def Release(self):
+        logger.debug("Release")
+        mainloop.quit()
+
+    @dbus.service.method("net.connman.iwd.Agent",
+                            in_signature='o',
+                            out_signature='s')
+    def RequestPassphrase(self, path):
+        utility.initLogger()
+        logger = logging.getLogger(__name__)
+        logger.info("RequestPassphrase")
+        return 'EasilyGuessedPassword'
+
+def startAgent(mainloop):
+    mainloop.run()
+
+def registerAgent(bus, mainloop):
+    twoNetworksTest.defineAgentVars()
+    manager = twoNetworksTest.getManager()
+    pathAgent = twoNetworksTest.getPathAgent()
+    object = Agent(bus, pathAgent)
+    try:
+        manager.RegisterAgent(pathAgent)
+        logger.debug("Registered iwd agent")
+    except:
+        logger.debug("Error in registering path")
+        logger.debug(traceback.print_exc(file=sys.stdout))
+
+    threading.Thread(target=delayedUnregister, args=(manager, pathAgent,
+                                                       mainloop,)).start()
+
+def unregisterAgent(manager, pathAgent):
+    try:
+        manager.UnregisterAgent(pathAgent)
+        logger.debug("UnRegistered iwd agent")
+    except:
+        logger.debug("Error in unregistering path")
+        logger.debug(traceback.print_exc(file=sys.stdout))
+
+def delayedUnregister(manager, path, mainloop):
+    counter = 1
+    while (utility.getCurrentlyConnectedDevice() == "" and counter < 10):
+         time.sleep(1)
+         counter = counter + 1
+         continue
+
+    time.sleep(1)
+    unregisterAgent(manager, path)
+    mainloop.quit()
+
+def init():
+    global logger
+    utility.initLogger()
+    logger = logging.getLogger(__name__)
+    dbus.mainloop.glib.DBusGMainLoop(set_as_default=True)
+    bus = dbus.SystemBus()
+    mainloop = GLib.MainLoop()
+    registerAgent(bus, mainloop)
+    threading.Thread(target=startAgent, args=(mainloop,)).start()
+
+if __name__ == '__main__':
+    init()