[v3,blktests,1/5] tests/throtl: add first test for blk-throttle

Commit Message

Yu Kuai April 20, 2024, 8:45 a.m. UTC

From: Yu Kuai <[e-mail]>

Test basic functionality.

Signed-off-by: Yu Kuai <[e-mail]>
---
 tests/throtl/001     | 39 ++++++++++++++++++++
 tests/throtl/001.out |  6 +++
 tests/throtl/rc      | 88 ++++++++++++++++++++++++++++++++++++++++++++
 3 files changed, 133 insertions(+)
 create mode 100755 tests/throtl/001
 create mode 100644 tests/throtl/001.out
 create mode 100644 tests/throtl/rc

Patch

diff --git a/tests/throtl/001 b/tests/throtl/001
new file mode 100755
index 0000000..835cac2
--- /dev/null
+++ b/tests/throtl/001
@@ -0,0 +1,39 @@ 
+#!/bin/bash
+# SPDX-License-Identifier: GPL-3.0+
+# Copyright (C) 2024 Yu Kuai
+#
+# Test basic functionality of blk-throttle
+
+. tests/throtl/rc
+
+DESCRIPTION="basic functionality"
+QUICK=1
+
+test() {
+	echo "Running ${TEST_NAME}"
+
+	if ! _set_up_throtl; then
+		return 1;
+	fi
+
+	local bps_limit=$((1024 * 1024))
+
+	_throtl_set_limits wbps=$bps_limit
+	_throtl_test_io write 4k 256
+	_throtl_remove_limits
+
+	_throtl_set_limits wiops=256
+	_throtl_test_io write 4k 256
+	_throtl_remove_limits
+
+	_throtl_set_limits rbps=$bps_limit
+	_throtl_test_io read 4k 256
+	_throtl_remove_limits
+
+	_throtl_set_limits riops=256
+	_throtl_test_io read 4k 256
+	_throtl_remove_limits
+
+	_clean_up_throtl
+	echo "Test complete"
+}
diff --git a/tests/throtl/001.out b/tests/throtl/001.out
new file mode 100644
index 0000000..a3edfdd
--- /dev/null
+++ b/tests/throtl/001.out
@@ -0,0 +1,6 @@ 
+Running throtl/001
+1
+1
+1
+1
+Test complete
diff --git a/tests/throtl/rc b/tests/throtl/rc
new file mode 100644
index 0000000..2ddbac8
--- /dev/null
+++ b/tests/throtl/rc
@@ -0,0 +1,88 @@ 
+#!/bin/bash
+# SPDX-License-Identifier: GPL-3.0+
+# Copyright (C) 2024 Yu Kuai
+#
+# Tests for blk-throttle
+
+. common/rc
+. common/null_blk
+. common/cgroup
+
+THROTL_DIR=$(echo "$TEST_NAME" | tr '/' '_')
+THROTL_DEV=dev_nullb
+
+group_requires() {
+	_have_root
+	_have_null_blk
+	_have_kernel_option BLK_DEV_THROTTLING
+	_have_cgroup2_controller io
+	_have_program bc
+}
+
+# Create a new null_blk device, and create a new blk-cgroup for test.
+_set_up_throtl() {
+
+	if ! _configure_null_blk $THROTL_DEV "$@" power=1; then
+		return 1
+	fi
+
+	if ! _init_cgroup2; then
+		_exit_null_blk
+		return 1
+	fi
+
+	echo "+io" > "$(_cgroup2_base_dir)/cgroup.subtree_control"
+	echo "+io" > "$CGROUP2_DIR/cgroup.subtree_control"
+
+	mkdir -p "$CGROUP2_DIR/$THROTL_DIR"
+	return 0;
+}
+
+_clean_up_throtl() {
+	rmdir "$CGROUP2_DIR/$THROTL_DIR"
+	echo "-io" > "$CGROUP2_DIR/cgroup.subtree_control"
+	echo "-io" > "$(_cgroup2_base_dir)/cgroup.subtree_control"
+
+	_exit_cgroup2
+	_exit_null_blk
+}
+
+_throtl_set_limits() {
+	echo "$(cat /sys/block/$THROTL_DEV/dev) $*" > \
+		"$CGROUP2_DIR/$THROTL_DIR/io.max"
+}
+
+_throtl_remove_limits() {
+	echo "$(cat /sys/block/$THROTL_DEV/dev) rbps=max wbps=max riops=max wiops=max" > \
+		"$CGROUP2_DIR/$THROTL_DIR/io.max"
+}
+
+# Create an asynchronous thread and bind it to the specified blk-cgroup, issue
+# IO and then print time elapsed to the second, blk-throttle limits should be
+# set before this function.
+_throtl_test_io() {
+	local pid
+
+	{
+		local start_time
+		local end_time
+		local elapsed
+
+		sleep 0.1
+		start_time=$(date +%s.%N)
+
+		if [ "$1" == "read" ]; then
+			dd if=/dev/$THROTL_DEV of=/dev/null bs="$2" count="$3" iflag=direct status=none
+		elif [ "$1" == "write" ]; then
+			dd of=/dev/$THROTL_DEV if=/dev/zero bs="$2" count="$3" oflag=direct status=none
+		fi
+
+		end_time=$(date +%s.%N)
+		elapsed=$(echo "$end_time - $start_time" | bc)
+		printf "%.0f\n" "$elapsed"
+	} &
+
+	pid=$!
+	echo "$pid" > "$CGROUP2_DIR/$THROTL_DIR/cgroup.procs"
+	wait $pid
+}