Added unti testing!

mainCode/session.py

@@ -6,7 +6,7 @@ logging.basicConfig(level=logging.DEBUG, format='%(asctime)s:%(levelname)s:%(thr
 
 USED_SESSION_IDS = []
 
-VERSION = "1.1.1"
+VERSION = "1.1.2"
 
 ## Role defs
 HOST = 0
@@ -80,9 +80,8 @@ class enlightSession():
                 self.sessionId = get_random_alphanumeric_string(24)
             logging.info("My session id is " + self.sessionId)
             USED_SESSION_IDS.append(self.sessionId)
-            self.__server__ = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)#, socket.IPPROTO_UDP)
+            self.__server__ = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
             self.__server__.setsockopt(socket.SOL_SOCKET, socket.SO_BROADCAST, 1)
-            ### self.__server__.bind(("", 37020))   # Bind it to a random port # 37020
             self.__activ__ = True
             self.allowJoin = True
             logging.info("Starting server thread")

testing/sessionDiscoveryTesting.py

@@ -1,10 +1,31 @@
-from src import session
+import sys
+sys.path.append('mainCode/../')
+import mainCode.session
 import unittest
+import time
 
 
-class NamesTestCase(unittest.TestCase):
+class SessionTesting(unittest.TestCase):
 
-   def test_first_last_name(self):
-       testSession = session.enlightSession("TestSession", role = session.HOST)
-       testSession.initConnection()
-       self.assertEqual(testSession.getSessionMembers(), [])
+    def setUp(self):
+        self.testSession = mainCode.session.enlightSession("TestSession", role = mainCode.session.HOST)
+        self.testSession.initConnection()
+
+    def test_check_if_members_are_empty(self):
+        time.sleep(1)
+        members = self.testSession.getSessionMembers()
+        self.assertEqual(members, [])
+    
+    def test_if_allow_join_is_okay(self):
+        time.sleep(2)
+        self.assertEqual(self.testSession.allowJoin, True)
+
+    def test_if_var_and_func_return_same(self):
+        time.sleep(3)
+        self.assertEqual(self.testSession.sessionId, self.testSession.getSessionId())
+    
+    def tearDown(self):
+        self.testSession.stopSession()
+
+if __name__ == '__main__':
+   unittest.main()