Add test module for rbackup.system

tests/test_system.py

@@ -0,0 +1,43 @@
+"""
+.. moduleauthor:: Eric Torres
+
+Tests for the rbackup.config module.
+"""
+import os
+import unittest
+
+import rbackup.system as system
+
+# ========== Constants ==========
+TESTING_PACKAGE = "rbackup"
+TESTING_MODULE = f"{TESTING_PACKAGE}.system"
+
+
+# ========== Tests ==========
+class TestUmask(unittest.TestCase):
+    @staticmethod
+    def get_current_umask():
+        """Obtain process umask, and then change it back for testing."""
+        orig_umask = os.umask(0)
+        os.umask(orig_umask)
+
+        return orig_umask
+
+    def setUp(self):
+        self.test_umask = 0o0027
+        self.orig_umask = self.get_current_umask()
+
+    def test_original_umask_remains(self):
+        # noinspection PyBroadException
+        try:
+            with system.change_umask(self.test_umask):
+                self.assertEqual(self.get_current_umask(), self.test_umask)
+
+                raise BaseException
+        except:
+            pass
+
+        self.assertEqual(self.get_current_umask(), self.orig_umask)
+
+    def tearDown(self):
+        os.umask(self.orig_umask)