nixos/tests/magic-wormhole-mailbox-server.nix


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

import ./make-test-python.nix ({ pkgs, ... }: {
  name = "magic-wormhole-mailbox-server";
  meta = with pkgs.stdenv.lib.maintainers; {
    maintainers = [ mmahut ];
  };

  nodes = {
    server = { ... }: {
      networking.firewall.allowedTCPPorts = [ 4000 ];
      services.magic-wormhole-mailbox-server.enable = true;
    };

    client_alice = { ... }: {
      networking.firewall.enable = false;
      environment.systemPackages = [ pkgs.magic-wormhole ];
    };

    client_bob = { ... }: {
      environment.systemPackages = [ pkgs.magic-wormhole ];
    };
  };

  testScript = ''
    start_all()

    # Start the wormhole relay server
    server.wait_for_unit("magic-wormhole-mailbox-server.service")
    server.wait_for_open_port(4000)

    # Create a secret file and send it to Bob
    client_alice.succeed("echo mysecret > secretfile")
    client_alice.succeed("wormhole --relay-url=ws://server:4000/v1 send -0 secretfile &")

    # Retrieve a secret file from Alice and check its content
    client_bob.succeed("wormhole --relay-url=ws://server:4000/v1 receive -0 --accept-file")
    client_bob.succeed("grep mysecret secretfile")
  '';
})