def test_set_indicator_simple() -> None: mainloop = mock.Mock(urwid.MainLoop) screen = ScreenManagerImpl(mainloop) screen.set_indicator("test", "Hello World", protocol.IndicatorLocation.LEFT, 0) view = _render(mainloop.widget) assert "Hello World" in view screen.remove_indicator("test") view = _render(mainloop.widget) assert "Hello World" not in view
def test_set_indicator_ordering() -> None: # GIVEN mainloop = mock.Mock(urwid.MainLoop) screen = ScreenManagerImpl(mainloop) screen.set_indicator("3", "Three", protocol.IndicatorLocation.RIGHT, 3) screen.set_indicator("1", "One", protocol.IndicatorLocation.RIGHT, 1) screen.set_indicator("2", ["T", "w", "o"], protocol.IndicatorLocation.RIGHT, 2) # WHEN view = _render(mainloop.widget) # THEN assert "One Two Three" in view