32 lines
1.5 KiB
Python
32 lines
1.5 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
|
||
|
|
from typing import List
|
||
|
|
from traceroute import _parse_traceroute, TraceElem
|
||
|
|
from datetime import timedelta
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
|
||
|
|
def parse_traceroute(lines):
|
||
|
|
"""Testing helper."""
|
||
|
|
return list(_parse_traceroute(lines))
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.parametrize("example, expected", [
|
||
|
|
# Basic case, one match
|
||
|
|
("3 10.60.142.2 (10.60.142.2) 117.502 ms",
|
||
|
|
[TraceElem("10.60.142.2", "10.60.142.2", timedelta(milliseconds=117.502))]),
|
||
|
|
# Multiple matches on one line
|
||
|
|
("3 10.60.142.2 (10.60.142.2) 117.502 ms 10.60.142.3 (10.60.142.3) 75.624 ms 10.60.142.2 (10.60.142.2) 117.709 ms",
|
||
|
|
[TraceElem("10.60.142.2", "10.60.142.2", timedelta(milliseconds=117.502)),
|
||
|
|
TraceElem("10.60.142.3", "10.60.142.3", timedelta(milliseconds=75.624)),
|
||
|
|
TraceElem("10.60.142.2", "10.60.142.2", timedelta(milliseconds=117.709))]),
|
||
|
|
# Context sensitive case - traceroute doesn't always print the host & IP.
|
||
|
|
("7 ae-501-ar01.denver.co.denver.comcast.net (96.216.22.130) 41.920 ms 41.893 ms 74.385 ms",
|
||
|
|
[TraceElem("ae-501-ar01.denver.co.denver.comcast.net", "96.216.22.130", timedelta(milliseconds=41.920)),
|
||
|
|
TraceElem("ae-501-ar01.denver.co.denver.comcast.net", "96.216.22.130", timedelta(milliseconds=41.893)),
|
||
|
|
TraceElem("ae-501-ar01.denver.co.denver.comcast.net", "96.216.22.130", timedelta(milliseconds=74.385))]),
|
||
|
|
])
|
||
|
|
def test_examples(example: str, expected: List[TraceElem]):
|
||
|
|
assert parse_traceroute(example.splitlines()) == expected
|