Add tests for device io limits

buildkite-plugins · Feb 5, 2024 · 245ea6b · 245ea6b
1 parent 2ab416e
commit 245ea6b
Showing 1 changed file with 68 additions and 0 deletions.
diff --git a/tests/command.bats b/tests/command.bats
@@ -1028,6 +1028,74 @@ EOF
   unstub docker
 }
 
+@test "Runs BUILDKITE_COMMAND with multiple added device read bps" {
+  export BUILDKITE_COMMAND="echo hello world"
+  export BUILDKITE_PLUGIN_DOCKER_DEVICE_READ_BPS_0='bps-0'
+  export BUILDKITE_PLUGIN_DOCKER_DEVICE_READ_BPS_1='bps-1'
+  export BUILDKITE_PLUGIN_DOCKER_DEVICE_READ_BPS_2='bps-2'
+
+  stub docker \
+    "run -t -i --rm --init --volume $PWD:/workdir --device-read-bps bps-0 --device-read-bps bps-1 --device-read-bps bps-2 --workdir /workdir --label com.buildkite.job-id=1-2-3-4 image:tag /bin/sh -e -c 'echo hello world' : echo ran command in docker"
+
+  run "$PWD"/hooks/command
+
+  assert_success
+  assert_output --partial "ran command in docker"
+
+  unstub docker
+}
+
+@test "Runs BUILDKITE_COMMAND with multiple added device write bps" {
+  export BUILDKITE_COMMAND="echo hello world"
+  export BUILDKITE_PLUGIN_DOCKER_DEVICE_WRITE_BPS_0='bps-0'
+  export BUILDKITE_PLUGIN_DOCKER_DEVICE_WRITE_BPS_1='bps-1'
+  export BUILDKITE_PLUGIN_DOCKER_DEVICE_WRITE_BPS_2='bps-2'
+
+  stub docker \
+    "run -t -i --rm --init --volume $PWD:/workdir --device-write-bps bps-0 --device-write-bps bps-1 --device-write-bps bps-2 --workdir /workdir --label com.buildkite.job-id=1-2-3-4 image:tag /bin/sh -e -c 'echo hello world' : echo ran command in docker"
+
+  run "$PWD"/hooks/command
+
+  assert_success
+  assert_output --partial "ran command in docker"
+
+  unstub docker
+}
+
+@test "Runs BUILDKITE_COMMAND with multiple added device read iops" {
+  export BUILDKITE_COMMAND="echo hello world"
+  export BUILDKITE_PLUGIN_DOCKER_DEVICE_READ_IOPS_0='iops-0'
+  export BUILDKITE_PLUGIN_DOCKER_DEVICE_READ_IOPS_1='iops-1'
+  export BUILDKITE_PLUGIN_DOCKER_DEVICE_READ_IOPS_2='iops-2'
+
+  stub docker \
+    "run -t -i --rm --init --volume $PWD:/workdir --device-read-bps iops-0 --device-read-bps iops-1 --device-read-bps iops-2 --workdir /workdir --label com.buildkite.job-id=1-2-3-4 image:tag /bin/sh -e -c 'echo hello world' : echo ran command in docker"
+
+  run "$PWD"/hooks/command
+
+  assert_success
+  assert_output --partial "ran command in docker"
+
+  unstub docker
+}
+
+@test "Runs BUILDKITE_COMMAND with multiple added device write iops" {
+  export BUILDKITE_COMMAND="echo hello world"
+  export BUILDKITE_PLUGIN_DOCKER_DEVICE_WRITE_IOPS_0='iops-0'
+  export BUILDKITE_PLUGIN_DOCKER_DEVICE_WRITE_IOPS_1='iops-1'
+  export BUILDKITE_PLUGIN_DOCKER_DEVICE_WRITE_IOPS_2='iops-2'
+
+  stub docker \
+    "run -t -i --rm --init --volume $PWD:/workdir --device-write-bps iops-0 --device-write-bps iops-1 --device-write-bps iops-2 --workdir /workdir --label com.buildkite.job-id=1-2-3-4 image:tag /bin/sh -e -c 'echo hello world' : echo ran command in docker"
+
+  run "$PWD"/hooks/command
+
+  assert_success
+  assert_output --partial "ran command in docker"
+
+  unstub docker
+}
+
 @test "Runs BUILDKITE_COMMAND with one added capability" {
   export BUILDKITE_COMMAND="echo hello world"
   export BUILDKITE_PLUGIN_DOCKER_ADD_CAPS_0='cap-0'