test: add unit tests

Tests written using unittest [1] and hypothesis [2] modules. We use
hypothesis v4.57.0 because it is a latest version with Python 2.7
support [3].

How-to run:

$ make test_unittest

1. https://docs.python.org/3/library/unittest.html
2. https://hypothesis.readthedocs.io/en/latest/
3. https://hypothesis.readthedocs.io/en/latest/changes.html#v4-57-0

Author: ligurio

.gitignore

@@ -41,6 +41,7 @@ htmlcov/
 .coverage
 .coverage.*
 .cache
+.hypothesis/
 nosetests.xml
 coverage.xml
 *,cover

Makefile

@@ -15,6 +15,9 @@ luacheck:
 test_integration:
 	$(PYTHON) test/test-run.py --force $(TEST_RUN_EXTRA_PARAMS)
 
-test: test_integration
+test_unittest:
+	$(PYTHON) -m unittest discover test/unittest/
 
-.PHONY: lint flake8 luacheck test test_integration
+test: test_unittest test_integration
+
+.PHONY: lint flake8 luacheck test test_integration test_unittest

requirements-test.txt

@@ -1 +1,2 @@
 flake8==3.7.9
+hypothesis==4.*

test/unittest/00000000000000000003.snap

@@ -0,0 +1,32 @@
+import socket
+import unittest
+
+from hypothesis import given, settings
+from hypothesis.strategies import integers
+
+import lib.utils as utils
+
+class TestUtils(unittest.TestCase):
+    def test_extract_schema_from_snapshot(self):
+        snapshot_path = 'test/unittest/00000000000000000003.snap'
+        v = utils.extract_schema_from_snapshot(snapshot_path)
+        self.assertEqual(v, (2, 3, 1))
+
+    @settings(max_examples=5)
+    @given(port=integers(65100, 65535))
+    def test_check_port(self, port):
+        def open_socket(p):
+            s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+            s.bind(('localhost', p))
+            s.listen(0)
+            return s
+        status = utils.check_port(port, rais=False, ipv4=True, ipv6=False)
+        self.assertEqual(status, True)
+        s = open_socket(port)
+        status = utils.check_port(port, rais=False, ipv4=True, ipv6=False)
+        s.close()
+        self.assertEqual(status, False)
+
+
+if __name__ == "__main__":
+    unittest.main()