pyrox.dev
1import ./make-test-python.nix ({ pkgs, lib, ... }:
2let
3 inherit (import ./ssh-keys.nix pkgs) snakeOilPrivateKey snakeOilPublicKey;
4 sshPort = 8231;
5 httpPort = 8232;
6 statsPort = 8233;
7 gitPort = 8418;
8in
9{
10 name = "soft-serve";
11 meta.maintainers = with lib.maintainers; [ dadada ];
12 nodes = {
13 client = { pkgs, ... }: {
14 environment.systemPackages = with pkgs; [
15 curl
16 git
17 openssh
18 ];
19 environment.etc.sshKey = {
20 source = snakeOilPrivateKey;
21 mode = "0600";
22 };
23 };
24
25 server =
26 { config, ... }:
27 {
28 services.soft-serve = {
29 enable = true;
30 settings = {
31 name = "TestServer";
32 ssh.listen_addr = ":${toString sshPort}";
33 git.listen_addr = ":${toString gitPort}";
34 http.listen_addr = ":${toString httpPort}";
35 stats.listen_addr = ":${toString statsPort}";
36 initial_admin_keys = [ snakeOilPublicKey ];
37 };
38 };
39 networking.firewall.allowedTCPPorts = [ sshPort httpPort statsPort ];
40 };
41 };
42
43 testScript =
44 { ... }:
45 ''
46 SSH_PORT = ${toString sshPort}
47 HTTP_PORT = ${toString httpPort}
48 STATS_PORT = ${toString statsPort}
49 KEY = "${snakeOilPublicKey}"
50 SSH_KEY = "/etc/sshKey"
51 SSH_COMMAND = f"ssh -p {SSH_PORT} -i {SSH_KEY} -o StrictHostKeyChecking=no"
52 TEST_DIR = "/tmp/test"
53 GIT = f"git -C {TEST_DIR}"
54
55 for machine in client, server:
56 machine.wait_for_unit("network.target")
57
58 server.wait_for_unit("soft-serve.service")
59 server.wait_for_open_port(SSH_PORT)
60
61 with subtest("Get info"):
62 status, test = client.execute(f"{SSH_COMMAND} server info")
63 if status != 0:
64 raise Exception("Failed to get SSH info")
65 key = " ".join(KEY.split(" ")[0:2])
66 if not key in test:
67 raise Exception("Admin key must be configured correctly")
68
69 with subtest("Create user"):
70 client.succeed(f"{SSH_COMMAND} server user create beatrice")
71 client.succeed(f"{SSH_COMMAND} server user info beatrice")
72
73 with subtest("Create repo"):
74 client.succeed(f"git init {TEST_DIR}")
75 client.succeed(f"{GIT} config --global user.email you@example.com")
76 client.succeed(f"touch {TEST_DIR}/foo")
77 client.succeed(f"{GIT} add foo")
78 client.succeed(f"{GIT} commit --allow-empty -m test")
79 client.succeed(f"{GIT} remote add origin git@server:test")
80 client.succeed(f"GIT_SSH_COMMAND='{SSH_COMMAND}' {GIT} push -u origin master")
81 client.execute("rm -r /tmp/test")
82
83 server.wait_for_open_port(HTTP_PORT)
84
85 with subtest("Clone over HTTP"):
86 client.succeed(f"curl --connect-timeout 10 http://server:{HTTP_PORT}/")
87 client.succeed(f"git clone http://server:{HTTP_PORT}/test /tmp/test")
88 client.execute("rm -r /tmp/test")
89
90 with subtest("Clone over SSH"):
91 client.succeed(f"GIT_SSH_COMMAND='{SSH_COMMAND}' git clone git@server:test /tmp/test")
92 client.execute("rm -r /tmp/test")
93
94 with subtest("Get stats over HTTP"):
95 server.wait_for_open_port(STATS_PORT)
96 status, test = client.execute(f"curl --connect-timeout 10 http://server:{STATS_PORT}/metrics")
97 if status != 0:
98 raise Exception("Failed to get metrics from status port")
99 if not "go_gc_duration_seconds_count" in test:
100 raise Exception("Metrics did not contain key 'go_gc_duration_seconds_count'")
101 '';
102})