File: test_parser.py

package info (click to toggle)
lexicon 3.21.1-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 72,684 kB
  • sloc: python: 20,075; sh: 94; makefile: 11
file content (43 lines) | stat: -rw-r--r-- 1,315 bytes parent folder | download
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
"""Unit tests for the Lexicon CLI parser"""

import pytest

from lexicon._private import parser


def test_base_provider_parser():
    baseparser = parser.generate_base_provider_parser()
    parsed = baseparser.parse_args(["list", "capsulecd.com", "TXT"])
    assert parsed.action == "list"
    assert parsed.domain == "capsulecd.com"
    assert parsed.type == "TXT"
    assert parsed.ttl is None
    assert parsed.output == "TABLE"


def test_base_provider_parser_without_domain():
    baseparser = parser.generate_base_provider_parser()
    with pytest.raises(SystemExit):
        baseparser.parse_args(["list"])


def test_base_provider_parser_without_options():
    baseparser = parser.generate_base_provider_parser()
    with pytest.raises(SystemExit):
        baseparser.parse_args([])


def test_cli_main_parser():
    baseparser = parser.generate_cli_main_parser()
    parsed = baseparser.parse_args(["cloudflare", "list", "capsulecd.com", "TXT"])
    assert parsed.provider_name == "cloudflare"
    assert parsed.action == "list"
    assert parsed.domain == "capsulecd.com"
    assert parsed.type == "TXT"
    assert parsed.output == "TABLE"


def test_cli_main_parser_without_args():
    baseparser = parser.generate_cli_main_parser()
    with pytest.raises(SystemExit):
        baseparser.parse_args([])