depot/third_party/nixpkgs/nixos/tests/db-rest.nix

import ./make-test-python.nix ({ pkgs, ... }:
{
  name = "db-rest";
  meta.maintainers = with pkgs.lib.maintainers; [ marie ];

  nodes = {
    database = {
      networking = {
        interfaces.eth1 = {
          ipv4.addresses = [
            { address = "192.168.2.10"; prefixLength = 24; }
          ];
        };
        firewall.allowedTCPPorts = [ 31638 ];
      };

      services.redis.servers.db-rest = {
        enable = true;
        bind = "0.0.0.0";
        requirePass = "choochoo";
        port = 31638;
      };
    };

    serverWithTcp = { pkgs, ... }: {
      environment = {
        etc = {
          "db-rest/password-redis-db".text = ''
            choochoo
          '';
        };
      };

      networking = {
        interfaces.eth1 = {
          ipv4.addresses = [
            { address = "192.168.2.11"; prefixLength = 24; }
          ];
        };
        firewall.allowedTCPPorts = [ 3000 ];
      };

      services.db-rest = {
        enable = true;
        host = "0.0.0.0";
        redis = {
          enable = true;
          createLocally = false;
          host = "192.168.2.10";
          port = 31638;
          passwordFile = "/etc/db-rest/password-redis-db";
          useSSL = false;
        };
      };
    };

    serverWithUnixSocket = { pkgs, ... }: {
      networking = {
        interfaces.eth1 = {
          ipv4.addresses = [
            { address = "192.168.2.12"; prefixLength = 24; }
          ];
        };
        firewall.allowedTCPPorts = [ 3000 ];
      };

      services.db-rest = {
        enable = true;
        host = "0.0.0.0";
        redis = {
          enable = true;
          createLocally = true;
        };
      };
    };

    client = {
      environment.systemPackages = [ pkgs.jq ];
      networking = {
        interfaces.eth1 = {
          ipv4.addresses = [
            { address = "192.168.2.13"; prefixLength = 24; }
          ];
        };
      };
    };
  };

  testScript = ''
    start_all()

    with subtest("db-rest redis with TCP socket"):
      database.wait_for_unit("redis-db-rest.service")
      database.wait_for_open_port(31638)

      serverWithTcp.wait_for_unit("db-rest.service")
      serverWithTcp.wait_for_open_port(3000)

      client.succeed("curl --fail --get http://192.168.2.11:3000/stations --data-urlencode 'query=Köln Hbf' | jq -r '.\"8000207\".name' | grep 'Köln Hbf'")

    with subtest("db-rest redis with Unix socket"):
      serverWithUnixSocket.wait_for_unit("db-rest.service")
      serverWithUnixSocket.wait_for_open_port(3000)

      client.succeed("curl --fail --get http://192.168.2.12:3000/stations --data-urlencode 'query=Köln Hbf' | jq -r '.\"8000207\".name' | grep 'Köln Hbf'")
  '';
})