59 lines
1.5 KiB
Nix
59 lines
1.5 KiB
Nix
|
{
|
||
|
|
pkgs,
|
||
|
|
makeTest,
|
||
|
|
}:
|
||
|
|
|
||
|
|
let
|
||
|
|
inherit (pkgs) lib;
|
||
|
|
|
||
|
|
makeTestFor =
|
||
|
|
package:
|
||
|
|
makeTest {
|
||
|
|
name = "postgresql-jit-${package.name}";
|
||
|
|
meta.maintainers = with lib.maintainers; [ ma27 ];
|
||
|
|
|
||
|
|
nodes.machine =
|
||
|
|
{ pkgs, ... }:
|
||
|
|
{
|
||
|
|
services.postgresql = {
|
||
|
|
inherit package;
|
||
|
|
enable = true;
|
||
|
|
enableJIT = true;
|
||
|
|
initialScript = pkgs.writeText "init.sql" ''
|
||
|
|
create table demo (id int);
|
||
|
|
insert into demo (id) select generate_series(1, 5);
|
||
|
|
'';
|
||
|
|
};
|
||
|
|
};
|
||
|
|
|
||
|
|
testScript = ''
|
||
|
|
machine.start()
|
||
|
|
machine.wait_for_unit("postgresql.service")
|
||
|
|
|
||
|
|
with subtest("JIT is enabled"):
|
||
|
|
machine.succeed("sudo -u postgres psql <<<'show jit;' | grep 'on'")
|
||
|
|
|
||
|
|
with subtest("Test JIT works fine"):
|
||
|
|
output = machine.succeed(
|
||
|
|
"cat ${pkgs.writeText "test.sql" ''
|
||
|
|
set jit_above_cost = 1;
|
||
|
|
EXPLAIN ANALYZE SELECT CONCAT('jit result = ', SUM(id)) FROM demo;
|
||
|
|
SELECT CONCAT('jit result = ', SUM(id)) from demo;
|
||
|
|
''} | sudo -u postgres psql"
|
||
|
|
)
|
||
|
|
assert "JIT:" in output
|
||
|
|
assert "jit result = 15" in output
|
||
|
|
|
||
|
|
machine.shutdown()
|
||
|
|
'';
|
||
|
|
};
|
||
|
|
in
|
||
|
|
lib.recurseIntoAttrs (
|
||
|
|
lib.concatMapAttrs (n: p: { ${n} = makeTestFor p; }) (
|
||
|
|
lib.filterAttrs (n: _: lib.hasSuffix "_jit" n) pkgs.postgresqlVersions
|
||
|
|
)
|
||
|
|
// {
|
||
|
|
passthru.override = p: makeTestFor p;
|
||
|
|
}
|
||
|
|
)
|