import ./make-test-python.nix (
  { pkgs, lib, ... }:
  let
    drbdPort = 7789;

    drbdConfig =
      { nodes, ... }:
      {
        virtualisation.emptyDiskImages = [ 1 ];
        networking.firewall.allowedTCPPorts = [ drbdPort ];

        services.drbd = {
          enable = true;
          config = ''
            global {
              usage-count yes;
            }

            common {
              net {
                protocol C;
                ping-int 1;
              }
            }

            resource r0 {
              volume 0 {
                device    /dev/drbd0;
                disk      /dev/vdb;
                meta-disk internal;
              }

              on drbd1 {
                address ${nodes.drbd1.networking.primaryIPAddress}:${toString drbdPort};
              }

              on drbd2 {
                address ${nodes.drbd2.networking.primaryIPAddress}:${toString drbdPort};
              }
            }
          '';
        };
      };
  in
  {
    name = "drbd";
    meta = with pkgs.lib.maintainers; {
      maintainers = [
        ryantm
        astro
        birkb
      ];
    };

    nodes.drbd1 = drbdConfig;
    nodes.drbd2 = drbdConfig;

    testScript =
      { nodes }:
      ''
        drbd1.start()
        drbd2.start()

        drbd1.wait_for_unit("network.target")
        drbd2.wait_for_unit("network.target")

        drbd1.succeed(
            "drbdadm create-md r0",
            "drbdadm up r0",
            "drbdadm primary r0 --force",
        )

        drbd2.succeed("drbdadm create-md r0", "drbdadm up r0")

        drbd1.succeed(
            "mkfs.ext4 /dev/drbd0",
            "mkdir -p /mnt/drbd",
            "mount /dev/drbd0 /mnt/drbd",
            "touch /mnt/drbd/hello",
            "umount /mnt/drbd",
            "drbdadm secondary r0",
        )
        drbd1.sleep(1)

        drbd2.succeed(
            "drbdadm primary r0",
            "mkdir -p /mnt/drbd",
            "mount /dev/drbd0 /mnt/drbd",
            "ls /mnt/drbd/hello",
        )
      '';
  }
)