nixpkgs/nixos/tests/avahi.nix

{
  pkgs,
  # bool: whether to use networkd in the tests
  networkd ? false,
  ...
}:

# Test whether `avahi-daemon' and `libnss-mdns' work as expected.
{
  name = "avahi";
  meta.maintainers = [ ];

  nodes =
    let
      cfg =
        { ... }:
        {
          services.avahi = {
            enable = true;
            nssmdns4 = true;
            publish.addresses = true;
            publish.domain = true;
            publish.enable = true;
            publish.userServices = true;
            publish.workstation = true;
            extraServiceFiles.ssh = "${pkgs.avahi}/etc/avahi/services/ssh.service";
          };
        }
        // pkgs.lib.optionalAttrs networkd {
          networking = {
            useNetworkd = true;
            useDHCP = false;
          };
        };
    in
    {
      one = cfg;
      two = cfg;
    };

  testScript = ''
    start_all()

    # mDNS.
    one.wait_for_unit("network.target")
    two.wait_for_unit("network.target")

    one.succeed("avahi-resolve-host-name one.local | tee out >&2")
    one.succeed('test "`cut -f1 < out`" = one.local')
    one.succeed("avahi-resolve-host-name two.local | tee out >&2")
    one.succeed('test "`cut -f1 < out`" = two.local')

    two.succeed("avahi-resolve-host-name one.local | tee out >&2")
    two.succeed('test "`cut -f1 < out`" = one.local')
    two.succeed("avahi-resolve-host-name two.local | tee out >&2")
    two.succeed('test "`cut -f1 < out`" = two.local')

    # Basic DNS-SD.
    one.succeed("avahi-browse -r -t _workstation._tcp | tee out >&2")
    one.succeed("test `wc -l < out` -gt 0")
    two.succeed("avahi-browse -r -t _workstation._tcp | tee out >&2")
    two.succeed("test `wc -l < out` -gt 0")

    # More DNS-SD.
    one.execute('avahi-publish -s "This is a test" _test._tcp 123 one=1 >&2 &')
    one.sleep(5)
    two.succeed("avahi-browse -r -t _test._tcp | tee out >&2")
    two.succeed("test `wc -l < out` -gt 0")

    # NSS-mDNS.
    one.succeed("getent hosts one.local >&2")
    one.succeed("getent hosts two.local >&2")
    two.succeed("getent hosts one.local >&2")
    two.succeed("getent hosts two.local >&2")

    # extra service definitions
    one.succeed("avahi-browse -r -t _ssh._tcp | tee out >&2")
    one.succeed("test `wc -l < out` -gt 0")
    two.succeed("avahi-browse -r -t _ssh._tcp | tee out >&2")
    two.succeed("test `wc -l < out` -gt 0")

    one.log(one.execute("systemd-analyze security avahi-daemon.service | grep -v ✓")[1])
  '';
}