Raise Exception for Azure Linux4 and RHEL10 not supoorted yet

Author: yashnap

src/core/src/bootstrap/EnvLayer.py

@@ -49,7 +49,7 @@ def is_distro_azure_linux(distro_name):
 
     def is_distro_azure_linux_3_or_beyond(self):
         # type: () -> bool
-        """ Checks if the current distro is Azure Linux 3 """
+        """ Checks if the current distro is Azure Linux 3 or beyond """
         if self.is_distro_azure_linux(self.platform.linux_distribution()):
             version = distro.os_release_attr('version')
             major = version.split('.')[0] if version else None
@@ -62,15 +62,33 @@ def get_package_manager(self):
         if platform.system() == 'Windows':
             return Constants.APT
 
+        version = distro.os_release_attr('version')
+        major = version.split('.')[0] if version else None
+        os_id = distro.id()
+
+        # Check for Azure Linux
         if self.is_distro_azure_linux(str(self.platform.linux_distribution())):
+            # Azure Linux 4 not yet supported
+            if major is not None and int(major) >= 4:
+                error_msg = "Azure Linux 4 is not yet supported in your region. Please review aka.ms/LinuxPatchExtension for more information."
+                print("Error: {0}".format(error_msg))
+                raise Exception(error_msg)
+
+            # Azure Linux 3 and below use TDNF
             code, out = self.run_command_output('which tdnf', False, False)
             if code == 0:
                 return Constants.TDNF
             else:
                 print("Error: Expected package manager tdnf not found on this Azure Linux VM.")
                 return str()
 
-        # choose default package manager
+        # Check for RHEL 10 (not yet supported)
+        if os_id == "rhel" and major is not None and int(major) >= 10:
+            error_msg = "RHEL 10 is not yet supported in your region. Please review aka.ms/LinuxPatchExtension for more information."
+            print("Error: {0}".format(error_msg))
+            raise Exception(error_msg)
+
+        # Choose default package manager
         package_manager_map = (('apt-get', Constants.APT),
                                ('yum', Constants.YUM),
                                ('zypper', Constants.ZYPPER))

src/core/tests/Test_EnvLayer.py

@@ -71,6 +71,18 @@ def mock_distro_os_release_attr_return_azure_linux_2(self, attribute):
 
     def mock_distro_os_release_attr_return_none(self, attribute):
         return None
+
+    def mock_distro_os_release_attr_return_azure_linux_4(self, attribute):
+        return '4.0.2'
+
+    def mock_distro_os_release_attr_return_rhel_10(self, attribute):
+        return '10.0'
+
+    def mock_linux_distribution_to_return_azure_linux_4(self):
+        return ['Microsoft Azure Linux', '4.0', '']
+
+    def mock_distro_id_return_rhel(self):
+        return 'rhel'
     # endregion
 
     def test_get_package_manager(self):
@@ -139,6 +151,43 @@ def test_platform(self):
         self.envlayer.platform.cpu_arch()
         self.envlayer.platform.vm_name()
 
+    def test_get_package_manager_azure_linux_4_not_supported(self):
+        """Test that Azure Linux 4 raises an exception"""
+        self.backup_linux_distribution = self.envlayer.platform.linux_distribution
+        self.backup_distro_os_release_attr = distro.os_release_attr
+
+        platform.system = self.mock_platform_system
+        self.envlayer.platform.linux_distribution = self.mock_linux_distribution_to_return_azure_linux_4
+        distro.os_release_attr = self.mock_distro_os_release_attr_return_azure_linux_4
+
+        with self.assertRaises(Exception) as context:
+            self.envlayer.get_package_manager()
+        self.assertEqual(str(context.exception), "Azure Linux 4 is not yet supported in your region. Please review aka.ms/LinuxPatchExtension for more information.")
+
+        # restore
+        distro.os_release_attr = self.backup_distro_os_release_attr
+        self.envlayer.platform.linux_distribution = self.backup_linux_distribution
+
+    def test_get_package_manager_rhel_10_not_supported(self):
+        """Test that RHEL 10 raises an exception"""
+        self.backup_linux_distribution = self.envlayer.platform.linux_distribution
+        self.backup_distro_os_release_attr = distro.os_release_attr
+        self.backup_distro_id = distro.id
+
+        platform.system = self.mock_platform_system
+        self.envlayer.platform.linux_distribution = self.mock_linux_distribution
+        distro.os_release_attr = self.mock_distro_os_release_attr_return_rhel_10
+        distro.id = self.mock_distro_id_return_rhel
+
+        with self.assertRaises(Exception) as context:
+            self.envlayer.get_package_manager()
+        self.assertEqual(str(context.exception), "RHEL 10 is not yet supported in your region. Please review aka.ms/LinuxPatchExtension for more information.")
+
+        # restore
+        distro.id = self.backup_distro_id
+        distro.os_release_attr = self.backup_distro_os_release_attr
+        self.envlayer.platform.linux_distribution = self.backup_linux_distribution
+
 
 if __name__ == '__main__':
     unittest.main()