32 lines
1,006 B
Nix
32 lines
1,006 B
Nix
{ system ? builtins.currentSystem
|
|
, config ? {}
|
|
, nixpkgs
|
|
, networkExpr
|
|
}:
|
|
|
|
let
|
|
nodes = builtins.mapAttrs (vm: module: {
|
|
_file = "${networkExpr}@node-${vm}";
|
|
imports = [ module ];
|
|
}) (import networkExpr);
|
|
|
|
pkgs = import nixpkgs { inherit system config; };
|
|
|
|
testing = import "${pkgs.path}/nixos/lib/testing-python.nix" {
|
|
inherit system pkgs;
|
|
};
|
|
|
|
interactiveDriver = (testing.makeTest { inherit nodes; name = "network"; testScript = "start_all(); join_all();"; }).test.driverInteractive;
|
|
in
|
|
|
|
|
|
pkgs.runCommand "nixos-build-vms" { nativeBuildInputs = [ pkgs.makeWrapper ]; } ''
|
|
mkdir -p $out/bin
|
|
ln -s ${interactiveDriver}/bin/nixos-test-driver $out/bin/nixos-test-driver
|
|
ln -s ${interactiveDriver}/bin/nixos-test-driver $out/bin/nixos-run-vms
|
|
wrapProgram $out/bin/nixos-test-driver \
|
|
--add-flags "--interactive"
|
|
wrapProgram $out/bin/nixos-run-vms \
|
|
--set testScript "${pkgs.writeText "start-all" "start_all(); join_all();"}" \
|
|
--add-flags "--no-interactive"
|
|
''
|