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
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123

import json
import re

import pytest

from testlib.fixtures.file_helper import CopyFile, with_files
from testlib.fixtures.nix import Nix
from testlib.utils import get_global_asset

_files = {
    "search.nix": CopyFile("assets/test_search/search.nix"),
    "config.nix": get_global_asset("config.nix"),
}
_exclude_args = ["search", "-f", "search.nix"]
_search_args = [*_exclude_args, ""]


@pytest.fixture(autouse=True)
def commands(nix: Nix):
    nix.settings.add_xp_feature("nix-command")


@with_files(_files)
def test_pkg_name(nix: Nix):
    res = nix.nix([*_search_args, "hello"]).run().ok()
    assert len(res.stdout_plain.splitlines()) > 1


@with_files(_files)
def test_description(nix: Nix):
    res = nix.nix([*_search_args, "broken"]).run().ok()
    assert len(res.stdout_plain.splitlines()) > 1


@with_files(_files)
def test_no_exist(nix: Nix):
    res = nix.nix([*_search_args, "nosuchpackageexits"]).run().expect(1)
    assert "error: no results for the given search" in res.stderr_plain


@with_files(_files)
def test_multiple_args_exists(nix: Nix):
    res = nix.nix([*_search_args, "hello", "empty"]).run().ok()
    assert len(res.stdout_plain.splitlines()) == 2


@with_files(_files)
def test_multiple_args_no_exist(nix: Nix):
    res = nix.nix([*_search_args, "hello", "broken"]).run().expect(1)
    assert "error: no results for the given search" in res.stderr_plain


@with_files(_files)
def test_no_args(nix: Nix):
    res = nix.nix([*_search_args]).run().expect(1)
    assert "error: Must provide at least one regex!" in res.stderr_plain


@with_files(_files)
def test_match_all(nix: Nix):
    res = nix.nix([*_search_args, "^"]).run().ok()
    out = res.stdout_plain
    for pkg in ["foo", "bar", "hello"]:
        assert pkg in out


# FIXME(Jade): possibly not test this with colour in the future
@with_files(_files)
def test_overlap_1(nix: Nix):
    nix.env.set_env("FORCE_COLOR", "1")
    res = nix.nix([*_search_args, "oo", "foo", "oo"]).run().ok()
    out = res.stdout
    assert out == b"* \x1b[0;1m\x1b[32;1mfoo\x1b[0;1m\x1b[0m (5)\n"


@with_files(_files)
def test_overlap_2(nix: Nix):
    nix.env.set_env("FORCE_COLOR", "1")
    res = nix.nix([*_search_args, "broken b", "en bar"]).run().ok()
    out = res.stdout
    assert out == b"* \x1b[0;1mbar\x1b[0m (3)\n  \x1b[32;1mbroken bar\x1b[0m\n"


@with_files(_files)
def test_multiple_finds_o(nix: Nix):
    """Searching for 'o' should yield the 'o' in 'broken bar', the 'oo' in foo and 'o' in hello"""
    nix.env.set_env("FORCE_COLOR", "1")
    res = nix.nix([*_search_args, "o"]).run().ok()
    out = res.stdout_s
    assert len(re.findall(r"\x1b\[32;1mo{1,2}\x1b\[(0|0;1)m", out)) == 3


@with_files(_files)
def test_multiple_finds_b(nix: Nix):
    """Searching for 'b' should yield the 'b' in bar and the two 'b's in 'broken bar'"""
    nix.env.set_env("FORCE_COLOR", "1")
    res = nix.nix([*_search_args, "b"]).run().ok()
    out = res.stdout_s
    assert len(re.findall(r"\x1b\[32;1mb\x1b\[(0|0;1)m", out)) == 3


@with_files(_files)
def test_exclude_hello(nix: Nix):
    res = nix.nix([*_exclude_args, "", "^", "-e", "hello"]).run().ok()
    assert res.stdout_plain.count("hello") == 0


@with_files(_files)
def test_exclude_foo_regex(nix: Nix):
    res = nix.nix([*_exclude_args, "foo", "^", "--exclude", "foo|bar"]).run().expect(1)
    assert "error: no results for the given search" in res.stderr_plain


@with_files(_files)
def test_exclude_foo_multiple_args(nix: Nix):
    res = nix.nix([*_exclude_args, "foo", "^", "-e", "foo", "--exclude", "bar"]).run().expect(1)
    assert "error: no results for the given search" in res.stderr_plain


@with_files(_files)
def test_exclude_bar(nix: Nix):
    res = nix.nix([*_exclude_args, "", "^", "-e", "bar", "--json"]).run().ok()
    assert json.loads(res.stdout_plain).keys() == {"foo", "hello"}