Adds a another allocation form helper test

OneGov · Aug 28, 2015 · bd95952 · bd95952
1 parent 9c4e65a
commit bd95952
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/onegov/town/tests/test_forms.py b/onegov/town/tests/test_forms.py
@@ -3,6 +3,7 @@
 from datetime import date, datetime, time
 from dateutil.rrule import MO, WE
 from libres.db.models import Allocation
+from onegov.core.utils import Bunch
 from onegov.town.forms import (
     DaypassAllocationForm,
     ReservationForm,
@@ -187,6 +188,21 @@ def test_generate_datetimes():
     ]
 
 
+def test_combine_datetime():
+    helper = AllocationFormHelpers()
+
+    helper.date = Bunch(data=date(2015, 1, 1))
+    helper.time = Bunch(data=None)
+    assert helper.combine_datetime('date', 'time') == datetime(2015, 1, 1)
+
+    helper.time = Bunch(data='12:00')
+    assert helper.combine_datetime('date', 'time') == datetime(2015, 1, 1, 12)
+
+    helper.date = Bunch(data=None)
+    helper.time = Bunch(data=None)
+    assert helper.combine_datetime('date', 'time') is None
+
+
 def test_reservation_form_partly_available():
     allocation = Allocation()