From cb84351270041449ee8a8984758e7e3ef86ea590 Mon Sep 17 00:00:00 2001
From: nico <nico@magicbroccoli.de>
Date: Thu, 12 Dec 2019 23:15:45 +0100
Subject: test code and small init corrections

+ added pytest test cases
* fixed TSGroupAssigner module init file
---
 TSGroupAssigner/__init__.py |  3 ++-
 setup.py                    |  1 -
 tests/__init__.py           |  0
 tests/test_group_assign.py  | 56 +++++++++++++++++++++++++++++++++++++++++++++
 4 files changed, 58 insertions(+), 2 deletions(-)
 create mode 100644 tests/__init__.py
 create mode 100644 tests/test_group_assign.py

diff --git a/TSGroupAssigner/__init__.py b/TSGroupAssigner/__init__.py
index a41cd05..04fc467 100644
--- a/TSGroupAssigner/__init__.py
+++ b/TSGroupAssigner/__init__.py
@@ -1,3 +1,4 @@
 # -*- coding: utf-8 -*-
 
-from TSGroupAssigner.group_assign import GroupAssigner
+from .group_assign import GroupAssigner
+from .group_assign import DateException
diff --git a/setup.py b/setup.py
index f296015..6136fdc 100644
--- a/setup.py
+++ b/setup.py
@@ -4,7 +4,6 @@ from setuptools import setup, find_packages
 with open("README.md", "r") as fh:
     long_description = fh.read()
 
-
 setup(
     name='TSGroupAssigner',
     version='0.0.1',
diff --git a/tests/__init__.py b/tests/__init__.py
new file mode 100644
index 0000000..e69de29
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()
-- 
cgit v1.2.3-54-g00ecf