[PATCH for f20 4/4] Add tests for file creation labeling functions

[Rastislav Hepner]

getfscreatecon( raw), setfscreatecon(_raw);
---
 tests/test_fileCreation_labeling.py | 129 ++++++++++++++++++++++++++++++++++++
 1 file changed, 129 insertions(+)
 create mode 100755 tests/test_fileCreation_labeling.py

diff --git a/tests/test_fileCreation_labeling.py
b/tests/test_fileCreation_labeling.py
new file mode 100755
index 0000000..276b307
--- /dev/null
+++ b/tests/test_fileCreation_labeling.py
@@ -0,0 +1,129 @@
+#!/usr/bin/env python
+
+#This program is free software: you can redistribute it and/or modify
+#it under the terms of the GNU General Public License as published by
+#the Free Software Foundation, either version 3 of the License, or
+#(at your option) any later version.
+#This program is distributed in the hope that it will be useful,
+#but WITHOUT ANY WARRANTY; without even the implied warranty of
+#MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+#GNU General Public License for more details.
+#
+#For more information see <http://www.gnu.org/licenses/>
+
+"""Tests for:
+getfscreatecon(_raw), setfscreatecon(_con)
+"""
+
+import helper
+import selinux
+import unittest
+import sys
+
+
+class auxiliaryTestCase(unittest.TestCase):
+    """Auxiliary class.
+    """
+
+    def __init__(self, test_method="runTest"):
+        unittest.TestCase.__init__(self, test_method)
+        self.raw_con = "unconfined_u:unconfined_r:unconfined_t:s0"
+        self.trans_con = "unconfined_u:unconfined_r:unconfined_t:SystemLow"
+        self.wrong_con = "WRONG CONTEXT"
+
+    def tearDown(self):
+        """Clears the fscreate file after every test
+        """
+        selinux.setfscreatecon_raw(None)
+
+    def read_fscreate(self, pid="self"):
+        """Reads selinux context from /proc/self/attr/fscreate file.
+
+        Args:
+            pid: PID of process whose /proc/.../fscreate file should be read.
+
+        Returns:
+            None if /proc/.../fscreate contains "".
+            Fscreate selinux context otherwise.
+        """
+        return helper.read_attr_file(pid, "fscreate")
+
+
+class getfscreateconRawTestCase(auxiliaryTestCase):
+    """TestCase for getfscreatecon_raw() funciton.
+    """
+
+    def test_getfscreateconRaw_FscreateIsEmpty_ReturnedNone(self):
+        #Make sure file is empty
+        selinux.setfscreatecon_raw(None)
+        self.assertEqual(self.read_fscreate(), None, "Gathered context from "
+                         "/proc/.../fscreate file should be None!")
+
+    def test_getfscreateconRaw_RawConInFscreate_ReturnsRawCon(self):
+        #Put raw_con into fscreate file first
+        selinux.setfscreatecon_raw(self.raw_con)
+        self.assertEqual(selinux.getfscreatecon_raw()[1],
self.raw_con, "raw_con was "
+                         "not returned as expected")
+
+
+class setfscreateconRawTestCase(auxiliaryTestCase):
+    """TestCase for setfscreatecon_raw() funciton.
+    """
+
+    def test_setfscreateconRaw_InsertRawCon_InsertedSuccessfully(self):
+        selinux.setfscreatecon_raw(self.raw_con)
+        self.assertEqual(self.read_fscreate(), self.raw_con, "raw_con was "
+                         "not put into /proc/.../fscreate file!")
+
+    def test_setfscreateconRaw_RawConRemoval_RawConRemoved(self):
+        selinux.setfscreatecon_raw(self.raw_con)
+        selinux.setfscreatecon_raw(None)
+        self.assertEqual(self.read_fscreate(), None, "Removal "
+                            "of raw_con has failed! setfscreatecon_raw(None) "
+                            "did not work!")
+
+    def test_setfscreateconRaw_WrongContextUsed_OSErrorRaised(self):
+        self.assertRaisesRegexp(OSError, '\[Errno 22\]',
selinux.setfscreatecon_raw,
+                                self.wrong_con)
+
+
+class getfscreateconTestCase(auxiliaryTestCase):
+    """TestCase for getfscreatecon() function.
+    """
+
+    @unittest.skipIf(helper.contextTranslation(), "Context-trans active!")
+    def test_getfscreatecon_RawContextInFscreate_ReturnsRawContext(self):
+        selinux.setfscreatecon_raw(self.raw_con)
+        self.assertEqual(selinux.getfscreatecon()[1], self.raw_con,
"raw_con was "
+                         "not returned as expected!")
+
+    @unittest.skipUnless(helper.contextTranslation(), "Context-trans
inactive!")
+    def test_getfscreatecon_RawContextInFscreate_ReturnsTransContext(self):
+        selinux.setfscreatecon_raw(self.raw_con)
+        self.assertEqual(selinux.getfscreatecon()[1], self.trans_con,
"trans_con was "
+                         "not returned as expected!")
+
+
+class setfscreateconTestCase(auxiliaryTestCase):
+    """TestCase for setfscreatecon() funciton.
+    """
+
+    @unittest.skipUnless(helper.contextTranslation(), "Context-trans
inactive!")
+    def test_setfscreatecon_InsertTransCon_InsertedSuccessfully(self):
+        selinux.setfscreatecon(self.trans_con)
+        self.assertEqual(self.read_fscreate(), self.raw_con, "trans_con was "
+                         "not put into /proc/.../fscreate file!")
+
+    def test_setfscreatecon_InsertRawCon_InsertedSuccessfully(self):
+        selinux.setfscreatecon(self.raw_con)
+        self.assertEqual(self.read_fscreate(), self.raw_con, "raw_con was "
+                         "not put into /proc/.../fscreate file!")
+
+    def test_setfscreateconRaw_WrongContextUsed_OSErrorRaised(self):
+        self.assertRaisesRegexp(OSError, '\[Errno 22\]',
selinux.setfscreatecon,
+                                self.wrong_con)
+
+
+if __name__ == "__main__":
+    suite =
unittest.TestLoader().loadTestsFromModule(sys.modules[auxiliaryTestCase.__module__])
+    unittest.TextTestRunner(verbosity=2).run(suite)
-- 
1.9.0