summaryrefslogtreecommitdiffstats
path: root/nixos/tests/pinnwand.nix
blob: f14a22f76686a4bf168d0e72afec0c72750fbe90 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
{
  pkgs,
  ...
}:
let
  port = 8000;
  baseUrl = "http://server:${toString port}";
in
{
  name = "pinnwand";
  meta = with pkgs.lib.maintainers; {
    maintainers = [ hexa ];
  };

  containers = {
    server =
      { config, ... }:
      {
        networking.firewall.allowedTCPPorts = [
          port
        ];

        services.pinnwand = {
          enable = true;
          port = port;
        };
      };

    client =
      { pkgs, ... }:
      {
        environment.systemPackages = [
          pkgs.steck

          (pkgs.writers.writePython3Bin "setup-steck.py"
            {
              libraries = with pkgs.python3.pkgs; [
                appdirs
                toml
              ];
              flakeIgnore = [
                "E501"
              ];
            }
            ''
              import appdirs
              import toml
              import os

              CONFIG = {
                  "base": "${baseUrl}/",
                  "confirm": False,
                  "magic": True,
                  "ignore": True
              }

              os.makedirs(appdirs.user_config_dir('steck'))
              with open(os.path.join(appdirs.user_config_dir('steck'), 'steck.toml'), "w") as fd:
                  toml.dump(CONFIG, fd)
            ''
          )
        ];
      };
  };

  testScript = ''
    start_all()

    server.wait_for_unit("pinnwand.service")
    client.wait_for_unit("network.target")

    # create steck.toml config file
    client.succeed("setup-steck.py")

    # wait until the server running pinnwand is reachable
    client.wait_until_succeeds("ping -c1 server")

    # make sure pinnwand is listening
    server.wait_for_open_port(${toString port})

    # send the contents of /etc/machine-id
    response = client.succeed("steck paste /etc/machine-id")

    # parse the steck response
    raw_url = None
    removal_link = None
    for line in response.split("\n"):
        if line.startswith("View link:"):
            raw_url = f"${baseUrl}/raw/{line.split('/')[-1]}"
        if line.startswith("Removal link:"):
            removal_link = line.split(":", 1)[1]

    # check whether paste matches what we sent
    client.succeed(f"curl {raw_url} > /tmp/machine-id")
    client.succeed("diff /tmp/machine-id /etc/machine-id")

    # remove paste and check that it's not available any more
    client.succeed(f"curl {removal_link}")
    client.fail(f"curl --fail {raw_url}")

    server.log(server.execute("systemd-analyze security pinnwand | grep '✗'")[1])
  '';
}