From 785ccbcb7a2381b3c2601608f158c69cc3e58ced Mon Sep 17 00:00:00 2001
From: David Wilson <dw@botanicus.net>
Date: Wed, 28 Feb 2018 14:11:36 +0545
Subject: [PATCH] tests: import very basic latch_test

---
 tests/latch_test.py | 65 +++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 65 insertions(+)
 create mode 100644 tests/latch_test.py

diff --git a/tests/latch_test.py b/tests/latch_test.py
new file mode 100644
index 00000000..2922cd71
--- /dev/null
+++ b/tests/latch_test.py
@@ -0,0 +1,65 @@
+
+import unittest2
+
+import mitogen.core
+
+import testlib
+
+
+class EmptyTest(testlib.TestCase):
+    klass = mitogen.core.Latch
+
+    def test_is_empty(self):
+        latch = self.klass()
+        self.assertTrue(latch.empty())
+
+    def test_is_nonempty(self):
+        latch = self.klass()
+        latch.put(None)
+        self.assertTrue(not latch.empty())
+
+
+class GetTest(testlib.TestCase):
+    klass = mitogen.core.Latch
+    # TODO: test multiple waiters.
+
+    def test_empty_noblock(self):
+        latch = self.klass()
+        exc = self.assertRaises(mitogen.core.TimeoutError,
+            lambda: latch.get(block=False))
+
+    def test_empty_zero_timeout(self):
+        latch = self.klass()
+        exc = self.assertRaises(mitogen.core.TimeoutError,
+            lambda: latch.get(timeout=0))
+
+    def test_nonempty(self):
+        obj = object()
+        latch = self.klass()
+        latch.put(obj)
+        self.assertEquals(obj, latch.get())
+
+    def test_nonempty_noblock(self):
+        obj = object()
+        latch = self.klass()
+        latch.put(obj)
+        self.assertEquals(obj, latch.get(block=False))
+
+    def test_nonempty_zero_timeout(self):
+        obj = object()
+        latch = self.klass()
+        latch.put(obj)
+        self.assertEquals(obj, latch.get(timeout=0))
+
+
+class PutTest(testlib.TestCase):
+    klass = mitogen.core.Latch
+
+    def test_put(self):
+        latch = self.klass()
+        latch.put(None)
+        self.assertEquals(None, latch.get())
+
+
+if __name__ == '__main__':
+    unittest2.main()