Skip to content

Commit

Permalink
Add device-passthrough testcase for CH platform
Browse files Browse the repository at this point in the history
Add test case to verify if passthrough-devices
are visible to guest for CH platform

Signed-off-by: Smit Gardhariya <[email protected]>
  • Loading branch information
smit-gardhariya committed Sep 18, 2024
1 parent a1358de commit 0d77e02
Showing 1 changed file with 110 additions and 0 deletions.
110 changes: 110 additions & 0 deletions microsoft/testsuites/device_passthrough/functional_tests.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,110 @@
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.
from pathlib import Path
from typing import Any, Dict, cast

from lisa import (
Environment,
Logger,
Node,
TestCaseMetadata,
TestSuite,
TestSuiteMetadata,
)
from lisa.sut_orchestrator import CLOUD_HYPERVISOR
from lisa.sut_orchestrator.libvirt.ch_platform import CloudHypervisorPlatform
from lisa.sut_orchestrator.libvirt.schema import BaseLibvirtNodeSchema
from lisa.testsuite import TestResult, simple_requirement
from lisa.tools import Lspci
from lisa.util import LisaException, SkippedException


@TestSuiteMetadata(
area="device_passthrough",
category="functional",
description="""
This test suite is for testing device passthrough functional tests.
""",
)
class DevicePassthroughFunctionalTests(TestSuite):
@TestCaseMetadata(
description="""
Check if passthrough device is visible to guest.
This testcase support only on CLOUD_HYPERVISOR
platform of LISA. Please refer below runbook snippet.
platform:
- type: cloud-hypervisor
cloud-hypervisor:
device_pools:
- type: "pci_net"
devices:
- vendor_id: xxx
device_id: xxx
requirement:
cloud-hypervisor:
device_passthrough:
- pool_type: "pci_net"
managed: "yes"
count: 1
We will check if sufficient devices are visible to guest or not.
Platform will create device pool based on given device/vendor id.
'device_passthrough' section will tell platform to create node
with appropriate num of devices being passthrough. Based on pool_type
value, platform will try to get devices from pool and assign it to node.
Testcase will verify if needed devices are present on node by reading
extended runbook. It will get the vendor/device id based on 'pool_type'
and check how many devices are present on node for that vendor/device id.
""",
priority=4,
requirement=simple_requirement(
supported_platform_type=[CLOUD_HYPERVISOR],
),
)
def verify_device_passthrough_on_guest(
self,
log: Logger,
node: Node,
environment: Environment,
log_path: Path,
result: TestResult,
variables: Dict[str, Any],
) -> None:
lspci = node.tools[Lspci]
platform = cast(CloudHypervisorPlatform, environment.platform)
pool_vendor_device_map = {}
assert platform.platform_runbook.device_pools, "Device pool cant be empty"
for pool in platform.platform_runbook.device_pools:
pool_type = str(pool.type.value)
vendor_device_id = {
"vendor_id": pool.devices[0].vendor_id,
"device_id": pool.devices[0].device_id,
}
pool_vendor_device_map[pool_type] = vendor_device_id

assert environment.runbook.nodes_requirement, "requirement cant be empty"
for node_space in environment.runbook.nodes_requirement:
node_runbook: BaseLibvirtNodeSchema = node_space.get_extended_runbook(
BaseLibvirtNodeSchema, CLOUD_HYPERVISOR
)
if not node_runbook.device_passthrough:
raise SkippedException("No device-passthrough is set for node")
for req in node_runbook.device_passthrough:
pool_type = str(req.pool_type.value)
ven_dev_id_of_pool = pool_vendor_device_map[pool_type]
ven_id = ven_dev_id_of_pool["vendor_id"]
dev_id = ven_dev_id_of_pool["device_id"]
devices = lspci.get_devices_by_vendor_device_id(
vendor_id=ven_id,
device_id=dev_id,
force_run=True,
)
if len(devices) >= req.count:
raise LisaException(
f"Device count don't match, got total: {len(devices)} "
f"As per runbook, Required device count: {req.count}, "
f"for pool_type: {pool_type} having Vendor/Device ID as "
f"vendor_id = {ven_id}, device_id={dev_id}"
)

0 comments on commit 0d77e02

Please sign in to comment.