Skip to content

Commit

Permalink
Fix test scripts
Browse files Browse the repository at this point in the history 
Signed-off-by: Wei Liu <[email protected]>
  • Loading branch information
@liuw1
liuw1 committed Dec 11, 2024
1 parent 0ff8022 commit c14bcb3
Show file tree
Hide file tree
Showing 3 changed files with 68 additions and 46 deletions.
98 changes: 59 additions & 39 deletions sh_script/test/integration_test.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -36,11 +36,12 @@ def test_function_001(device_type):
migtd_dst = "../../Bin/migtd_001.bin"

with migtd_context() as ctx:
ctx.start_mig_td(bios_img=migtd_src, type="src", device=device_type)
ctx.start_mig_td(bios_img=migtd_dst, type="dst", device=device_type)
ctx.start_mig_td(bios_img=migtd_src, type="src", device=device_type)
ctx.start_user_td(type="src")
ctx.start_user_td(type="dst")
ctx.connect()
if device_type == "vsock":
ctx.connect()
ctx.pre_migration()
ctx.check_migration_result()

Expand All @@ -56,11 +57,12 @@ def test_function_002(device_type):
migtd_dst = "../../Bin/migtd_001.bin"

with migtd_context() as ctx:
ctx.start_mig_td(bios_img=migtd_src, type="src", device=device_type)
ctx.start_mig_td(bios_img=migtd_dst, type="dst", device=device_type)
ctx.start_mig_td(bios_img=migtd_src, type="src", device=device_type)
ctx.start_user_td(type="src")
ctx.start_user_td(type="dst")
ctx.connect()
if device_type == "vsock":
ctx.connect()
ctx.pre_migration()
ctx.check_migration_result()

Expand All @@ -76,11 +78,12 @@ def test_function_negative_003(device_type):
migtd_dst = "../../Bin/migtd_001.bin"

with migtd_context() as ctx:
ctx.start_mig_td(bios_img=migtd_src, type="src", device=device_type)
ctx.start_mig_td(bios_img=migtd_dst, type="dst", device=device_type)
ctx.start_mig_td(bios_img=migtd_src, type="src", device=device_type)
ctx.start_user_td(type="src")
ctx.start_user_td(type="dst")
ctx.connect()
if device_type == "vsock":
ctx.connect()
ctx.pre_migration()
ctx.check_migration_result(negative=True)

Expand All @@ -96,11 +99,12 @@ def test_function_004(device_type):
migtd_dst = "../../Bin/migtd_004.bin"

with migtd_context() as ctx:
ctx.start_mig_td(bios_img=migtd_src, type="src", device=device_type)
ctx.start_mig_td(bios_img=migtd_dst, type="dst", device=device_type)
ctx.start_mig_td(bios_img=migtd_src, type="src", device=device_type)
ctx.start_user_td(type="src")
ctx.start_user_td(type="dst")
ctx.connect()
if device_type == "vsock":
ctx.connect()
ctx.pre_migration()
ctx.check_migration_result()

Expand All @@ -116,11 +120,12 @@ def test_function_005(device_type):
migtd_dst = "../../Bin/migtd_005.bin"

with migtd_context() as ctx:
ctx.start_mig_td(bios_img=migtd_src, type="src", device=device_type)
ctx.start_mig_td(bios_img=migtd_dst, type="dst", device=device_type)
ctx.start_mig_td(bios_img=migtd_src, type="src", device=device_type)
ctx.start_user_td(type="src")
ctx.start_user_td(type="dst")
ctx.connect()
if device_type == "vsock":
ctx.connect()
ctx.pre_migration()
ctx.check_migration_result()

Expand All @@ -136,11 +141,12 @@ def test_function_negative_006(device_type):
migtd_dst = "../../Bin/migtd_006.bin"

with migtd_context() as ctx:
ctx.start_mig_td(bios_img=migtd_src, type="src", device=device_type)
ctx.start_mig_td(bios_img=migtd_dst, type="dst", device=device_type)
ctx.start_mig_td(bios_img=migtd_src, type="src", device=device_type)
ctx.start_user_td(type="src")
ctx.start_user_td(type="dst")
ctx.connect()
if device_type == "vsock":
ctx.connect()
ctx.pre_migration()
ctx.check_migration_result(negative=True)

Expand All @@ -156,11 +162,12 @@ def test_function_negative_007(device_type):
migtd_dst = "../../Bin/migtd_no.bin"

with migtd_context() as ctx:
ctx.start_mig_td(bios_img=migtd_src, type="src", device=device_type)
ctx.start_mig_td(bios_img=migtd_dst, type="dst", device=device_type)
ctx.start_mig_td(bios_img=migtd_src, type="src", device=device_type)
ctx.start_user_td(type="src")
ctx.start_user_td(type="dst")
ctx.connect()
if device_type == "vsock":
ctx.connect()
ctx.pre_migration()
ctx.check_migration_result(negative=True)

Expand All @@ -176,11 +183,12 @@ def test_function_negative_008(device_type):
migtd_dst = "../../Bin/migtd_no.bin"

with migtd_context() as ctx:
ctx.start_mig_td(bios_img=migtd_src, type="src", device=device_type)
ctx.start_mig_td(bios_img=migtd_dst, type="dst", device=device_type)
ctx.start_mig_td(bios_img=migtd_src, type="src", device=device_type)
ctx.start_user_td(type="src")
ctx.start_user_td(type="dst")
ctx.connect()
if device_type == "vsock":
ctx.connect()
ctx.pre_migration()
ctx.check_migration_result(negative=True)

Expand All @@ -196,11 +204,12 @@ def test_function_negative_009(device_type):
migtd_dst = "../../Bin/migtd_009.bin"

with migtd_context() as ctx:
ctx.start_mig_td(bios_img=migtd_dst, type="dst", device=device_type)
ctx.start_mig_td(bios_img=migtd_src, type="src", no_device=True, device=device_type)
ctx.start_mig_td(bios_img=migtd_dst, type="dst", device=device_type)
ctx.start_user_td(type="src")
ctx.start_user_td(type="dst")
ctx.connect()
if device_type == "vsock":
ctx.connect()
ctx.pre_migration()
ctx.check_migration_result(negative=True, wait_time=10)

Expand All @@ -212,11 +221,12 @@ def test_function_010(device_type):
migtd_dst = "../../Bin/migtd_010.bin"

with migtd_context() as ctx:
ctx.start_mig_td(bios_img=migtd_src, type="src", device=device_type)
ctx.start_mig_td(bios_img=migtd_dst, type="dst", device=device_type)
ctx.start_mig_td(bios_img=migtd_src, type="src", device=device_type)
ctx.start_user_td(type="src")
ctx.start_user_td(type="dst")
ctx.connect()
if device_type == "vsock":
ctx.connect()
ctx.pre_migration()
ctx.check_migration_result()

Expand All @@ -228,11 +238,12 @@ def test_function_011(device_type):
migtd_dst = "../../Bin/migtd_011.bin"

with migtd_context() as ctx:
ctx.start_mig_td(bios_img=migtd_src, type="src", device=device_type)
ctx.start_mig_td(bios_img=migtd_dst, type="dst", device=device_type)
ctx.start_mig_td(bios_img=migtd_src, type="src", device=device_type)
ctx.start_user_td(type="src")
ctx.start_user_td(type="dst")
ctx.connect()
if device_type == "vsock":
ctx.connect()
ctx.pre_migration()
ctx.check_migration_result()

Expand All @@ -248,11 +259,12 @@ def test_function_012(device_type):
migtd_dst = "../../Bin/migtd_dst_012.bin"

with migtd_context() as ctx:
ctx.start_mig_td(bios_img=migtd_src, type="src", device=device_type)
ctx.start_mig_td(bios_img=migtd_dst, type="dst", device=device_type)
ctx.start_mig_td(bios_img=migtd_src, type="src", device=device_type)
ctx.start_user_td(type="src")
ctx.start_user_td(type="dst")
ctx.connect()
if device_type == "vsock":
ctx.connect()
ctx.pre_migration()
ctx.check_migration_result()

Expand All @@ -268,11 +280,12 @@ def test_function_013(device_type):
migtd_dst = "../../Bin/migtd_dst_013.bin"

with migtd_context() as ctx:
ctx.start_mig_td(bios_img=migtd_src, type="src", device=device_type)
ctx.start_mig_td(bios_img=migtd_dst, type="dst", device=device_type)
ctx.start_mig_td(bios_img=migtd_src, type="src", device=device_type)
ctx.start_user_td(type="src")
ctx.start_user_td(type="dst")
ctx.connect()
if device_type == "vsock":
ctx.connect()
ctx.pre_migration()
ctx.check_migration_result()

Expand All @@ -288,11 +301,12 @@ def test_function_negative_014(device_type):
migtd_dst = "../../Bin/migtd_dst_014.bin"

with migtd_context() as ctx:
ctx.start_mig_td(bios_img=migtd_src, type="src", device=device_type)
ctx.start_mig_td(bios_img=migtd_dst, type="dst", device=device_type)
ctx.start_mig_td(bios_img=migtd_src, type="src", device=device_type)
ctx.start_user_td(type="src")
ctx.start_user_td(type="dst")
ctx.connect()
if device_type == "vsock":
ctx.connect()
ctx.pre_migration()
ctx.check_migration_result(negative=True)

Expand All @@ -308,11 +322,12 @@ def test_function_negative_015(device_type):
migtd_dst = "../../Bin/migtd_015.bin"

with migtd_context() as ctx:
ctx.start_mig_td(bios_img=migtd_src, type="src", device=device_type)
ctx.start_mig_td(bios_img=migtd_dst, type="dst", device=device_type)
ctx.start_mig_td(bios_img=migtd_src, type="src", device=device_type)
ctx.start_user_td(type="src")
ctx.start_user_td(type="dst")
ctx.connect()
if device_type == "vsock":
ctx.connect()
ctx.pre_migration()
ctx.check_migration_result(negative=True)

Expand All @@ -328,11 +343,12 @@ def test_function_negative_016(device_type):
migtd_dst = "../../Bin/migtd_016.bin"

with migtd_context() as ctx:
ctx.start_mig_td(bios_img=migtd_src, type="src", device=device_type)
ctx.start_mig_td(bios_img=migtd_dst, type="dst", device=device_type)
ctx.start_mig_td(bios_img=migtd_src, type="src", device=device_type)
ctx.start_user_td(type="src")
ctx.start_user_td(type="dst")
ctx.connect()
if device_type == "vsock":
ctx.connect()
ctx.pre_migration()
ctx.check_migration_result(negative=True)

Expand All @@ -348,11 +364,12 @@ def test_function_negative_017(device_type):
migtd_dst = "../../Bin/migtd_017.bin"

with migtd_context() as ctx:
ctx.start_mig_td(bios_img=migtd_src, type="src", device=device_type)
ctx.start_mig_td(bios_img=migtd_dst, type="dst", device=device_type)
ctx.start_mig_td(bios_img=migtd_src, type="src", device=device_type)
ctx.start_user_td(type="src")
ctx.start_user_td(type="dst")
ctx.connect()
if device_type == "vsock":
ctx.connect()
ctx.pre_migration()
ctx.check_migration_result(negative=True)

Expand All @@ -372,7 +389,8 @@ def test_function_negative_018(device_type):
ctx.start_mig_td(bios_img=migtd_dst, type="dst", device=device_type)
ctx.start_user_td(type="src")
ctx.start_user_td(type="dst")
ctx.connect()
if device_type == "vsock":
ctx.connect()
ctx.pre_migration()
ctx.check_migration_result(negative=True)

Expand Down Expand Up @@ -400,9 +418,10 @@ def test_pre_binding(device_type, servtd_hash):
with migtd_context() as ctx:
ctx.start_user_td(type="src", is_pre_binding=True, hash=servtd_hash)
ctx.start_user_td(type="dst", is_pre_binding=True, hash=servtd_hash)
ctx.start_mig_td(bios_img=migtd_src, type="src", device=device_type)
ctx.start_mig_td(bios_img=migtd_dst, type="dst", device=device_type)
ctx.connect()
ctx.start_mig_td(bios_img=migtd_src, type="src", device=device_type)
if device_type == "vsock":
ctx.connect()

ctx.pre_migration(is_pre_binding=True)
ctx.check_migration_result()
Expand All @@ -417,9 +436,10 @@ def test_cycle(device_type):
migtd_dst = "../../target/release/migtd.bin"

with migtd_context() as ctx:
ctx.start_mig_td(bios_img=migtd_src, type="src", device=device_type)
ctx.start_mig_td(bios_img=migtd_dst, type="dst", device=device_type)
ctx.connect()
ctx.start_mig_td(bios_img=migtd_src, type="src", device=device_type)
if device_type == "vsock":
ctx.connect()

for i in range(ctx.stress_test_cycles):
LOG.debug(f"#### Cycle Test: {i} ####")
Expand Down
8 changes: 5 additions & 3 deletions sh_script/test/mig-td.sh
View file
Original file line number Diff line number Diff line change
Expand Up @@ -6,9 +6,9 @@
#
set -e

QEMU_EXEC="/usr/local/bin/qemu-system-x86_64"
QEMU_EXEC="/usr/libexec/qemu-kvm"
GUEST_CID=18
MIGTD="/usr/share/td-migration/migtd.bin"
MIGTD=""
VIRTIO_SERIAL=false
MIGTD_TYPE=""
DEST_IP="127.0.0.1"
Expand All @@ -19,6 +19,8 @@ Usage: $(basename "$0") [OPTION]...
-q <qemu path> QEMU path
-m <migtd file> MigTD file
-t <src|dst> Must set migtd type, src or dst
-s Use virtio serial, default is vsock
-i <dest_ip> IP of destination
-h Show this help
EOM
}
Expand Down Expand Up @@ -93,7 +95,7 @@ QEMU_CMD="${QEMU_EXEC} \
if [[ $NO_DEVICE != true ]]; then
if [[ ${VIRTIO_SERIAL} == true ]]; then
QEMU_CMD+=" -device virtio-serial-pci,id=virtio-serial0 "
if [[ ${MIGTD_TYPE} == "src" ]]; then
if [[ ${MIGTD_TYPE} == "dst" ]]; then
QEMU_CMD+=" -chardev socket,host=0.0.0.0,port=1236,server=on,id=foo "
else
QEMU_CMD+=" -chardev socket,host=${DEST_IP},port=1236,server=off,id=foo "
Expand Down
8 changes: 4 additions & 4 deletions sh_script/test/user-td.sh
View file
Original file line number Diff line number Diff line change
Expand Up @@ -6,11 +6,11 @@
#
set -e

QEMU_EXEC="/usr/local/bin/qemu-system-x86_64"
QEMU_EXEC="/usr/libexec/qemu-kvm"
TD_TYPE=""
KERNEL=""
GUEST_IMG=""
OVMF="/home/env/OVMF.fd"
OVMF="/usr/share/qemu/OVMF.fd"
BOOT_TYPE="direct"
TARGET_PID=""
ROOT_PARTITION="/dev/vda3"
Expand All @@ -31,7 +31,7 @@ Usage: $(basename "$0") [OPTION]...
-k <kernel file> Kernel file
-b [direct|grub] Boot type, default is "direct" which requires kernel binary specified via "-k"
-p incoming port
-q [tdvmcall|vsock] Support for TD quote using tdvmcall or vsock
-a [tdvmcall|vsock] Support for TD quote using tdvmcall or vsock
-r <root partition> root partition for direct boot, default is /dev/vda1
-t <src|dst> Must set userTD type, src or dst
-g [true|false] Use pre-binding or not, default is "false"
Expand All @@ -41,7 +41,7 @@ EOM
}

process_args() {
while getopts "i:k:b:o:p:q:r:t:g:m:h" option; do
while getopts "i:k:b:o:p:q:a:r:t:g:m:h" option; do
case "${option}" in
i) GUEST_IMG=$OPTARG;;
k) KERNEL=$OPTARG;;
Expand Down

0 comments on commit c14bcb3

Please sign in to comment.