summaryrefslogtreecommitdiff
path: root/tests/test_yang.py
blob: e2449b59f7415f4722f72cff4c53fe8b4c6f1c92 (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
93
94
95
96
97
98
99
100
101
102
103
# -*- coding: utf-8 -*-
"""
    Basic Yang Test
    ~~~~~~~~~~~~~~~~~~~~

    :copyright: Copyright 2006-2020 by the Pygments team, see AUTHORS.
    :license: BSD, see LICENSE for details.
"""

import pytest

from pygments.token import Operator, Number, Text, Token
from pygments.lexers import YangLexer

@pytest.fixture(scope='module')
def lexer():
    yield YangLexer()

def test_namespace_1(lexer):
    """
    Namespace `urn:test:std:yang` should not be explicitly highlighted
    """
    fragment = 'namespace urn:test:std:yang;\n'
    tokens = [
        (Token.Keyword, 'namespace'),
        (Token.Text.Whitespace, ' '),
        (Token.Name.Variable, 'urn:test:std:yang'),
        (Token.Punctuation, ';'),
        (Token.Text.Whitespace, '\n'),
    ]
    assert list(lexer.get_tokens(fragment)) == tokens

def test_namespace_2(lexer):
    """
    namespace-prefix `yang` should be explicitly highlighted
    """
    fragment = 'type yang:counter64;\n'
    tokens = [
        (Token.Keyword, 'type'),
        (Token.Text.Whitespace, ' '),
        (Token.Name.Namespace, 'yang'),
        (Token.Punctuation, ':'),
        (Token.Name.Variable, 'counter64'),
        (Token.Punctuation, ';'),
        (Token.Text.Whitespace, '\n'),
    ]
    assert list(lexer.get_tokens(fragment)) == tokens

def test_revision_date(lexer):
    """
    Revision-date `2020-08-03` should be explicitly highlighted
    """
    fragment = 'revision 2020-03-08{\n'
    tokens = [
        (Token.Keyword, 'revision'),
        (Token.Text.Whitespace, ' '),
        (Token.Name.Label, '2020-03-08'),
        (Token.Punctuation, '{'),
        (Token.Text.Whitespace, '\n'),
    ]
    assert list(lexer.get_tokens(fragment)) == tokens

def test_integer_value(lexer):
    """
    Integer value `5` should be explicitly highlighted
    """
    fragment = 'value 5;\n'
    tokens = [
        (Token.Keyword, 'value'),
        (Token.Text.Whitespace, ' '),
        (Token.Number.Integer, '5'),
        (Token.Punctuation, ';'),
        (Token.Text.Whitespace, '\n'),
    ]
    assert list(lexer.get_tokens(fragment)) == tokens

def test_string_value(lexer):
    """
    String value `"5"` should be not explicitly highlighted
    """
    fragment = 'value "5";\n'
    tokens = [
        (Token.Keyword, 'value'),
        (Token.Text.Whitespace, ' '),
        (Token.String.Double, '"5"'),
        (Token.Punctuation, ';'),
        (Token.Text.Whitespace, '\n'),
    ]
    assert list(lexer.get_tokens(fragment)) == tokens

def test_float_value(lexer):
    """
    Float value `1.1` should be explicitly highlighted
    """
    fragment = 'yang-version 1.1;\n'
    tokens = [
        (Token.Keyword, 'yang-version'),
        (Token.Text.Whitespace, ' '),
        (Token.Number.Float, '1.1'),
        (Token.Punctuation, ';'),
        (Token.Text.Whitespace, '\n'),
    ]
    assert list(lexer.get_tokens(fragment)) == tokens