Updated tests for is_aarch64 returns

saltstack · Aug 9, 2024 · 6e021e2 · 6e021e2
1 parent 575f523
commit 6e021e2
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/changelog/36.improvement.rst b/changelog/36.improvement.rst
@@ -1 +1 @@
-Added support for is_arm64 for MacOS and is_x86_64, and updated pre-commit, drop support for Python 3.7
+Updated is_aarch64 to allow for MacOS Arm64 using 'arm64' instead of 'aarch64', updated pre-commit, drop support for Python 3.7
diff --git a/tests/unit/utils/test_platform.py b/tests/unit/utils/test_platform.py
@@ -144,13 +144,21 @@ def test_is_not_aix():
 
 def test_is_aarch64():
     return_value = True
-    with mock.patch("platform.machine", return_value="aarch64"):
+    # Allow for MacOS Arm64 platform returns 'arm64' not 'aarch64', different than Linux
+    with mock.patch("platform.machine", return_value="arm64"):
         assert pytestskipmarkers.utils.platform.is_aarch64() is return_value
 
 
 def test_is_not_aarch64():
     return_value = False
-    with mock.patch("platform.machine", return_value="not_aarch64"):
+    with mock.patch("platform.machine", return_value="not_arm64"):
+        assert pytestskipmarkers.utils.platform.is_aarch64() is return_value
+
+
+def test_is_not_aarch64_string_aarch64():
+    return_value = False
+    # Allow for MacOS Arm64 platform returning differently from Linux
+    with mock.patch("platform.machine", return_value="aarch64"):
         assert pytestskipmarkers.utils.platform.is_aarch64() is return_value