iwd/autotests/test1AP/APShutdownTest.py

#!/usr/bin/python3

#built-in python libraries
import unittest
from gi.repository import GLib
import dbus
import time
import logging
import os
from subprocess import Popen, PIPE, STDOUT, call, check_output
import sys
sys.path.append('../utility') #needed to import all the utility modules
import utility
import pty
import signal

class TestAPShutdown(unittest.TestCase):
    def doConnectDisconnect(self):
        objectList = utility.getObjectList(bus)

        # start simpleAgent
        master, slave = pty.openpty()
        proc = Popen([sys.executable, '../utility/simpleAgent.py'],
                     stdin=PIPE, stdout=slave, close_fds=True)
        stdout_handle = os.fdopen(master)
        if stdout_handle.readline().rstrip() == "AGENT_REGISTERED":
            logger.debug("Agent Registered")
        else:
            logger.debug("Agent failed to register")

        # close the handles
        stdout_handle.close()
        os.close(slave)

        networkToConnect = utility.getNetworkToConnectTo(objectList,
                                                         "IntelWIFI")
        # 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 == ""):
            time.sleep(2)
            logger.debug("RESTART PROGRAM")
            os.execl(sys.executable, sys.executable, * sys.argv)

        self.assertNotEqual(networkToConnect, "")
        network = dbus.Interface(bus.get_object(utility.IWD_SERVICE,
                                                networkToConnect),
                                 utility.IWD_NETWORK_INTERFACE)

        status = utility.connect(networkToConnect, self, mainloop, bus)

        if status == False:
            #terminate proc
            proc.terminate()
            return
        logger.info("Currently connected to: %s",
                   utility.getCurrentlyConnectedNetworkName())
        self.assertEqual(utility.getCurrentlyConnectedNetworkName(),
                          "IntelWIFI")

        #kill AP
        pid_bytes = check_output("pgrep -o -x hostapd", shell=True)
        pid = pid_bytes.decode('utf-8')
        logger.debug("shutting down AP with pid: %s", pid)
        os.system("kill " + pid)

        #should not be connected to any network after AP shutdown
        logger.info("After AP shutdown, connected to: %s",
                   utility.getCurrentlyConnectedNetworkName())
        self.assertEqual(utility.getCurrentlyConnectedNetworkName(),
                          "")
        #terminate proc
        proc.terminate()

    # connect to network A. Wait for 2 seconds. Disconnect.
    def test_APShutdown(self):
        logger.info(sys._getframe().f_code.co_name)
        while (True):
            if bus.name_has_owner(utility.IWD_SERVICE) == True:
                break
        self.doConnectDisconnect()

    @classmethod
    def setUpClass(cls):
        global logger, bus, mainloop
        os.system("rm -rf /var/lib/iwd")
        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)