# Copyright 2018 Amazon.com, Inc. or its affiliates. All Rights Reserved.
# SPDX-License-Identifier: Apache-2.0
"""Tests for initrd."""

from framework.microvm import Serial

INITRD_FILESYSTEM = "rootfs"


def test_microvm_initrd_with_serial(test_microvm_with_initrd):
    """
    Test that a boot using initrd successfully loads the root filesystem.
    """
    vm = test_microvm_with_initrd
    vm.jailer.daemonize = False
    vm.spawn()
    vm.memory_monitor = None

    vm.basic_config(
        add_root_device=False,
        vcpu_count=1,
        boot_args="console=ttyS0 reboot=k panic=1 pci=off",
        use_initrd=True,
    )

    vm.start()
    serial = Serial(vm)
    serial.open()
    serial.rx(token="login: ")
    serial.tx("root")

    serial.rx(token="Password: ")
    serial.tx("root")

    serial.rx(token="# ")

    serial.tx("findmnt /")
    serial.rx(token=f"/      {INITRD_FILESYSTEM} {INITRD_FILESYSTEM}")