import unittest
from utils import *
from ksc import Ksc
from mock import Mock, patch


class ReadListTest(unittest.TestCase):
    """
    Test reading whitelist
    """
    def runTest(self):
        data, _ = read_list("x86_64", "kabi-current")
        assert len(data) != 0


class ReadTotalListTest(unittest.TestCase):
    """
    Test reading all symbol names
    """
    def runTest(self):
        data = read_total_list()
        assert len(data) != 0


class RunCommandTest(unittest.TestCase):
    """
    To test our own set function
    """
    def runTest(self):
        data = run('uname -a')
        self.assertTrue(data.startswith('Linux'))


class GetConfigTest(unittest.TestCase):
    """
    To test our own set function
    """
    def runTest(self):
        data = getconfig('./data/ksc.conf', True)
        assert 'user' in data
        assert 'partner' in data
        assert 'group' in data
        assert 'server' in data


class CreateBugTest(unittest.TestCase):
    """
    Code to test createbug function
    """
    def runTest(self):
        bugid = createbug('./data/ksc.conf', 'x86_64', True)  # This is mock


class ParseKOTest(unittest.TestCase):
    """
    Code to test parse_ko
    """
    @patch('ksc.run')
    def runTest(self, mock_run):
        mock_run.return_value = 'U add_disk\nU add_drv\nU call_rcu_bh'
        k = Ksc(mock=True)
        k.read_data('x86_64', 'kabi-current')
        k.parse_ko('./ksc.py')
        assert len(k.all_symbols_used) == 2
        assert len(k.nonwhite_symbols_used) == 1
        assert len(k.white_symbols) == 1


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