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
|
# SPDX-License-Identifier: GPL-2.0-or-later
#
# Copyright 2023 Quentin Young
import functools
from lib.topogen import get_topogen
from lib.topolog import logger
from lib import topotest
def check_ping(name, dest_addr, expect_connected, count, wait):
"""
Assert that ping to dest_addr is expected
* 'name': the router to set the ping from
* 'dest_addr': The destination ip address to ping
* 'expect_connected': True if ping is expected to pass
* 'count': how many echos to send
* 'wait': how long ping should wait to receive all replies
"""
def _check(name, dest_addr, match):
tgen = get_topogen()
output = tgen.gears[name].run("ping {} -c 1 -w 1".format(dest_addr))
logger.info(output)
if match not in output:
return "ping fail"
match = ", {} packet loss".format("0%" if expect_connected else "100%")
logger.info("[+] check {} {} {}".format(name, dest_addr, match))
tgen = get_topogen()
func = functools.partial(_check, name, dest_addr, match)
success, result = topotest.run_and_expect(func, None, count=count, wait=wait)
assert result is None, "Failed"
|
