Add simple test for fcntl.flock()

tiran · tiran · commit 0a956f1aec60 · 2013-12-05T16:13:03.000+01:00
diff --git a/Lib/test/test_fcntl.py b/Lib/test/test_fcntl.py
@@ -115,6 +115,21 @@ def test_fcntl_64_bit(self):
         finally:
             os.close(fd)
 
+    def test_flock(self):
+        self.f = open(TESTFN, 'wb')
+        fileno = self.f.fileno()
+        fcntl.flock(fileno, fcntl.LOCK_SH)
+        fcntl.flock(fileno, fcntl.LOCK_UN)
+        fcntl.flock(self.f, fcntl.LOCK_SH | fcntl.LOCK_NB)
+        fcntl.flock(self.f, fcntl.LOCK_UN)
+        fcntl.flock(fileno, fcntl.LOCK_EX)
+        fcntl.flock(fileno, fcntl.LOCK_UN)
+
+        self.assertRaises(ValueError, fcntl.flock, -1, fcntl.LOCK_SH)
+        self.assertRaises(TypeError, fcntl.flock, 'spam', fcntl.LOCK_SH)
+        self.assertRaises(OverflowError, fcntl.flock, _testcapi.INT_MAX+1,
+                          fcntl.LOCK_SH)
+
 
 def test_main():
     run_unittest(TestFcntl)
