26 lines
613 B
Python
26 lines
613 B
Python
import unittest
|
|
|
|
from elironwars.console import ConsoleUi
|
|
from elironwars.entrypoints import Entrypoints
|
|
|
|
|
|
class TestEntrypoints(unittest.TestCase):
|
|
|
|
def test_load_console_ui(self):
|
|
console_ep = 'console'
|
|
|
|
ep = Entrypoints()
|
|
self.assertIn(console_ep, ep.ui.names)
|
|
self.assertIsInstance(ep.load_ui(console_ep), ConsoleUi)
|
|
|
|
def test_load_unknown_ui(self):
|
|
unknown_ep = 'unknown'
|
|
|
|
ep = Entrypoints()
|
|
self.assertNotIn(unknown_ep, ep.ui.names)
|
|
self.assertIsNone(ep.load_ui(unknown_ep), ConsoleUi)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|