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
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
|
#!/usr/bin/env python
# -*- coding: utf-8 eval: (blacken-mode 1) -*-
# SPDX-License-Identifier: ISC
#
# <template>.py
# Part of NetDEF Topology Tests
#
# Copyright (c) 2017 by
# Network Device Education Foundation, Inc. ("NetDEF")
#
"""
<template>.py: Test <template>.
"""
import sys
import pytest
from lib.topogen import Topogen, TopoRouter
from lib.topolog import logger
# TODO: select markers based on daemons used during test
# pytest module level markers
pytestmark = [
# pytest.mark.babeld,
# pytest.mark.bfdd,
# pytest.mark.bgpd,
# pytest.mark.eigrpd,
# pytest.mark.isisd,
# pytest.mark.ldpd,
# pytest.mark.nhrpd,
# pytest.mark.ospf6d,
pytest.mark.ospfd,
# pytest.mark.pathd,
# pytest.mark.pbrd,
# pytest.mark.pimd,
# pytest.mark.ripd,
# pytest.mark.ripngd,
# pytest.mark.sharpd,
# pytest.mark.staticd,
# pytest.mark.vrrpd,
]
# Function we pass to Topogen to create the topology
def build_topo(tgen):
"Build function"
# Create 2 routers
r1 = tgen.add_router("r1")
r2 = tgen.add_router("r2")
# Create a p2p connection between r1 and r2
tgen.add_link(r1, r2)
# Create a switch with one router connected to it to simulate a empty network.
switch = tgen.add_switch("s1")
switch.add_link(r1)
# Create a p2p connection between r1 and r2
switch = tgen.add_switch("s2")
switch.add_link(r1)
switch.add_link(r2)
# New form of setup/teardown using pytest fixture
@pytest.fixture(scope="module")
def tgen(request):
"Setup/Teardown the environment and provide tgen argument to tests"
# This function initiates the topology build with Topogen...
tgen = Topogen(build_topo, request.module.__name__)
# A basic topology similar to the above could also have be more easily specified
# using a # dictionary, remove the build_topo function and use the following
# instead:
#
# topodef = {
# "s1": "r1"
# "s2": ("r1", "r2")
# }
# tgen = Topogen(topodef, request.module.__name__)
# ... and here it calls initialization functions.
tgen.start_topology()
# This is a sample of configuration loading.
router_list = tgen.routers()
# For all routers arrange for:
# - starting zebra using config file from <rtrname>/zebra.conf
# - starting ospfd using an empty config file.
for rname, router in router_list.items():
router.load_config(TopoRouter.RD_ZEBRA, "zebra.conf")
router.load_config(TopoRouter.RD_OSPF)
# Start and configure the router daemons
tgen.start_router()
# Provide tgen as argument to each test function
yield tgen
# Teardown after last test runs
tgen.stop_topology()
# Fixture that executes before each test
@pytest.fixture(autouse=True)
def skip_on_failure(tgen):
if tgen.routers_have_failure():
pytest.skip("skipped because of previous test failure")
# ===================
# The tests functions
# ===================
def test_get_version(tgen):
"Test the logs the FRR version"
r1 = tgen.gears["r1"]
version = r1.vtysh_cmd("show version")
logger.info("FRR version is: " + version)
def test_connectivity(tgen):
"Test the logs the FRR version"
r1 = tgen.gears["r1"]
r2 = tgen.gears["r2"]
output = r1.cmd_raises("ping -c1 192.168.1.2")
output = r2.cmd_raises("ping -c1 192.168.3.1")
@pytest.mark.xfail
def test_expect_failure(tgen):
"A test that is current expected to fail but should be fixed"
assert False, "Example of temporary expected failure that will eventually be fixed"
@pytest.mark.skip
def test_will_be_skipped(tgen):
"A test that will be skipped"
assert False
# Memory leak test template
def test_memory_leak(tgen):
"Run the memory leak test and report results."
if not tgen.is_memleak_enabled():
pytest.skip("Memory leak test/report is disabled")
tgen.report_memory_leaks()
if __name__ == "__main__":
args = ["-s"] + sys.argv[1:]
sys.exit(pytest.main(args))
|
