import os
import sys
import unittest

sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..', 'payload', 'user', 'remote_access', 'pager-webui'))
import server


def setUpModule():
    __import__('importlib').reload(server)


class ChannelBandTest(unittest.TestCase):
    def test_2g_channels(self):
        for ch in (1, 6, 11, 14):
            self.assertEqual(server.channel_band(ch), server.BAND_2G)

    def test_5g_channels(self):
        for ch in (36, 48, 100, 149, 165, 177):
            self.assertEqual(server.channel_band(ch), server.BAND_5G)

    def test_6g_channels(self):
        for ch in (181, 189, 197, 205, 213, 225, 233):
            self.assertEqual(server.channel_band(ch), server.BAND_6G)

    def test_invalid(self):
        for ch in (0, 15, 35, 178, 234, None, 'x'):
            self.assertIsNone(server.channel_band(ch))


class ChannelFreqTest(unittest.TestCase):
    def test_freqs(self):
        self.assertEqual(server.channel_freq(server.BAND_2G, 1), 2412)
        self.assertEqual(server.channel_freq(server.BAND_2G, 11), 2462)
        self.assertEqual(server.channel_freq(server.BAND_5G, 36), 5180)
        self.assertEqual(server.channel_freq(server.BAND_5G, 165), 5825)
        self.assertEqual(server.channel_freq(server.BAND_6G, 1), 5955)
        self.assertEqual(server.channel_freq(server.BAND_6G, 233), 7115)


class BandAuxTest(unittest.TestCase):
    def test_htmode(self):
        self.assertEqual(server.band_htmode(server.BAND_2G), 'HT20')
        self.assertEqual(server.band_htmode(server.BAND_5G), 'VHT80')
        self.assertEqual(server.band_htmode(server.BAND_6G), 'HE80')

    def test_radio(self):
        self.assertEqual(server.band_radio(server.BAND_2G), 'radio0')
        self.assertEqual(server.band_radio(server.BAND_5G), 'radio1')
        self.assertEqual(server.band_radio(server.BAND_6G), 'radio1')

    def test_dfs_marker(self):
        self.assertIn(52, server.DFS_CHANNELS)
        self.assertIn(144, server.DFS_CHANNELS)
        self.assertNotIn(149, server.DFS_CHANNELS)


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