ark/tests/default.nix

{ self, pkgs }:
let
  inherit (self.inputs) nixos;
  inherit (self.nixosConfigurations.NixOS.config.lib) testModule specialArgs;

  # current release 20.09 does not support the specialArgs required for us
  # to use tests as we would normally use hosts. Using the "testing-python.nix"
  # from the override flake would build the test-vm from an unstable os
  # different than the one our systems are running. Instead simply patch nixpkgs
  # to include the updated version. This can be removed in the next release
  patchedNixpkgs =
    pkgs.stdenv.mkDerivation {
      name = "nixpkgs-patched";

      src = nixos;
      patches = [ ./0004-nixos-testing-Add-support-for-specialArgs.patch ];

      dontBuild = true;
      dontFixup = true;

      versionSuffix = "pre${
    if nixos ? lastModified
    then builtins.substring 0 8 (nixos.lastModifiedDate or nixos.lastModified)
    else toString nixos.revCount}.${nixos.shortRev or "dirty"}";

      configurePhase = ''
        echo -n $VERSION_SUFFIX > .version-suffix
        echo -n ${nixos.rev or nixos.shortRev or "dirty"} > .git-revision
      '';

      installPhase = ''
        cp -r $PWD $out
      '';
    };

  mkTest =
    let
      nixosTesting =
        (import "${patchedNixpkgs}/nixos/lib/testing-python.nix" {
          inherit (pkgs.stdenv.hostPlatform) system;
          inherit specialArgs;
          inherit pkgs;
          extraConfigurations = [
            testModule
          ];
        });
    in
    test:
    let
      loadedTest =
        if builtins.typeOf test == "path"
        then import test
        else test;
      calledTest =
        if pkgs.lib.isFunction loadedTest
        then pkgs.callPackage loadedTest { }
        else loadedTest;
    in
    nixosTesting.makeTest calledTest;
in
{
  profilesTest = mkTest {
    name = "profiles";

    machine = { suites, ... }: {
      imports = suites.allProfiles ++ suites.allUsers;
    };

    testScript = ''
      machine.systemctl("is-system-running --wait")
    '';
  };

  libTests = pkgs.runCommandNoCC "devos-lib-tests"
    {
      buildInputs = [
        pkgs.nix
        (
          let tests = import ./lib.nix { inherit self pkgs; };
          in
          if tests == [ ]
          then null
          else throw (builtins.toJSON tests)
        )
      ];
    } ''
    datadir="${pkgs.nix}/share"
    export TEST_ROOT=$(pwd)/test-tmp
    export NIX_BUILD_HOOK=
    export NIX_CONF_DIR=$TEST_ROOT/etc
    export NIX_LOCALSTATE_DIR=$TEST_ROOT/var
    export NIX_LOG_DIR=$TEST_ROOT/var/log/nix
    export NIX_STATE_DIR=$TEST_ROOT/var/nix
    export NIX_STORE_DIR=$TEST_ROOT/store
    export PAGER=cat
    cacheDir=$TEST_ROOT/binary-cache
    nix-store --init

    touch $out
  '';
}