import unittest

from utils import *

from ksc import Ksc

from mock import Mock, patch

class ReadListTest(unittest.TestCase):

    """

    Test reading stablelist

    """

    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.nonstable_symbols_used) == 1

        assert len(k.stable_symbols) == 1

if __name__ == '__main__':

    unittest.main()