Created a test-directory, colonialized by a single testcase, and a script to run them

run_tests.sh

@@ -0,0 +1,14 @@
+#/bin/bash
+
+# the directory where the code lies in. It will be added to sys.path so
+# tests can import from where normally
+CODE_DIRECTORY=code
+
+# the directory where the tests lie, relative to CODE_DIRECTORY. This
+# value this is relative because i have no idea how to do file-based-
+# string-manipulation in bash and i don't want to find out.
+TEST_DIRECTORY_RELATIVE=../test
+
+cd $CODE_DIRECTORY
+python3 -m unittest discover -s $TEST_DIRECTORY_RELATIVE
+

test/test_ipgenerator.py

@@ -0,0 +1,43 @@
+from ID2TLib.IPGenerator import IPGenerator
+import unittest
+
+class IPGeneratorTestCase(unittest.TestCase):
+	IP_GENERATOR = None
+	IP_SAMPLES = None
+	IP_SAMPLES_NUM = 1000
+	
+	@classmethod
+	def setUpClass(cls):
+		cls.IP_GENERATOR = IPGenerator()
+		cls.IP_SAMPLES = [cls.IP_GENERATOR.random_ip() for _ in range(cls.IP_SAMPLES_NUM)]
+	
+	def test_valid_ips(self):
+		ip = None
+		
+		try:
+			for ip in self.IP_SAMPLES:
+				parts = ip.split(".")
+				self.assertTrue(len(parts) == 4)
+				
+				numbers = [int(i) for i in parts]
+				self.assertTrue(all(n in range(256) for n in numbers))
+		except:
+			self.fail("%s is not a valid IPv4" % ip)
+	
+	def test_generates_localhost_ip(self):
+		self.assertFalse(any(ip.startswith("127.") for ip in self.IP_SAMPLES))
+	
+	def test_generates_private_ip(self):
+		def private_ip(ip):
+			private_starts = ["10.", "192.168."] + ["172.%i." % i for i in range(16, 32)]
+			return any(ip.startswith(start) for start in private_starts)
+		
+		self.assertFalse(any(map(private_ip, self.IP_SAMPLES)))
+	
+	def test_unique_ips(self):
+		self.assertTrue(len(self.IP_SAMPLES) == len(set(self.IP_SAMPLES)))
+	
+	def test_blacklist(self):
+		generator = IPGenerator(blacklist = ["42.0.0.0/8"])
+		self.assertFalse(any(generator.random_ip().startswith("42.") for _ in range(self.IP_SAMPLES_NUM)))
+