Onboard MDE for Linux to LISA

lisa/sut_orchestrator/azure/tools.py

@@ -2,6 +2,7 @@
 # Licensed under the MIT license.
 
 import re
+import json
 from pathlib import PurePath
 from typing import Any, Dict, List, Optional, Tuple, Type
 
@@ -561,3 +562,32 @@ def get_pool_records(self, pool_id: int, force_run: bool = False) -> Dict[str, s
             records[content_split[i]] = content_split[i + 1]
 
         return records
+
+class mdatp(Tool):
+    @property
+    def command(self) -> str:
+        return "mdatp"
+
+    @property
+    def can_install(self) -> bool:
+        return False
+
+    def get_result(
+        self,
+        arg: str,
+        json_out: bool = False,
+        sudo: bool = False,
+    ) -> None:
+        if json_out:
+            arg += ' --output json'
+        result = self.run(
+            arg,
+            sudo=sudo,
+            shell=True,
+        )
+
+        result.assert_exit_code()
+        if json_out:
+            return json.loads(result.stdout)
+        return result.stdout.split()
+

microsoft/testsuites/vm_extensions/mde.py

@@ -0,0 +1,54 @@
+import time
+from typing import cast
+
+from assertpy import assert_that
+from retry import retry
+
+from lisa import (
+    Logger,
+    Node,
+    TestCaseMetadata,
+    TestSuite,
+    TestSuiteMetadata,
+    simple_requirement,
+)
+from lisa.operating_system import BSD, Posix
+from lisa.sut_orchestrator.azure.common import (
+    add_tag_for_vm,
+    add_user_assign_identity,
+    get_managed_service_identity_client,
+    get_node_context,
+)
+from lisa.sut_orchestrator.azure.features import AzureExtension
+from lisa.sut_orchestrator.azure.platform_ import AzurePlatform
+from lisa.sut_orchestrator.azure.tools import Azsecd
+from lisa.sut_orchestrator.azure.tools import mdatp
+from lisa.testsuite import TestResult
+from lisa.tools import Cat, Journalctl, Service
+from lisa.util import LisaException, UnsupportedDistroException
+
+
+@TestSuiteMetadata(
+    area="vm_extension",
+    category="functional",
+    description="""
+    MDE Test Suite
+    """,
+)
+class MDE(TestSuite):
+    @TestCaseMetadata(
+        description="""
+            Verify if MDE is healthy
+        """,
+        priority=1,
+        requirement=simple_requirement(
+            #supported_features=[AzureExtension], unsupported_os=[BSD]
+        ),
+    )
+    def verify_health(self, node: Node, log: Logger, result: TestResult) -> None:
+        output = node.tools[mdatp].get_result('health', json_out=True)
+
+        log.info(output)
+
+        assert_that(output['healthy']).is_equal_to(True)
+