1 files changed, 51 insertions, 0 deletions
diff --git a/python/mozterm/test/test_terminal.py b/python/mozterm/test/test_terminal.py
new file mode 100644
index 0000000000..25b63545bb
--- /dev/null
+++ b/python/mozterm/test/test_terminal.py
@@ -0,0 +1,51 @@
+# This Source Code Form is subject to the terms of the Mozilla Public
+# License, v. 2.0. If a copy of the MPL was not distributed with this
+# file, You can obtain one at http://mozilla.org/MPL/2.0/.
+
+from __future__ import absolute_import, unicode_literals
+
+import os
+import sys
+
+import mozunit
+import pytest
+
+from mozterm import Terminal, NullTerminal
+
+
+def test_terminal():
+    blessings = pytest.importorskip("blessings")
+    term = Terminal()
+    assert isinstance(term, blessings.Terminal)
+
+    term = Terminal(disable_styling=True)
+    assert isinstance(term, NullTerminal)
+
+    del sys.modules["blessings"]
+    orig = sys.path[:]
+    for path in orig:
+        if "blessings" in path:
+            sys.path.remove(path)
+
+    term = Terminal()
+    assert isinstance(term, NullTerminal)
+
+    with pytest.raises(ImportError):
+        term = Terminal(raises=True)
+
+    sys.path = orig
+
+
+def test_null_terminal():
+    term = NullTerminal()
+    assert term.red("foo") == "foo"
+    assert term.red == ""
+    assert term.color(1) == ""
+    assert term.number_of_colors == 0
+    assert term.width == 0
+    assert term.height == 0
+    assert term.is_a_tty == os.isatty(sys.stdout.fileno())
+
+
+if __name__ == "__main__":
+    mozunit.main()