summaryrefslogtreecommitdiffstats
path: root/python/mozterm/test/test_widgets.py
blob: f1664f62c57736c5d03fa3930873440653cf0b89 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
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

from io import StringIO

import mozunit
import pytest

from mozterm import Terminal
from mozterm.widgets import Footer


@pytest.fixture
def terminal(monkeypatch):
    blessings = pytest.importorskip("blessings")

    kind = "xterm-256color"
    try:
        term = Terminal(stream=StringIO(), force_styling=True, kind=kind)
    except blessings.curses.error:
        pytest.skip("terminal '{}' not found".format(kind))

    # For some reason blessings returns None for width/height though a comment
    # says that shouldn't ever happen.
    monkeypatch.setattr(term, "_height_and_width", lambda: (100, 100))
    return term


def test_footer(terminal):
    footer = Footer(terminal=terminal)
    footer.write(
        [
            ("dim", "foo"),
            ("green", "bar"),
        ]
    )
    value = terminal.stream.getvalue()
    expected = "\x1b7\x1b[2mfoo\x1b(B\x1b[m \x1b[32mbar\x1b(B\x1b[m\x1b8"
    assert value == expected

    footer.clear()
    value = terminal.stream.getvalue()[len(value) :]
    expected = "\x1b[1G\x1b[K"
    assert value == expected


if __name__ == "__main__":
    mozunit.main()