Skip to content

Commit

Permalink
[guest-test] New Test: add 3 lazy accept memory TCs
Browse files Browse the repository at this point in the history 
add 3 lazy accept memory functional TCs base on guest-test framework

[Test Components] tdx
[Test Types] func
[Supported Devices] spr,emr,gnr,srf

Signed-off-by: Hongyu Ning <[email protected]>
  • Loading branch information
@hongyuni @ysun
hongyuni authored and ysun committed Dec 21, 2023
1 parent 1a806e7 commit f091173
Show file tree
Hide file tree
Showing 3 changed files with 111 additions and 27 deletions.
24 changes: 24 additions & 0 deletions guest-test/tdx/tdx.test_executor.sh
View file
Original file line number Diff line number Diff line change
Expand Up @@ -224,6 +224,30 @@ case "$TESTCASE" in
guest_test_close || { die "Failed on close guest VM"; return 1; }
fi
;;
TD_MEM_ACPT_FUNC)
guest_test_prepare tdx_mem_test.sh
guest_test_entry tdx_mem_test.sh "-t MEM_ACPT_FUNC" || \
{ die "Failed on $TESTCASE tdx_mem_test.sh -t MEM_ACPT_FUNC"; return 1; }
if [[ $GCOV == "off" ]]; then
guest_test_close || { die "Failed on close guest VM"; return 1; }
fi
;;
TD_MEM_ACPT_CAL)
guest_test_prepare tdx_mem_test.sh
guest_test_entry tdx_mem_test.sh "-t MEM_ACPT_CAL" || \
{ die "Failed on $TESTCASE tdx_mem_test.sh -t MEM_ACPT_CAL"; return 1; }
if [[ $GCOV == "off" ]]; then
guest_test_close || { die "Failed on close guest VM"; return 1; }
fi
;;
TD_MEM_ACPT_NEG)
guest_test_prepare tdx_mem_test.sh
guest_test_entry tdx_mem_test.sh "-t MEM_ACPT_NEG" || \
{ die "Failed on $TESTCASE tdx_mem_test.sh -t MEM_ACPT_NEG"; return 1; }
if [[ $GCOV == "off" ]]; then
guest_test_close || { die "Failed on close guest VM"; return 1; }
fi
;;
:)
test_print_err "Must specify the test scenario option by [-t]"
usage && exit 1
Expand Down
109 changes: 83 additions & 26 deletions guest-test/tdx/tdx_mem_test.sh
View file
Original file line number Diff line number Diff line change
Expand Up @@ -45,40 +45,88 @@ ebizzy_func() {

# function based on stress-ng calculate total remained mem accepted time
mem_accepted_time() {
# common expected time consumed in seconds
expected_time=$1
# stress-ng mem stress process number
workers=$2
# prepare for prerequisites
# common expected time consumed in seconds
expected_time=$1
# stress-ng mem stress process number
workers=$2
# prepare for prerequisites
if [ ! "$(which stress-ng)" ]; then
dnf install -y stress-ng > /dev/null
apt install -y stress-ng > /dev/null
else
test_print_trc "stress-ng prerequisites is ready for use"
test_print_trc "unaccepted memory drained time calculation is starting now..."
fi
# calculate memory accepted fully completed time
SECONDS=0
stress-ng --vm "$workers" --vm-bytes 100% &
while (true); do
if [[ $(grep "nr_unaccepted" /proc/vmstat | awk '{print $2}') -eq 0 ]]; then
actual_time=$SECONDS;
pkill stress-ng;
break;
# calculate memory accepted fully completed time
SECONDS=0
stress-ng --vm "$workers" --vm-bytes 100% &
while (true); do
if [[ $(grep "nr_unaccepted" /proc/vmstat | awk '{print $2}') -eq 0 ]]; then
actual_time=$SECONDS;
pkill stress-ng;
break;
fi
done
# check if memory accept time far exceed expected (passed in value)
compare_result=$(echo "scale=2; ($actual_time/$expected_time)" | bc)
baseline_result=1.1
result=$(awk -v n1="$compare_result" -v n2="$baseline_result" 'BEGIN {if (n1>n2) print 1; else print 0}')
if [ "$result" -eq 0 ]; then
test_print_trc "Memory accepted full time consumed: $actual_time"
return 0
else
die "Memory accepted full time consumed: $actual_time seconds"
die "It's over expectation $expected_time seconds 10% more!!!"
return 1
fi
}

# function based on stress-ng do basic mem lazy accept function check
mem_accept_func() {
# prepare for prerequisites
if [ ! "$(which stress-ng)" ]; then
dnf install -y stress-ng > /dev/null
apt install -y stress-ng > /dev/null
else
test_print_trc "stress-ng prerequisites is ready for use"
test_print_trc "unaccepted memory drained time calculation is starting now..."
fi
bootup_vmstat=$(grep "nr_unaccepted" /proc/vmstat | cut -d' ' -f2)
stress-ng --vm 1 --vm-bytes 10% --timeout 3s
stress_vmstat=$(grep "nr_unaccepted" /proc/vmstat | cut -d' ' -f2)
if [[ "$bootup_vmstat" -gt "$stress_vmstat" ]]; then
test_print_trc "TD VM unaccepted memory func test PASS"
else
die "TD VM unaccepted memory func test FAIL"
return 1
fi
}

# function mem lazy accept info calculation
mem_accept_cal() {
test_print_trc "Start TD VM unaccepted memory info calculation check"
vmstat=$(grep "nr_unaccepted" /proc/vmstat | cut -d' ' -f2)
meminfo=$(awk '/Unaccepted/{printf "%d\n", $2;}' </proc/meminfo)
pagesize=$(getconf PAGESIZE)
test_print_trc "vmstat: $vmstat; meminfo= $meminfo"
if [[ $((vmstat * pagesize / 1024)) -eq "$meminfo" ]]; then
test_print_trc "TD VM unaccepted memory info calculation PASS"
else
die "TD VM unaccepted memory info calculation FAIL"
return 1
fi
}

# function mem lazy accept negative test
mem_accept_neg() {
test_print_trc "Start TD VM unaccepted memory negative test"
vmstat_disable=$(grep "nr_unaccepted" /proc/vmstat | cut -d' ' -f2)
if [[ "$vmstat_disable" -eq 0 ]]; then
test_print_trc "TD VM unaccepted memory negative test"
else
die "TD VM unaccepted memory negative test FAIL with nr_unaccepted: $vmstat_disable"
return 1
fi
done
# check if memory accept time far exceed expected (passed in value)
compare_result=$(echo "scale=2; ($actual_time/$expected_time)" | bc)
baseline_result=1.1
result=$(awk -v n1="$compare_result" -v n2="$baseline_result" 'BEGIN {if (n1>n2) print 1; else print 0}')
if [ "$result" -eq 0 ]; then
test_print_trc "Memory accepted full time consumed: $actual_time"
return 0
else
die "Memory accepted full time consumed: $actual_time seconds"
die "It's over expectation $expected_time seconds 10% more!!!"
return 1
fi
}

###################### Do Works ######################
Expand Down Expand Up @@ -146,6 +194,15 @@ case "$MEM_CASE" in
# 32VCPU + 96G MEM + 256 mem stress processes
mem_accepted_time 92 256
;;
MEM_ACPT_FUNC)
mem_accept_func
;;
MEM_ACPT_CAL)
mem_accept_cal
;;
MEM_ACPT_NEG)
mem_accept_neg
;;
:)
test_print_err "Must specify the memory case option by [-t]"
exit 1
Expand Down
5 changes: 4 additions & 1 deletion guest-test/tdx/tests
View file
Original file line number Diff line number Diff line change
Expand Up @@ -31,4 +31,7 @@ guest.test_launcher.sh -v 32 -s 1 -m 8 -d on -t tdx -f tdx -x TD_MEM_ACPT_T_32C_
guest.test_launcher.sh -v 32 -s 1 -m 32 -d on -t tdx -f tdx -x TD_MEM_ACPT_T_32C_32G_32W -c "accept_memory=lazy" -p off
guest.test_launcher.sh -v 32 -s 1 -m 32 -d on -t tdx -f tdx -x TD_MEM_ACPT_T_32C_32G_256W -c "accept_memory=lazy" -p off
guest.test_launcher.sh -v 32 -s 1 -m 96 -d on -t tdx -f tdx -x TD_MEM_ACPT_T_32C_96G_32W -c "accept_memory=lazy" -p off
guest.test_launcher.sh -v 32 -s 1 -m 96 -d on -t tdx -f tdx -x TD_MEM_ACPT_T_32C_96G_256W -c "accept_memory=lazy" -p off
guest.test_launcher.sh -v 32 -s 1 -m 96 -d on -t tdx -f tdx -x TD_MEM_ACPT_T_32C_96G_256W -c "accept_memory=lazy" -p off
guest.test_launcher.sh -v 1 -s 1 -m 16 -d on -t tdx -f tdx -x TD_MEM_ACPT_FUNC -c "accept_memory=lazy" -p off
guest.test_launcher.sh -v 1 -s 1 -m 16 -d on -t tdx -f tdx -x TD_MEM_ACPT_CAL -c "accept_memory=lazy" -p off
guest.test_launcher.sh -v 1 -s 1 -m 16 -d on -t tdx -f tdx -x TD_MEM_ACPT_NEG -c "accept_memory=eager" -p off

0 comments on commit f091173

Please sign in to comment.