firecracker

1
#!/usr/bin/env python3
2
# Copyright 2023 Amazon.com, Inc. or its affiliates. All Rights Reserved.
3
# SPDX-License-Identifier: Apache-2.0
4

5
# pylint:disable=invalid-name
6

7
"""
8
Test all the ext4 rootfs in the current directory
9
"""
10

11
import os
12
import sys
13
from pathlib import Path
14

15
# Hack to be able to import testing framework functions.
16
sys.path.append(os.path.join(os.getcwd(), "tests"))
17

18
# pylint: disable=wrong-import-position
19
from framework.artifacts import kernels
20
from framework.microvm import MicroVMFactory
21
from host_tools.cargo_build import get_firecracker_binaries
22

23
# pylint: enable=wrong-import-position
24

25
kernels = list(kernels("vmlinux-*"))
26
# Use the latest guest kernel
27
kernel = kernels[-1]
28

29
vmfcty = MicroVMFactory(*get_firecracker_binaries())
30
# (may take a while to compile Firecracker...)
31

32
for rootfs in Path(".").glob("*.ext4"):
33
    print(f">>>> Testing {rootfs}")
34
    uvm = vmfcty.build(kernel, rootfs)
35
    uvm.spawn()
36
    uvm.add_net_iface()
37
    uvm.basic_config()
38
    uvm.start()
39
    rc, stdout, stderr = uvm.ssh.run("cat /etc/issue")
40
    print(rc, stdout, stderr)
41