summaryrefslogtreecommitdiff
path: root/pkgs/servers/sql/postgresql/postgresqlTestExtension.nix
blob: 734e7b4e3f9024bc1573d7f61a6da102cd33c3cf (plain)
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
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
{
  lib,
  postgresql,
  postgresqlTestHook,
  stdenvNoCC,
}:

{
  asserts ? [ ],
  finalPackage,
  sql,
  withPackages ? [ ],
  ...
}@extraArgs:
stdenvNoCC.mkDerivation (
  {
    name = "${finalPackage.name}-test-extension";
    dontUnpack = true;
    doCheck = true;
    nativeCheckInputs = [
      postgresqlTestHook
      (postgresql.withPackages (ps: [ finalPackage ] ++ (map (p: ps."${p}") withPackages)))
    ];
    postgresqlTestUserOptions = "LOGIN SUPERUSER";
    sql =
      sql
      + lib.concatMapStrings (
        {
          query,
          expected,
          description,
        }:
        ''
          DO $$ BEGIN
            ASSERT (${query}) = (${expected}), '${lib.replaceStrings [ "'" ] [ "''" ] description}';
          END $$;
        ''
      ) asserts;
    checkPhase = ''
      runHook preCheck
      sqlPath=$TMPDIR/test.sql
      printf "%s" "$sql" > $sqlPath
      psql -a -v ON_ERROR_STOP=1 -f "$sqlPath"
      runHook postCheck
    '';
    installPhase = "touch $out";
    __structuredAttrs = true;
  }
  // lib.removeAttrs extraArgs [
    "asserts"
    "finalPackage"
    "sql"
    "withPackages"
  ]
)