Fix crash on startup when terminal doesnt support colors

2017-11-04 14:10:50 -04:00
parent 331492c766
commit 00df45bd34
3 changed files with 15 additions and 4 deletions
--- a/tests/test_terminal.py
+++ b/tests/test_terminal.py
@@ -596,3 +596,16 @@ def test_set_theme(terminal, stdscr):
    terminal.set_theme(theme=theme)
    assert terminal.theme.monochrome
    stdscr.bkgd.assert_called_once_with(' ', 0)
+
+
+def test_set_theme_no_colors(terminal, stdscr):
+
+    with mock.patch('curses.has_colors') as has_colors:
+        has_colors.return_value = False
+
+        terminal.set_theme()
+        assert terminal.theme.monochrome
+
+        theme = Theme(monochrome=False)
+        terminal.set_theme(theme=theme)
+        assert terminal.theme.monochrome