25 lines
856 B
Python
25 lines
856 B
Python
import unittest
|
|
|
|
from dungeonbattle.menus import ArbitraryMenu, MainMenu, MainMenuValues
|
|
|
|
|
|
class TestMenus(unittest.TestCase):
|
|
def test_scroll_menu(self) -> None:
|
|
"""
|
|
Test to scroll the menu.
|
|
"""
|
|
arbitrary_menu = ArbitraryMenu([])
|
|
self.assertEqual(arbitrary_menu.position, 0)
|
|
|
|
main_menu = MainMenu()
|
|
self.assertEqual(main_menu.position, 0)
|
|
self.assertEqual(main_menu.validate(), MainMenuValues.START)
|
|
main_menu.go_up()
|
|
self.assertEqual(main_menu.validate(), MainMenuValues.START)
|
|
main_menu.go_down()
|
|
self.assertEqual(main_menu.validate(), MainMenuValues.SETTINGS)
|
|
main_menu.go_down()
|
|
self.assertEqual(main_menu.validate(), MainMenuValues.EXIT)
|
|
main_menu.go_down()
|
|
self.assertEqual(main_menu.validate(), MainMenuValues.EXIT)
|