summaryrefslogtreecommitdiffstats
path: root/python/mozlint/test/test_editor.py
blob: 7a15a613a6efcb36c661ab73dcda60a2729651b7 (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
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
# 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/.

import os
import subprocess

import mozunit
import pytest

from mozlint import editor
from mozlint.result import Issue, ResultSummary

here = os.path.abspath(os.path.dirname(__file__))


@pytest.fixture
def capture_commands(monkeypatch):
    def inner(commands):
        def fake_subprocess_call(*args, **kwargs):
            commands.append(args[0])

        monkeypatch.setattr(subprocess, "call", fake_subprocess_call)

    return inner


@pytest.fixture
def result():
    result = ResultSummary("/fake/root")
    result.issues["foo.py"].extend(
        [
            Issue(
                linter="no-foobar",
                path="foo.py",
                lineno=1,
                message="Oh no!",
            ),
            Issue(
                linter="no-foobar",
                path="foo.py",
                lineno=3,
                column=10,
                message="To Yuma!",
            ),
        ]
    )
    return result


def test_no_editor(monkeypatch, capture_commands, result):
    commands = []
    capture_commands(commands)

    monkeypatch.delenv("EDITOR", raising=False)
    editor.edit_issues(result)
    assert commands == []


def test_no_issues(monkeypatch, capture_commands, result):
    commands = []
    capture_commands(commands)

    monkeypatch.setenv("EDITOR", "generic")
    result.issues = {}
    editor.edit_issues(result)
    assert commands == []


def test_vim(monkeypatch, capture_commands, result):
    commands = []
    capture_commands(commands)

    monkeypatch.setenv("EDITOR", "vim")
    editor.edit_issues(result)
    assert len(commands) == 1
    assert commands[0][0] == "vim"


def test_generic(monkeypatch, capture_commands, result):
    commands = []
    capture_commands(commands)

    monkeypatch.setenv("EDITOR", "generic")
    editor.edit_issues(result)
    assert len(commands) == len(result.issues)
    assert all(c[0] == "generic" for c in commands)
    assert all("foo.py" in c for c in commands)


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