test: Wrap optional library tests with skiptest.

google · Jun 3, 2023 · 60b8886 · 60b8886
1 parent 4f9d297
commit 60b8886
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/nss_cache/sources/gcssource_test.py b/nss_cache/sources/gcssource_test.py
@@ -8,10 +8,14 @@
 from nss_cache.maps import group
 from nss_cache.maps import passwd
 from nss_cache.maps import shadow
-from nss_cache.sources import gcssource
 from nss_cache.util import file_formats
 from nss_cache.util import timestamps
 
+try:
+    from nss_cache.sources import gcssource
+except Exception as e:
+    raise unittest.SkipTest("`gcssource` unabled to be imported: {}".format(e))
+
 
 class TestGcsSource(unittest.TestCase):
     def setUp(self):

diff --git a/nss_cache/sources/s3source_test.py b/nss_cache/sources/s3source_test.py
@@ -8,7 +8,11 @@
 from nss_cache.maps import group
 from nss_cache.maps import passwd
 from nss_cache.maps import shadow
-from nss_cache.sources import s3source
+
+try:
+    from nss_cache.sources import s3source
+except Exception as e:
+    raise unittest.SkipTest("s3source unable to be imported: {}".format(e))
 
 
 class TestS3Source(unittest.TestCase):