import os
import unittest

from aprsd import plugin
from aprsd.utils import package


class TestPackage(unittest.TestCase):
    def test_plugin_type(self):
        self.assertEqual(
            package.plugin_type(plugin.APRSDRegexCommandPluginBase), 'RegexCommand'
        )
        self.assertEqual(
            package.plugin_type(plugin.APRSDWatchListPluginBase), 'WatchList'
        )
        self.assertEqual(package.plugin_type(plugin.APRSDPluginBase), 'APRSDPluginBase')

    def test_is_plugin(self):
        class TestPlugin(plugin.APRSDPluginBase):
            def setup(self):
                pass

            def filter(self, packet):
                pass

            def process(self, packet):
                pass

        class NonPlugin:
            pass

        self.assertTrue(package.is_plugin(TestPlugin))
        self.assertFalse(package.is_plugin(NonPlugin))

    def test_walk_package(self):
        import aprsd.utils

        result = package.walk_package(aprsd.utils)
        # walk_package returns an iterator, so we just check it's not None
        self.assertIsNotNone(result)

    def test_get_module_info(self):
        # Test with a specific, limited directory to avoid hanging
        # Use the aprsd/utils directory which is small and safe
        import aprsd.utils

        package_name = 'aprsd.utils'
        module_name = 'package'
        # Get the actual path to aprsd/utils directory
        module_path = os.path.dirname(aprsd.utils.__file__)
        module_info = package.get_module_info(package_name, module_name, module_path)
        # The result should be a list (even if empty)
        self.assertIsInstance(module_info, list)

    def test_is_aprsd_package(self):
        self.assertTrue(package.is_aprsd_package('aprsd_plugin'))
        self.assertFalse(package.is_aprsd_package('other'))

    def test_is_aprsd_extension(self):
        self.assertTrue(package.is_aprsd_extension('aprsd_extension_plugin'))
        self.assertFalse(package.is_aprsd_extension('other'))

    def test_get_installed_aprsd_items(self):
        plugins, extensions = package.get_installed_aprsd_items()
        self.assertIsNotNone(plugins)
        self.assertIsNotNone(extensions)

    def test_get_installed_plugins(self):
        plugins = package.get_installed_plugins()
        self.assertIsNotNone(plugins)

    def test_get_installed_extensions(self):
        extensions = package.get_installed_extensions()
        self.assertIsNotNone(extensions)

    def test_get_pypi_packages(self):
        packages = package.get_pypi_packages()
        self.assertIsNotNone(packages)

    def test_log_installed_extensions_and_plugins(self):
        package.log_installed_extensions_and_plugins()


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