nixtest/lib/default.nix

{
  pkgs,
  lib ? pkgs.lib,
  self ? "",
  ...
}: {
  mkTest = {
    type ? "unit",
    name,
    description ? "",
    format ? "json",
    expected ? null,
    actual ? null,
    actualDrv ? null,
    script ? null,
    pos ? null,
  }: let
    fileRelative = lib.removePrefix ((toString self) + "/") pos.file;
    actual' =
      if format == "json"
      then actual
      else lib.generators.toPretty {} actual;
    expected' =
      if format == "json"
      then expected
      else lib.generators.toPretty {} expected;
  in {
    inherit type name description;
    actual = actual';
    expected = expected';
    # discard string context, otherwise it's being built instantly which we don't want
    actualDrv = builtins.unsafeDiscardStringContext (actualDrv.drvPath or "");
    script =
      if script != null
      then builtins.unsafeDiscardStringContext (pkgs.writeShellScript "nixtest-${name}" script).drvPath
      else null;
    pos =
      if pos == null
      then ""
      else "${fileRelative}:${toString pos.line}";
  };
  mkSuite = name: tests: {
    inherit name tests;
  };
  exportSuites = suites: let
    suitesList =
      if builtins.isList suites
      then suites
      else [suites];
    testsMapped = builtins.toJSON suitesList;
  in
    pkgs.runCommand "tests.json" {} ''
      echo '${testsMapped}' > $out
    '';
}