From 4ad0154bcf86a09b4bb839ae6e336425771ad403 Mon Sep 17 00:00:00 2001
From: Florian Schweikert <kelvan@logic.at>
Date: Wed, 7 Dec 2011 22:01:42 +0100
Subject: [PATCH] new unittests

---
 gotovienna/tests/departures.py |   45 ++++++++++++++++++++++++++++++++++++++++
 gotovienna/tests/realtime.py   |    8 +++----
 2 files changed, 49 insertions(+), 4 deletions(-)
 create mode 100644 gotovienna/tests/departures.py

diff --git a/gotovienna/tests/departures.py b/gotovienna/tests/departures.py
new file mode 100644
index 0000000..9da88d1
--- /dev/null
+++ b/gotovienna/tests/departures.py
@@ -0,0 +1,45 @@
+from nose.tools import assert_equal
+import sys
+import os
+from datetime import time, datetime
+
+# bananas for the monkey
+class datetime_static(datetime):
+    @classmethod
+    def now(cls):
+        return datetime(2000, 1, 1, 11, 50)
+
+from gotovienna import realtime
+realtime.datetime = datetime_static
+# </bananas>
+
+sys.path.insert(0, os.path.dirname(os.path.dirname(__file__)))
+from gotovienna.realtime import Departure
+from utils import sort_departures
+
+departures = [Departure('U2', 'Karlsplatz', 'Aspernstrasse', time(13, 37), True),
+              Departure('U1', 'Karlsplatz', 'Reumannplatz', time(12, 24), False),
+              Departure('U4', 'Karlsplatz', 'Huetteldorf', time(11, 42), False),
+              Departure('U4', 'Karlsplatz', 'Heiligenstadt', 5, True),
+              Departure('U2', 'Karlsplatz', 'Aspernstrasse', time(13, 38), True)]
+
+departures_sorted = [Departure('U4', 'Karlsplatz', 'Huetteldorf', time(11, 42), False),
+                     Departure('U4', 'Karlsplatz', 'Heiligenstadt', 5, True),
+                     Departure('U1', 'Karlsplatz', 'Reumannplatz', time(12, 24), False),
+                     Departure('U2', 'Karlsplatz', 'Aspernstrasse', time(13, 37), True),
+                     Departure('U2', 'Karlsplatz', 'Aspernstrasse', time(13, 38), True)]
+
+def test_sort():
+    assert_equal(departures_sorted, sort_departures(departures))
+
+def test_atime():
+    for dep in departures:
+        assert_equal(time, type(dep['atime']))
+
+def test_ftime():
+    for dep in departures:
+        assert_equal(str, type(dep['ftime']))
+
+def test_deltatime():
+    for dep in departures:
+        assert_equal(int, type(dep['deltatime']))
diff --git a/gotovienna/tests/realtime.py b/gotovienna/tests/realtime.py
index d7cf44a..a0c514c 100644
--- a/gotovienna/tests/realtime.py
+++ b/gotovienna/tests/realtime.py
@@ -1,22 +1,22 @@
 import unittest
 import sys
 import os
-sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..'))
+sys.path.insert(0, os.path.dirname(os.path.dirname(__file__)))
 
 from gotovienna.realtime import *
 
 parser = ITipParser()
 
 class ParseTest(unittest.TestCase):
-    
+
     def test_lines(self):
         lines = parser.lines
         self.assertTrue(type(lines) == dict)
         self.assertTrue(lines)
-    
+
     def test_stations(self):
         lines = parser.lines
-        
+
         s = []
         for line in lines:
             s.append(parser.get_stations(line[0]))
-- 
1.7.9.5