Don't import spwd for tests on non Linux OS

saltstack · May 7, 2021 · 5dfb901 · 5dfb901
1 parent 0f777f5
commit 5dfb901
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/tests/unit/modules/test_linux_shadow.py b/tests/unit/modules/test_linux_shadow.py
@@ -2,7 +2,6 @@
     :codeauthor: Erik Johnson <erik@saltstack.com>
 """
 
-import spwd
 import textwrap
 
 import pytest
@@ -12,14 +11,18 @@
 from tests.support.mock import DEFAULT, MagicMock, mock_open, patch
 from tests.support.unit import TestCase, skipIf
 
+try:
+    import spwd
+except ImportError:
+    pass
+
 try:
     import salt.modules.linux_shadow as shadow
 
     HAS_SHADOW = True
 except ImportError:
     HAS_SHADOW = False
 
-
 _PASSWORD = "lamepassword"
 
 # Not testing blowfish as it is not available on most Linux distros