storage/tests: add tests for splitfdstream

Signed-off-by: Giuseppe Scrivano <gscrivan@redhat.com>

Author: giuseppe

storage/tests/splitfdstream.bats

@@ -0,0 +1,107 @@
+#!/usr/bin/env bats
+
+load helpers
+
+SPLITFDSTREAM_SERVER_PID=
+
+splitfdstream_teardown() {
+	if [[ -n "$SPLITFDSTREAM_SERVER_PID" ]]; then
+		kill $SPLITFDSTREAM_SERVER_PID 2>/dev/null || true
+		wait $SPLITFDSTREAM_SERVER_PID 2>/dev/null || true
+		SPLITFDSTREAM_SERVER_PID=
+	fi
+}
+
+@test "splitfdstream json-rpc-server and apply-splitfdstream" {
+	case "$STORAGE_DRIVER" in
+	overlay*)
+		;;
+	*)
+		skip "driver $STORAGE_DRIVER does not support splitfdstream"
+		;;
+	esac
+
+	# Create and populate a test layer
+	populate
+
+	# Get the socket path from runroot
+	local runroot=`storage status 2>&1 | awk '/^Run Root:/{print $3}'`
+	local socket_path="$runroot/json-rpc.sock"
+
+	# Start the JSON-RPC server in the background
+	storage json-rpc-server --socket "$socket_path" &
+	SPLITFDSTREAM_SERVER_PID=$!
+
+	# Wait for socket to be created (max 10 seconds)
+	local count=0
+	while [[ ! -S "$socket_path" && $count -lt 50 ]]; do
+		sleep 0.2
+		count=$((count + 1))
+	done
+
+	# Check that the socket exists
+	[ -S "$socket_path" ]
+
+	# Create a new layer using apply-splitfdstream
+	# This should connect to our JSON-RPC server and fetch the layer
+	run storage --debug=false apply-splitfdstream --socket "$socket_path" "$lowerlayer"
+	echo "apply-splitfdstream output: $output"
+	[ "$status" -eq 0 ]
+	[ "$output" != "" ]
+
+	applied_layer="$output"
+
+	# Verify the layer was created
+	run storage --debug=false layers
+	[ "$status" -eq 0 ]
+	[[ "$output" =~ "$applied_layer" ]]
+
+	# Check that we can mount the applied layer
+	run storage --debug=false mount "$applied_layer"
+	[ "$status" -eq 0 ]
+	[ "$output" != "" ]
+	local applied_mount="$output"
+
+	# Verify some expected content exists (from populate function)
+	[ -f "$applied_mount/layer1file1" ]
+	[ -f "$applied_mount/layer1file2" ]
+	[ -d "$applied_mount/layerdir1" ]
+
+	# Unmount the layer
+	run storage unmount "$applied_layer"
+	[ "$status" -eq 0 ]
+
+	# Clean up
+	splitfdstream_teardown
+}
+
+@test "splitfdstream server socket path uses runroot" {
+	case "$STORAGE_DRIVER" in
+	overlay*)
+		;;
+	*)
+		skip "driver $STORAGE_DRIVER does not support splitfdstream"
+		;;
+	esac
+
+	# Get the expected socket path from runroot
+	local runroot=`storage status 2>&1 | awk '/^Run Root:/{print $3}'`
+	local expected_socket="$runroot/json-rpc.sock"
+
+	# Start the JSON-RPC server in the background
+	storage json-rpc-server &
+	SPLITFDSTREAM_SERVER_PID=$!
+
+	# Wait for socket to be created (max 10 seconds)
+	local count=0
+	while [[ ! -S "$expected_socket" && $count -lt 50 ]]; do
+		sleep 0.2
+		count=$((count + 1))
+	done
+
+	# Verify the socket is created in the correct location
+	[ -S "$expected_socket" ]
+
+	# Clean up
+	splitfdstream_teardown
+}