test code and small init corrections

+ added pytest test cases
* fixed TSGroupAssigner module init file

1 files changed, 56 insertions, 0 deletions

diff --git a/tests/test_group_assign.py b/tests/test_group_assign.py
new file mode 100644
index 0000000..1f86574
--- /dev/null
+++ b/tests/test_group_assign.py
@@ -0,0 +1,56 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+import datetime as dt
+
+import pytest
+
+from TSGroupAssigner import GroupAssigner, DateException
+
+# sample input
+creds = {
+    'host': 'localhost',
+    'port': 10011,
+    'user': 'serveradmin',
+    'password': '5up3r_53cr37',
+    'sid': 1,
+    'gid': 24
+}
+
+
+class TestGroupAssigner:
+    def test_missing_input(self):
+        """
+        this should fail due to missing arguments
+        """
+        with pytest.raises(TypeError):
+            GroupAssigner().start()
+
+    def test_sleepstart_startdate(self):
+        # startdate is too far in the future sleepstart should produce a DateException
+
+        # start date 3 days in the future
+        startdate = dt.date.today() + dt.timedelta(days=3)
+        duration = dt.timedelta()
+
+        with pytest.raises(DateException):
+            GroupAssigner(date=startdate, delta=duration,**creds).start()
+
+    def test_datecheck_enddate(self):
+        # this should produce a exit code 0 SystemExit as the end date is in the past
+
+        # start date 2 days in the past
+        startdate = dt.date.today() + dt.timedelta(days=-2)
+        duration = dt.timedelta()
+
+        with pytest.raises(SystemExit):
+            GroupAssigner(date=startdate, delta=duration,**creds).start()
+
+    def test_connect_noconnection(self):
+        # connect should fail with ConnectionRefusedError
+
+        # start date is today
+        startdate = dt.date.today()
+        duration = dt.timedelta()
+
+        with pytest.raises(ConnectionRefusedError):
+            GroupAssigner(date=startdate, delta=duration,**creds).start()