nixpkgs/nixos/tests/ulogd/ulogd.py
2024-01-18 16:28:42 -08:00

50 lines
3.0 KiB
Python

start_all()
machine.wait_for_unit("ulogd.service")
machine.systemctl("start network-online.target")
machine.wait_for_unit("network-online.target")
with subtest("Ulogd is running"):
machine.succeed("pgrep ulogd >&2")
# All packets show up twice in the logs
with subtest("Logs are collected"):
machine.succeed("ping -f 127.0.0.1 -c 5 >&2")
machine.succeed("sleep 2")
machine.wait_until_succeeds("du /var/log/ulogd.pcap")
_, echo_request_packets = machine.execute("tcpdump -r /var/log/ulogd.pcap icmp[0] == 8 and host 127.0.0.1")
expected, actual = 5 * 2, len(echo_request_packets.splitlines())
assert expected == actual, f"Expected {expected} ICMP request packets from pcap, got: {actual}"
_, echo_reply_packets = machine.execute("tcpdump -r /var/log/ulogd.pcap icmp[0] == 0 and host 127.0.0.1")
expected, actual = 5 * 2, len(echo_reply_packets.splitlines())
assert expected == actual, f"Expected {expected} ICMP reply packets from pcap, got: {actual}"
machine.wait_until_succeeds("du /var/log/ulogd_pkts.log")
_, echo_request_packets = machine.execute("grep TYPE=8 /var/log/ulogd_pkts.log")
expected, actual = 5 * 2, len(echo_request_packets.splitlines())
assert expected == actual, f"Expected {expected} ICMP request packets from logfile, got: {actual}"
_, echo_reply_packets = machine.execute("grep TYPE=0 /var/log/ulogd_pkts.log")
expected, actual = 5 * 2, len(echo_reply_packets.splitlines())
assert expected == actual, f"Expected {expected} ICMP reply packets from logfile, got: {actual}"
with subtest("Reloading service reopens log file"):
machine.succeed("mv /var/log/ulogd.pcap /var/log/old_ulogd.pcap")
machine.succeed("mv /var/log/ulogd_pkts.log /var/log/old_ulogd_pkts.log")
machine.succeed("systemctl reload ulogd.service")
machine.succeed("ping -f 127.0.0.1 -c 5 >&2")
machine.succeed("sleep 2")
machine.wait_until_succeeds("du /var/log/ulogd.pcap")
_, echo_request_packets = machine.execute("tcpdump -r /var/log/ulogd.pcap icmp[0] == 8 and host 127.0.0.1")
expected, actual = 5 * 2, len(echo_request_packets.splitlines())
assert expected == actual, f"Expected {expected} packets, got: {actual}"
_, echo_reply_packets = machine.execute("tcpdump -r /var/log/ulogd.pcap icmp[0] == 0 and host 127.0.0.1")
expected, actual = 5 * 2, len(echo_reply_packets.splitlines())
assert expected == actual, f"Expected {expected} packets, got: {actual}"
machine.wait_until_succeeds("du /var/log/ulogd_pkts.log")
_, echo_request_packets = machine.execute("grep TYPE=8 /var/log/ulogd_pkts.log")
expected, actual = 5 * 2, len(echo_request_packets.splitlines())
assert expected == actual, f"Expected {expected} ICMP request packets from logfile, got: {actual}"
_, echo_reply_packets = machine.execute("grep TYPE=0 /var/log/ulogd_pkts.log")
expected, actual = 5 * 2, len(echo_reply_packets.splitlines())
assert expected == actual, f"Expected {expected} ICMP reply packets from logfile, got: {actual}"