iwd/autotests/testHiddenNetworks/wpa_network_test.py

#!/usr/bin/python3

import unittest
import sys

sys.path.append('../util')
import iwd
import validation
from validation import TestHiddenNetworks
from iwd import IWD
from hostapd import HostapdCLI

class TestWpaNetwork(unittest.TestCase):
    '''
    The bellow test cases excesise the following connection scenarios:

    Network config is
    present at start time:  Connect:  AutoConnect:  Result:
    --------------------------------------------------------------------------
    False                   True                    Connection succeeds
    True                              True          Connection succeeds
    '''

    def test_wpa(self):
        tca = TestHiddenNetworks()
        tca.validate('ssidHiddenWPA', False, None, True)
        tca.validate('ssidHiddenWPA', True, None, True)

    @classmethod
    def setUpClass(cls):
        cls.disabled = [HostapdCLI('ssidHiddenOpen.conf'),
                        HostapdCLI('ssidOpen.conf'),
                        HostapdCLI('ssidOverlap1.conf'),
                        HostapdCLI('ssidOverlap2.conf'),
                        HostapdCLI('ssidSomeHidden.conf')]

        for hapd in cls.disabled:
            hapd.disable()

    @classmethod
    def tearDownClass(cls):
        IWD.clear_storage()

        for hapd in cls.disabled:
            hapd.reload()

if __name__ == '__main__':
    unittest.main(exit=True)