2021-05-16 16:20:03 +00:00
|
|
|
{ system ? builtins.currentSystem
|
|
|
|
, config ? { }
|
|
|
|
, pkgs ? import ../.. { inherit system config; }
|
|
|
|
}:
|
|
|
|
|
|
|
|
with import ../lib/testing-python.nix { inherit system pkgs; };
|
|
|
|
with pkgs.lib;
|
|
|
|
|
|
|
|
let assertions = rec {
|
|
|
|
path = program: path: ''
|
|
|
|
with subtest("The path of ${program} should be ${path}"):
|
|
|
|
p = machine.succeed("type -p \"${program}\" | head -c -1")
|
|
|
|
assert p == "${path}", f"${program} is {p}, expected ${path}"
|
|
|
|
'';
|
|
|
|
unit = name: state: ''
|
|
|
|
with subtest("Unit ${name} should be ${state}"):
|
2021-05-21 21:11:20 +00:00
|
|
|
if "${state}" == "active":
|
|
|
|
machine.wait_for_unit("${name}")
|
|
|
|
else:
|
|
|
|
machine.require_unit_state("${name}", "${state}")
|
2021-05-16 16:20:03 +00:00
|
|
|
'';
|
|
|
|
version = ''
|
|
|
|
import re
|
|
|
|
|
|
|
|
with subtest("binary should report the correct version"):
|
|
|
|
pkgver = "${pkgs.atop.version}"
|
|
|
|
ver = re.sub(r'(?s)^Version: (\d\.\d\.\d).*', r'\1', machine.succeed("atop -V"))
|
|
|
|
assert ver == pkgver, f"Version is `{ver}`, expected `{pkgver}`"
|
|
|
|
'';
|
|
|
|
atoprc = contents:
|
|
|
|
if builtins.stringLength contents > 0 then ''
|
|
|
|
with subtest("/etc/atoprc should have the correct contents"):
|
|
|
|
f = machine.succeed("cat /etc/atoprc")
|
|
|
|
assert f == "${contents}", f"/etc/atoprc contents: '{f}', expected '${contents}'"
|
|
|
|
'' else ''
|
|
|
|
with subtest("/etc/atoprc should not be present"):
|
|
|
|
machine.succeed("test ! -e /etc/atoprc")
|
|
|
|
'';
|
|
|
|
wrapper = present:
|
|
|
|
if present then path "atop" "/run/wrappers/bin/atop" + ''
|
|
|
|
with subtest("Wrapper should be setuid root"):
|
|
|
|
stat = machine.succeed("stat --printf '%a %u' /run/wrappers/bin/atop")
|
|
|
|
assert stat == "4511 0", f"Wrapper stat is {stat}, expected '4511 0'"
|
|
|
|
''
|
|
|
|
else path "atop" "/run/current-system/sw/bin/atop";
|
|
|
|
atopService = present:
|
|
|
|
if present then
|
|
|
|
unit "atop.service" "active"
|
|
|
|
+ ''
|
2021-05-21 21:11:20 +00:00
|
|
|
with subtest("atop.service should write some data to /var/log/atop"):
|
|
|
|
|
|
|
|
def has_data_files(last: bool) -> bool:
|
|
|
|
files = int(machine.succeed("ls -1 /var/log/atop | wc -l"))
|
|
|
|
if files == 0:
|
|
|
|
machine.log("Did not find at least one 1 data file")
|
|
|
|
if not last:
|
|
|
|
machine.log("Will retry...")
|
|
|
|
return False
|
|
|
|
return True
|
|
|
|
|
|
|
|
with machine.nested("Waiting for data files"):
|
|
|
|
retry(has_data_files)
|
2021-05-16 16:20:03 +00:00
|
|
|
'' else unit "atop.service" "inactive";
|
|
|
|
atopRotateTimer = present:
|
|
|
|
unit "atop-rotate.timer" (if present then "active" else "inactive");
|
|
|
|
atopacctService = present:
|
|
|
|
if present then
|
|
|
|
unit "atopacct.service" "active"
|
|
|
|
+ ''
|
|
|
|
with subtest("atopacct.service should enable process accounting"):
|
2021-05-21 21:11:20 +00:00
|
|
|
machine.wait_until_succeeds("test -f /run/pacct_source")
|
2021-05-16 16:20:03 +00:00
|
|
|
|
|
|
|
with subtest("atopacct.service should write data to /run/pacct_shadow.d"):
|
2021-05-21 21:11:20 +00:00
|
|
|
|
|
|
|
def has_data_files(last: bool) -> bool:
|
|
|
|
files = int(machine.succeed("ls -1 /run/pacct_shadow.d | wc -l"))
|
|
|
|
if files == 0:
|
|
|
|
machine.log("Did not find at least one 1 data file")
|
|
|
|
if not last:
|
|
|
|
machine.log("Will retry...")
|
|
|
|
return False
|
|
|
|
return True
|
|
|
|
|
|
|
|
with machine.nested("Waiting for data files"):
|
|
|
|
retry(has_data_files)
|
2021-05-16 16:20:03 +00:00
|
|
|
'' else unit "atopacct.service" "inactive";
|
|
|
|
netatop = present:
|
|
|
|
if present then
|
|
|
|
unit "netatop.service" "active"
|
|
|
|
+ ''
|
|
|
|
with subtest("The netatop kernel module should be loaded"):
|
|
|
|
out = machine.succeed("modprobe -n -v netatop")
|
|
|
|
assert out == "", f"Module should be loaded already, but modprobe would have done {out}."
|
|
|
|
'' else ''
|
|
|
|
with subtest("The netatop kernel module should be absent"):
|
|
|
|
machine.fail("modprobe -n -v netatop")
|
|
|
|
'';
|
|
|
|
atopgpu = present:
|
|
|
|
if present then
|
|
|
|
(unit "atopgpu.service" "active") + (path "atopgpud" "/run/current-system/sw/bin/atopgpud")
|
|
|
|
else (unit "atopgpu.service" "inactive") + ''
|
|
|
|
with subtest("atopgpud should not be present"):
|
|
|
|
machine.fail("type -p atopgpud")
|
|
|
|
'';
|
|
|
|
};
|
|
|
|
in
|
|
|
|
{
|
|
|
|
justThePackage = makeTest {
|
|
|
|
name = "atop-justThePackage";
|
2022-03-20 23:15:30 +00:00
|
|
|
nodes.machine = {
|
2021-05-16 16:20:03 +00:00
|
|
|
environment.systemPackages = [ pkgs.atop ];
|
|
|
|
};
|
|
|
|
testScript = with assertions; builtins.concatStringsSep "\n" [
|
|
|
|
version
|
|
|
|
(atoprc "")
|
|
|
|
(wrapper false)
|
|
|
|
(atopService false)
|
|
|
|
(atopRotateTimer false)
|
|
|
|
(atopacctService false)
|
|
|
|
(netatop false)
|
|
|
|
(atopgpu false)
|
|
|
|
];
|
|
|
|
};
|
|
|
|
defaults = makeTest {
|
|
|
|
name = "atop-defaults";
|
2022-03-20 23:15:30 +00:00
|
|
|
nodes.machine = {
|
2021-05-14 00:11:05 +00:00
|
|
|
programs.atop = {
|
|
|
|
enable = true;
|
|
|
|
};
|
|
|
|
};
|
2021-05-16 16:20:03 +00:00
|
|
|
testScript = with assertions; builtins.concatStringsSep "\n" [
|
|
|
|
version
|
|
|
|
(atoprc "")
|
|
|
|
(wrapper false)
|
|
|
|
(atopService true)
|
|
|
|
(atopRotateTimer true)
|
|
|
|
(atopacctService true)
|
|
|
|
(netatop false)
|
|
|
|
(atopgpu false)
|
|
|
|
];
|
|
|
|
};
|
|
|
|
minimal = makeTest {
|
|
|
|
name = "atop-minimal";
|
2022-03-20 23:15:30 +00:00
|
|
|
nodes.machine = {
|
2021-05-14 00:11:05 +00:00
|
|
|
programs.atop = {
|
|
|
|
enable = true;
|
2021-05-16 16:20:03 +00:00
|
|
|
atopService.enable = false;
|
|
|
|
atopRotateTimer.enable = false;
|
|
|
|
atopacctService.enable = false;
|
2021-05-14 00:11:05 +00:00
|
|
|
};
|
|
|
|
};
|
2021-05-16 16:20:03 +00:00
|
|
|
testScript = with assertions; builtins.concatStringsSep "\n" [
|
|
|
|
version
|
|
|
|
(atoprc "")
|
|
|
|
(wrapper false)
|
|
|
|
(atopService false)
|
|
|
|
(atopRotateTimer false)
|
|
|
|
(atopacctService false)
|
|
|
|
(netatop false)
|
|
|
|
(atopgpu false)
|
|
|
|
];
|
|
|
|
};
|
|
|
|
netatop = makeTest {
|
|
|
|
name = "atop-netatop";
|
2022-03-20 23:15:30 +00:00
|
|
|
nodes.machine = {
|
2021-05-14 00:11:05 +00:00
|
|
|
programs.atop = {
|
|
|
|
enable = true;
|
2021-05-16 16:20:03 +00:00
|
|
|
netatop.enable = true;
|
2021-05-14 00:11:05 +00:00
|
|
|
};
|
|
|
|
};
|
2021-05-16 16:20:03 +00:00
|
|
|
testScript = with assertions; builtins.concatStringsSep "\n" [
|
|
|
|
version
|
|
|
|
(atoprc "")
|
|
|
|
(wrapper false)
|
|
|
|
(atopService true)
|
|
|
|
(atopRotateTimer true)
|
|
|
|
(atopacctService true)
|
|
|
|
(netatop true)
|
|
|
|
(atopgpu false)
|
|
|
|
];
|
|
|
|
};
|
|
|
|
atopgpu = makeTest {
|
|
|
|
name = "atop-atopgpu";
|
2022-03-20 23:15:30 +00:00
|
|
|
nodes.machine = {
|
2021-05-16 16:20:03 +00:00
|
|
|
nixpkgs.config.allowUnfreePredicate = pkg: builtins.elem (getName pkg) [
|
|
|
|
"cudatoolkit"
|
|
|
|
];
|
2021-05-14 00:11:05 +00:00
|
|
|
|
|
|
|
programs.atop = {
|
|
|
|
enable = true;
|
2021-05-16 16:20:03 +00:00
|
|
|
atopgpu.enable = true;
|
2021-05-14 00:11:05 +00:00
|
|
|
};
|
|
|
|
};
|
2021-05-16 16:20:03 +00:00
|
|
|
testScript = with assertions; builtins.concatStringsSep "\n" [
|
|
|
|
version
|
|
|
|
(atoprc "")
|
|
|
|
(wrapper false)
|
|
|
|
(atopService true)
|
|
|
|
(atopRotateTimer true)
|
|
|
|
(atopacctService true)
|
|
|
|
(netatop false)
|
|
|
|
(atopgpu true)
|
|
|
|
];
|
|
|
|
};
|
|
|
|
everything = makeTest {
|
|
|
|
name = "atop-everthing";
|
2022-03-20 23:15:30 +00:00
|
|
|
nodes.machine = {
|
2021-05-16 16:20:03 +00:00
|
|
|
nixpkgs.config.allowUnfreePredicate = pkg: builtins.elem (getName pkg) [
|
2021-05-14 00:11:05 +00:00
|
|
|
"cudatoolkit"
|
|
|
|
];
|
2021-05-16 16:20:03 +00:00
|
|
|
|
2021-05-14 00:11:05 +00:00
|
|
|
programs.atop = {
|
|
|
|
enable = true;
|
2021-05-16 16:20:03 +00:00
|
|
|
settings = {
|
|
|
|
flags = "faf1";
|
|
|
|
interval = 2;
|
|
|
|
};
|
|
|
|
setuidWrapper.enable = true;
|
|
|
|
netatop.enable = true;
|
2021-05-14 00:11:05 +00:00
|
|
|
atopgpu.enable = true;
|
|
|
|
};
|
|
|
|
};
|
2021-05-16 16:20:03 +00:00
|
|
|
testScript = with assertions; builtins.concatStringsSep "\n" [
|
|
|
|
version
|
|
|
|
(atoprc "flags faf1\\ninterval 2\\n")
|
|
|
|
(wrapper true)
|
|
|
|
(atopService true)
|
|
|
|
(atopRotateTimer true)
|
|
|
|
(atopacctService true)
|
|
|
|
(netatop true)
|
|
|
|
(atopgpu true)
|
|
|
|
];
|
2021-05-14 00:11:05 +00:00
|
|
|
};
|
2021-05-16 16:20:03 +00:00
|
|
|
}
|