Using Azure SAS Uri for getting onboarding script

lisa/tools/mde.py

@@ -2,17 +2,14 @@
 # Licensed under the MIT license.
 
 import os
-import sys
 import json
 import requests
-from pathlib import Path, PurePath
+from pathlib import Path
 from typing import Any
 
 from lisa.executable import Tool
 from lisa.executable import CustomScriptBuilder, CustomScript
-#from . import RemoteCopy, Whoami, Curl
-from .remote_copy import RemoteCopy
-from .whoami import Whoami
+from lisa.base_tools import Wget
 
 class MDE(Tool):
     @property
@@ -50,23 +47,25 @@ def _install(self) -> bool:
         return self._check_exists()
 
 
-    def onboard(self, onboarding_script: PurePath) -> bool:
+    def onboard(self, onboarding_script_sas_uri: str) -> bool:
         if not self._check_exists():
             self._log.error("MDE is not installed, onboarding not possible")
             return False
 
-        username = self.node.tools[Whoami].get_username()
+        wget = self.node.tools[Wget]
 
-        remote_copy = self.node.tools[RemoteCopy]
-        remote_copy.copy_to_remote(onboarding_script, PurePath(f"/home/{username}/MicrosoftDefenderATPOnboardingLinuxServer.py"))
+        download_path = wget.get(
+            url=onboarding_script_sas_uri,
+            filename="MicrosoftDefenderATPOnboardingLinuxServer.py",
+        )
 
         if not self.get_mde_installer():
            self._log.error("Unable to download mde_installer.sh script. MDE can't be onboarded")
 
         script: CustomScript = self.node.tools[self._mde_installer]
 
         self._log.info('Onboarding MDE')
-        result1 = script.run(parameters=f"--onboard /home/{username}/MicrosoftDefenderATPOnboardingLinuxServer.py/MicrosoftDefenderATPOnboardingLinuxServer.py", sudo=True)
+        result1 = script.run(parameters=f"--onboard {download_path}", sudo=True)
         self._log.info(result1)
 
         output = self.get_result('health --field licensed')

microsoft/runbook/my.yml

@@ -10,6 +10,10 @@ variable:
     is_secret: true
     is_case_visible: True
     value: ""
+  - name: onboarding_script_sas_uri
+    is_secret: true
+    is_case_visible: True
+    value: ""
   - name: location
     value: "westus3"
   - name: keep_environment

microsoft/testsuites/vm_extensions/mde.py

@@ -29,9 +29,9 @@ class MDE(TestSuite):
 
     def before_case(self, log: Logger, **kwargs: Any) -> None:
         variables = kwargs["variables"]
-        self.onboarding_script = variables.get("onboarding_script", "")
-        if not self.onboarding_script:
-            raise SkippedException("Onboarding script is not provided.")
+        self.onboarding_script_sas_uri = variables.get("onboarding_script_sas_uri", "")
+        if not self.onboarding_script_sas_uri:
+            raise SkippedException("Onboarding script SAS URI is not provided.")
 
     @TestCaseMetadata(
         description="""
@@ -61,7 +61,7 @@ def verify_mde(self, node: Node, log: Logger, result: TestResult) -> None:
 
     def verify_onboard(self, node: Node, log: Logger, result: TestResult) -> None:
 
-        onboarding_result = node.tools[mdatp].onboard(PurePath(self.onboarding_script))
+        onboarding_result = node.tools[mdatp].onboard(self.onboarding_script_sas_uri)
 
         assert_that(onboarding_result).is_equal_to(True)