depot/third_party/nixpkgs/nixos/tests/matrix/conduit.nix

import ../make-test-python.nix (
  { pkgs, ... }:
  let
    name = "conduit";
  in
  {
    name = "matrix-conduit";

    nodes = {
      conduit = args: {
        services.matrix-conduit = {
          enable = true;
          settings.global.server_name = name;
          settings.global.allow_registration = true;
          extraEnvironment.RUST_BACKTRACE = "yes";
        };
        services.nginx = {
          enable = true;
          virtualHosts.${name} = {
            enableACME = false;
            forceSSL = false;
            enableSSL = false;

            locations."/_matrix" = {
              proxyPass = "http://[::1]:6167";
            };
          };
        };
        networking.firewall.allowedTCPPorts = [ 80 ];
      };
      client =
        { pkgs, ... }:
        {
          environment.systemPackages = [
            (pkgs.writers.writePython3Bin "do_test" { libraries = [ pkgs.python3Packages.matrix-nio ]; } ''
              import asyncio

              from nio import AsyncClient


              async def main() -> None:
                  # Connect to conduit
                  client = AsyncClient("http://conduit:80", "alice")

                  # Register as user alice
                  response = await client.register("alice", "my-secret-password")

                  # Log in as user alice
                  response = await client.login("my-secret-password")

                  # Create a new room
                  response = await client.room_create(federate=False)
                  room_id = response.room_id

                  # Join the room
                  response = await client.join(room_id)

                  # Send a message to the room
                  response = await client.room_send(
                      room_id=room_id,
                      message_type="m.room.message",
                      content={
                          "msgtype": "m.text",
                          "body": "Hello conduit!"
                      }
                  )

                  # Sync responses
                  response = await client.sync(timeout=30000)

                  # Check the message was received by conduit
                  last_message = response.rooms.join[room_id].timeline.events[-1].body
                  assert last_message == "Hello conduit!"

                  # Leave the room
                  response = await client.room_leave(room_id)

                  # Close the client
                  await client.close()

              asyncio.get_event_loop().run_until_complete(main())
            '')
          ];
        };
    };

    testScript = ''
      start_all()

      with subtest("start conduit"):
            conduit.wait_for_unit("conduit.service")
            conduit.wait_for_open_port(80)

      with subtest("ensure messages can be exchanged"):
            client.succeed("do_test")
    '';
  }
)