You can now run a test in the nixos/tests directory directly using nix-build, e.g. $ nix-build '<nixos/tests/login.nix>' -A test This gets rid of having to add the test to nixos/tests/default.nix. (Of course, you still need to add it to nixos/release.nix if you want Hydra to run the test.)
		
			
				
	
	
		
			48 lines
		
	
	
		
			1.3 KiB
		
	
	
	
		
			Nix
		
	
	
	
	
	
			
		
		
	
	
			48 lines
		
	
	
		
			1.3 KiB
		
	
	
	
		
			Nix
		
	
	
	
	
	
# Test the firewall module.
 | 
						|
 | 
						|
import ./make-test.nix {
 | 
						|
 | 
						|
  nodes =
 | 
						|
    { walled =
 | 
						|
        { config, pkgs, nodes, ... }:
 | 
						|
        { networking.firewall.enable = true;
 | 
						|
          networking.firewall.logRefusedPackets = true;
 | 
						|
          services.httpd.enable = true;
 | 
						|
          services.httpd.adminAddr = "foo@example.org";
 | 
						|
        };
 | 
						|
 | 
						|
      attacker =
 | 
						|
        { config, pkgs, ... }:
 | 
						|
        { services.httpd.enable = true;
 | 
						|
          services.httpd.adminAddr = "foo@example.org";
 | 
						|
          networking.firewall.enable = false;
 | 
						|
        };
 | 
						|
    };
 | 
						|
 | 
						|
  testScript =
 | 
						|
    { nodes, ... }:
 | 
						|
    ''
 | 
						|
      startAll;
 | 
						|
 | 
						|
      $walled->waitForUnit("firewall");
 | 
						|
      $walled->waitForUnit("httpd");
 | 
						|
      $attacker->waitForUnit("network.target");
 | 
						|
 | 
						|
      # Local connections should still work.
 | 
						|
      $walled->succeed("curl -v http://localhost/ >&2");
 | 
						|
 | 
						|
      # Connections to the firewalled machine should fail.
 | 
						|
      $attacker->fail("curl --fail --connect-timeout 2 http://walled/ >&2");
 | 
						|
      $attacker->fail("ping -c 1 walled >&2");
 | 
						|
 | 
						|
      # Outgoing connections/pings should still work.
 | 
						|
      $walled->succeed("curl -v http://attacker/ >&2");
 | 
						|
      $walled->succeed("ping -c 1 attacker >&2");
 | 
						|
 | 
						|
      # If we stop the firewall, then connections should succeed.
 | 
						|
      $walled->stopJob("firewall");
 | 
						|
      $attacker->succeed("curl -v http://walled/ >&2");
 | 
						|
    '';
 | 
						|
 | 
						|
}
 |