98 lines
2.8 KiB
Nix
98 lines
2.8 KiB
Nix
|
let
|
||
|
|
clients = [
|
||
|
|
"ircclient1"
|
||
|
|
"ircclient2"
|
||
|
|
];
|
||
|
|
server = "ergochat";
|
||
|
|
ircPort = 6667;
|
||
|
|
channel = "nixos-cat";
|
||
|
|
iiDir = "/tmp/irc";
|
||
|
|
in
|
||
|
|
|
||
|
|
import ./make-test-python.nix ({ pkgs, lib, ... }: {
|
||
|
|
name = "ergochat";
|
||
|
|
nodes = {
|
||
|
|
"${server}" = {
|
||
|
|
networking.firewall.allowedTCPPorts = [ ircPort ];
|
||
|
|
services.ergochat = {
|
||
|
|
enable = true;
|
||
|
|
settings.server.motd = pkgs.writeText "ergo.motd" ''
|
||
|
|
The default MOTD doesn't contain the word "nixos" in it.
|
||
|
|
This one does.
|
||
|
|
'';
|
||
|
|
};
|
||
|
|
};
|
||
|
|
} // lib.listToAttrs (builtins.map (client: lib.nameValuePair client {
|
||
|
|
imports = [
|
||
|
|
./common/user-account.nix
|
||
|
|
];
|
||
|
|
|
||
|
|
systemd.services.ii = {
|
||
|
|
requires = [ "network.target" ];
|
||
|
|
wantedBy = [ "default.target" ];
|
||
|
|
|
||
|
|
serviceConfig = {
|
||
|
|
Type = "simple";
|
||
|
|
ExecPreStartPre = "mkdir -p ${iiDir}";
|
||
|
|
ExecStart = ''
|
||
|
|
${lib.getBin pkgs.ii}/bin/ii -n ${client} -s ${server} -i ${iiDir}
|
||
|
|
'';
|
||
|
|
User = "alice";
|
||
|
|
};
|
||
|
|
};
|
||
|
|
}) clients);
|
||
|
|
|
||
|
|
testScript =
|
||
|
|
let
|
||
|
|
msg = client: "Hello, my name is ${client}";
|
||
|
|
clientScript = client: [
|
||
|
|
''
|
||
|
|
${client}.wait_for_unit("network.target")
|
||
|
|
${client}.systemctl("start ii")
|
||
|
|
${client}.wait_for_unit("ii")
|
||
|
|
${client}.wait_for_file("${iiDir}/${server}/out")
|
||
|
|
''
|
||
|
|
# look for the custom text in the MOTD.
|
||
|
|
''
|
||
|
|
${client}.wait_until_succeeds("grep 'nixos' ${iiDir}/${server}/out")
|
||
|
|
''
|
||
|
|
# wait until first PING from server arrives before joining,
|
||
|
|
# so we don't try it too early
|
||
|
|
''
|
||
|
|
${client}.wait_until_succeeds("grep 'PING' ${iiDir}/${server}/out")
|
||
|
|
''
|
||
|
|
# join ${channel}
|
||
|
|
''
|
||
|
|
${client}.succeed("echo '/j #${channel}' > ${iiDir}/${server}/in")
|
||
|
|
${client}.wait_for_file("${iiDir}/${server}/#${channel}/in")
|
||
|
|
''
|
||
|
|
# send a greeting
|
||
|
|
''
|
||
|
|
${client}.succeed(
|
||
|
|
"echo '${msg client}' > ${iiDir}/${server}/#${channel}/in"
|
||
|
|
)
|
||
|
|
''
|
||
|
|
# check that all greetings arrived on all clients
|
||
|
|
] ++ builtins.map (other: ''
|
||
|
|
${client}.succeed(
|
||
|
|
"grep '${msg other}$' ${iiDir}/${server}/#${channel}/out"
|
||
|
|
)
|
||
|
|
'') clients;
|
||
|
|
|
||
|
|
# foldl', but requires a non-empty list instead of a start value
|
||
|
|
reduce = f: list:
|
||
|
|
builtins.foldl' f (builtins.head list) (builtins.tail list);
|
||
|
|
in ''
|
||
|
|
start_all()
|
||
|
|
${server}.systemctl("status ergochat")
|
||
|
|
${server}.wait_for_open_port(${toString ircPort})
|
||
|
|
|
||
|
|
# run clientScript for all clients so that every list
|
||
|
|
# entry is executed by every client before advancing
|
||
|
|
# to the next one.
|
||
|
|
'' + lib.concatStrings
|
||
|
|
(reduce
|
||
|
|
(lib.zipListsWith (cs: c: cs + c))
|
||
|
|
(builtins.map clientScript clients));
|
||
|
|
})
|