import ../make-test-python.nix (
  { pkgs, ... }:
  {
    name = "nifi";
    meta.maintainers = with pkgs.lib.maintainers; [ izorkin ];

    nodes = {
      nifi =
        { pkgs, ... }:
        {
          virtualisation = {
            memorySize = 2048;
            diskSize = 4096;
          };
          services.nifi = {
            enable = true;
            enableHTTPS = false;
          };
        };
    };

    testScript = ''
      nifi.start()

      nifi.wait_for_unit("nifi.service")
      nifi.wait_for_open_port(8080)

      # Check if NiFi is running
      nifi.succeed("curl --fail http://127.0.0.1:8080/nifi/login 2> /dev/null | grep 'NiFi Login'")

      nifi.shutdown()
    '';
  }
)