test_rtc.py

# Copyright 2020 Amazon.com, Inc. or its affiliates. All Rights Reserved.
# SPDX-License-Identifier: Apache-2.0
"""Check the well functioning af the RTC device on aarch64 platforms."""
import re
import platform
import pytest

import framework.utils as utils
from host_tools.network import SSHConnection

DMESG_LOG_REGEX = r'rtc-pl031\s+(\d+).rtc: setting system clock to'


@pytest.mark.skipif(
    platform.machine() != "aarch64",
    reason="RTC exists only on aarch64."
)
def test_rtc(test_microvm_with_ssh, network_config):
    """Test RTC functionality on aarch64."""
    vm = test_microvm_with_ssh
    vm.spawn()
    vm.memory_monitor = None
    vm.basic_config()
    _tap, _, _ = vm.ssh_network_config(network_config, '1')

    vm.start()
    conn = SSHConnection(vm.ssh_config)

    # check that the kernel creates an rtcpl031 base device.
    _, stdout, _ = conn.execute_command("dmesg")
    rtc_log = re.findall(DMESG_LOG_REGEX, stdout.read())
    assert rtc_log is not None

    _, stdout, _ = conn.execute_command("stat /dev/rtc0")
    assert "character special file" in stdout.read()

    _, host_stdout, _ = utils.run_cmd("date +%s")
    _, guest_stdout, _ = conn.execute_command("date +%s")
    assert abs(int(guest_stdout.read()) - int(host_stdout)) < 5