nixos/tests/matrix/dendrite.nix

   1 import ../make-test-python.nix (
   2   { pkgs, ... }:
   3     let
   4       homeserverUrl = "http://homeserver:8008";
   5
   6       private_key = pkgs.runCommand "matrix_key.pem" {
   7         buildInputs = [ pkgs.dendrite ];
   8       } "generate-keys --private-key $out";
   9     in
  10       {
  11         name = "dendrite";
  12         meta = with pkgs.lib; {
  13           maintainers = teams.matrix.members;
  14         };
  15
  16         nodes = {
  17           homeserver = { pkgs, ... }: {
  18             services.dendrite = {
  19               enable = true;
  20               loadCredential = [ "test_private_key:${private_key}" ];
  21               openRegistration = true;
  22               settings = {
  23                 global.server_name = "test-dendrite-server.com";
  24                 global.private_key = "$CREDENTIALS_DIRECTORY/test_private_key";
  25                 client_api.registration_disabled = false;
  26               };
  27             };
  28
  29             networking.firewall.allowedTCPPorts = [ 8008 ];
  30           };
  31
  32           client = { pkgs, ... }: {
  33             environment.systemPackages = [
  34               (
  35                 pkgs.writers.writePython3Bin "do_test"
  36                   { libraries = [ pkgs.python3Packages.matrix-nio ]; } ''
  37                   import asyncio
  38
  39                   from nio import AsyncClient
  40
  41
  42                   async def main() -> None:
  43                       # Connect to dendrite
  44                       client = AsyncClient("http://homeserver:8008", "alice")
  45
  46                       # Register as user alice
  47                       response = await client.register("alice", "my-secret-password")
  48
  49                       # Log in as user alice
  50                       response = await client.login("my-secret-password")
  51
  52                       # Create a new room
  53                       response = await client.room_create(federate=False)
  54                       room_id = response.room_id
  55
  56                       # Join the room
  57                       response = await client.join(room_id)
  58
  59                       # Send a message to the room
  60                       response = await client.room_send(
  61                           room_id=room_id,
  62                           message_type="m.room.message",
  63                           content={
  64                               "msgtype": "m.text",
  65                               "body": "Hello world!"
  66                           }
  67                       )
  68
  69                       # Sync responses
  70                       response = await client.sync(timeout=30000)
  71
  72                       # Check the message was received by dendrite
  73                       last_message = response.rooms.join[room_id].timeline.events[-1].body
  74                       assert last_message == "Hello world!"
  75
  76                       # Leave the room
  77                       response = await client.room_leave(room_id)
  78
  79                       # Close the client
  80                       await client.close()
  81
  82                   asyncio.get_event_loop().run_until_complete(main())
  83                 ''
  84               )
  85             ];
  86           };
  87         };
  88
  89         testScript = ''
  90           start_all()
  91
  92           with subtest("start the homeserver"):
  93               homeserver.wait_for_unit("dendrite.service")
  94               homeserver.wait_for_open_port(8008)
  95
  96           with subtest("ensure messages can be exchanged"):
  97               client.succeed("do_test")
  98         '';
  99
 100       }
 101 )