summaryrefslogtreecommitdiffstats
path: root/nixos/tests/nvmetcfg.nix
blob: 4720ac75437802ccb442d137b88b837c18966be0 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
{ lib, ... }:
{
  name = "nvmetcfg";

  meta = {
    maintainers = with lib.maintainers; [ nickcao ];
  };

  nodes = {
    server =
      { pkgs, ... }:
      {
        boot.kernelModules = [ "nvmet" ];
        environment.systemPackages = [ pkgs.nvmetcfg ];
        networking.firewall.allowedTCPPorts = [ 4420 ];
        virtualisation.emptyDiskImages = [ 512 ];
      };
    client =
      { pkgs, ... }:
      {
        boot.kernelModules = [ "nvme-fabrics" ];
        environment.systemPackages = [ pkgs.nvme-cli ];
      };
  };

  testScript =
    let
      subsystem = "nqn.2014-08.org.nixos:server";
    in
    ''
      import json

      with subtest("Create subsystem and namespace"):
        server.succeed("nvmet subsystem add ${subsystem}")
        server.succeed("nvmet namespace add ${subsystem} 1 /dev/vdb")

      with subtest("Bind subsystem to port"):
        server.wait_for_unit("network-online.target")
        server.succeed("nvmet port add 1 tcp [::]:4420")
        server.succeed("nvmet port add-subsystem 1 ${subsystem}")

      with subtest("Discover and connect to available subsystems"):
        client.wait_for_unit("network-online.target")
        assert "subnqn:  ${subsystem}" in client.succeed("nvme discover --transport=tcp --traddr=server --trsvcid=4420")
        client.succeed("nvme connect-all --transport=tcp --traddr=server --trsvcid=4420")

      with subtest("Write to the connected subsystem"):
        devices = json.loads(client.succeed("lsblk --nvme --paths --json"))["blockdevices"]
        assert len(devices) == 1
        client.succeed(f"dd if=/dev/zero of={devices[0]['name']} bs=1M count=64")
    '';
}