depot/third_party/nixpkgs/nixos/tests/signal-desktop.nix

import ./make-test-python.nix ({ pkgs, ...} :

let
  sqlcipher-signal = pkgs.writeShellScriptBin "sqlcipher" ''
    set -eu

    readonly CFG=~/.config/Signal/config.json
    readonly KEY="$(${pkgs.jq}/bin/jq --raw-output '.key' $CFG)"
    readonly DB="$1"
    readonly SQL="SELECT * FROM sqlite_master where type='table'"
    ${pkgs.sqlcipher}/bin/sqlcipher "$DB" "PRAGMA key = \"x'$KEY'\"; $SQL"
  '';
in {
  name = "signal-desktop";
  meta = with pkgs.lib.maintainers; {
    maintainers = [ flokli primeos ];
  };

  nodes.machine = { ... }:

  {
    imports = [
      ./common/user-account.nix
      ./common/x11.nix
    ];

    services.xserver.enable = true;
    test-support.displayManager.auto.user = "alice";
    environment.systemPackages = with pkgs; [
      signal-desktop file sqlite sqlcipher-signal
    ];
  };

  enableOCR = true;

  testScript = { nodes, ... }: let
    user = nodes.machine.config.users.users.alice;
  in ''
    start_all()
    machine.wait_for_x()

    # start signal desktop
    machine.execute("su - alice -c signal-desktop >&2 &")

    # Wait for the Signal window to appear. Since usually the tests
    # are run sandboxed and therfore with no internet, we can not wait
    # for the message "Link your phone ...". Nor should we wait for
    # the "Failed to connect to server" message, because when manually
    # running this test it will be not sandboxed.
    machine.wait_for_text("Signal")
    machine.wait_for_text("File Edit View Window Help")
    machine.screenshot("signal_desktop")

    # Test if the database is encrypted to prevent these issues:
    # - https://github.com/NixOS/nixpkgs/issues/108772
    # - https://github.com/NixOS/nixpkgs/pull/117555
    print(machine.succeed("su - alice -c 'file ~/.config/Signal/sql/db.sqlite'"))
    machine.fail(
        "su - alice -c 'file ~/.config/Signal/sql/db.sqlite' | grep -e SQLite -e database"
    )
    # Only SQLCipher should be able to read the encrypted DB:
    machine.fail(
        "su - alice -c 'sqlite3 ~/.config/Signal/sql/db.sqlite .tables'"
    )
    print(machine.succeed(
        "su - alice -c 'sqlcipher ~/.config/Signal/sql/db.sqlite'"
    ))
  '';
})