-
Notifications
You must be signed in to change notification settings - Fork 54
Expand file tree
/
Copy pathtest_diff.py
More file actions
114 lines (106 loc) · 4.6 KB
/
test_diff.py
File metadata and controls
114 lines (106 loc) · 4.6 KB
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
# Copyright (c) 2019 Robin Jarry
# SPDX-License-Identifier: MIT
import os
import unittest
from libyang import (
BaseTypeAdded,
BaseTypeRemoved,
Context,
DefaultAdded,
EnumRemoved,
EnumStatusAdded,
EnumStatusRemoved,
ExtensionAdded,
NodeTypeAdded,
NodeTypeRemoved,
SNodeAdded,
SNodeRemoved,
StatusAdded,
StatusRemoved,
schema_diff,
)
OLD_YANG_DIR = os.path.join(os.path.dirname(__file__), "yang-old")
NEW_YANG_DIR = os.path.join(os.path.dirname(__file__), "yang")
# -------------------------------------------------------------------------------------
class DiffTest(unittest.TestCase):
expected_diffs = frozenset(
(
(BaseTypeAdded, "/yolo-system:conf/speed"),
(BaseTypeAdded, "/yolo-system:state/speed"),
(BaseTypeRemoved, "/yolo-system:conf/speed"),
(BaseTypeRemoved, "/yolo-system:state/speed"),
(DefaultAdded, "/yolo-system:conf/speed"),
(DefaultAdded, "/yolo-system:state/speed"),
(NodeTypeAdded, "/yolo-system:conf/number"),
(NodeTypeAdded, "/yolo-system:state/number"),
(NodeTypeRemoved, "/yolo-system:conf/number"),
(NodeTypeRemoved, "/yolo-system:state/number"),
(SNodeAdded, "/yolo-system:conf/full"),
(SNodeAdded, "/yolo-system:state/full"),
(SNodeAdded, "/yolo-system:conf/hostname-ref"),
(SNodeAdded, "/yolo-system:conf/url/enabled"),
(SNodeAdded, "/yolo-system:conf/url/fetch"),
(
SNodeAdded,
"/yolo-system:conf/url/fetch/input/timeout",
),
(
SNodeAdded,
"/yolo-system:conf/url/fetch/output/result",
),
(SNodeAdded, "/yolo-system:state/hostname-ref"),
(SNodeAdded, "/yolo-system:state/url/enabled"),
(SNodeAdded, "/yolo-system:state/url/fetch"),
(
SNodeAdded,
"/yolo-system:state/url/fetch/input/timeout",
),
(
SNodeAdded,
"/yolo-system:state/url/fetch/output/result",
),
(StatusAdded, "/yolo-system:conf/deprecated-leaf"),
(StatusAdded, "/yolo-system:conf/obsolete-leaf"),
(StatusAdded, "/yolo-system:state/deprecated-leaf"),
(StatusAdded, "/yolo-system:state/obsolete-leaf"),
(StatusRemoved, "/yolo-system:conf/deprecated-leaf"),
(StatusRemoved, "/yolo-system:conf/obsolete-leaf"),
(StatusRemoved, "/yolo-system:state/deprecated-leaf"),
(StatusRemoved, "/yolo-system:state/obsolete-leaf"),
(SNodeAdded, "/yolo-system:alarm-triggered"),
(SNodeAdded, "/yolo-system:alarm-triggered/severity"),
(SNodeAdded, "/yolo-system:alarm-triggered/description"),
(SNodeAdded, "/yolo-system:config-change"),
(SNodeAdded, "/yolo-system:config-change/edit"),
(SNodeAdded, "/yolo-system:config-change/edit/target"),
(EnumRemoved, "/yolo-system:conf/url/proto"),
(EnumRemoved, "/yolo-system:state/url/proto"),
(EnumStatusAdded, "/yolo-system:conf/url/proto"),
(EnumStatusAdded, "/yolo-system:state/url/proto"),
(ExtensionAdded, "/yolo-system:conf/url/proto"),
(ExtensionAdded, "/yolo-system:state/url/proto"),
(EnumStatusRemoved, "/yolo-system:conf/url/proto"),
(EnumStatusRemoved, "/yolo-system:state/url/proto"),
(SNodeAdded, "/yolo-system:conf/pill/red/out"),
(SNodeAdded, "/yolo-system:state/pill/red/out"),
(SNodeAdded, "/yolo-system:conf/pill/blue/in"),
(SNodeAdded, "/yolo-system:state/pill/blue/in"),
(SNodeAdded, "/yolo-system:alarm-triggered/severity"),
(SNodeAdded, "/yolo-system:alarm-triggered/description"),
)
)
def test_diff(self):
with Context(OLD_YANG_DIR, compile_obsolete=True) as ctx_old, Context(
NEW_YANG_DIR, compile_obsolete=True
) as ctx_new:
mod = ctx_old.load_module("yolo-system")
mod.feature_enable_all()
mod = ctx_new.load_module("yolo-system")
mod.feature_enable_all()
diffs = []
for d in schema_diff(ctx_old, ctx_new):
if isinstance(d, (SNodeAdded, SNodeRemoved)):
diffs.append((d.__class__, d.node.schema_path()))
else:
diffs.append((d.__class__, d.new.schema_path()))
self.assertEqual(frozenset(diffs), self.expected_diffs)