Fixed #5851 -- Allowed specifying different HTML attrs for SplitDateTimeWidget subwidgets.

Thanks Tim Graham and Nick Pope for review.

2 files changed, 24 insertions, 0 deletions

diff --git a/tests/forms_tests/widget_tests/test_splitdatetimewidget.py b/tests/forms_tests/widget_tests/test_splitdatetimewidget.py
index 172bcbbe8d..3b06fd5958 100644
--- a/tests/forms_tests/widget_tests/test_splitdatetimewidget.py
+++ b/tests/forms_tests/widget_tests/test_splitdatetimewidget.py
@@ -37,6 +37,18 @@ class SplitDateTimeWidgetTest(WidgetTest):
             '<input type="text" class="pretty" value="07:30:00" name="date_1" />'
         ))
 
+    def test_constructor_different_attrs(self):
+        html = (
+            '<input type="text" class="foo" value="2006-01-10" name="date_0" />'
+            '<input type="text" class="bar" value="07:30:00" name="date_1" />'
+        )
+        widget = SplitDateTimeWidget(date_attrs={'class': 'foo'}, time_attrs={'class': 'bar'})
+        self.check_html(widget, 'date', datetime(2006, 1, 10, 7, 30), html=html)
+        widget = SplitDateTimeWidget(date_attrs={'class': 'foo'}, attrs={'class': 'bar'})
+        self.check_html(widget, 'date', datetime(2006, 1, 10, 7, 30), html=html)
+        widget = SplitDateTimeWidget(time_attrs={'class': 'bar'}, attrs={'class': 'foo'})
+        self.check_html(widget, 'date', datetime(2006, 1, 10, 7, 30), html=html)
+
     def test_formatting(self):
         """
         Use 'date_format' and 'time_format' to change the way a value is
diff --git a/tests/forms_tests/widget_tests/test_splithiddendatetimewidget.py b/tests/forms_tests/widget_tests/test_splithiddendatetimewidget.py
index 07ee15690c..e3adb11520 100644
--- a/tests/forms_tests/widget_tests/test_splithiddendatetimewidget.py
+++ b/tests/forms_tests/widget_tests/test_splithiddendatetimewidget.py
@@ -40,3 +40,15 @@ class SplitHiddenDateTimeWidgetTest(WidgetTest):
             <input type="hidden" name="date_1" value="12:51:00" />
             """
         ))
+
+    def test_constructor_different_attrs(self):
+        html = (
+            '<input type="hidden" class="foo" value="2006-01-10" name="date_0" />'
+            '<input type="hidden" class="bar" value="07:30:00" name="date_1" />'
+        )
+        widget = SplitHiddenDateTimeWidget(date_attrs={'class': 'foo'}, time_attrs={'class': 'bar'})
+        self.check_html(widget, 'date', datetime(2006, 1, 10, 7, 30), html=html)
+        widget = SplitHiddenDateTimeWidget(date_attrs={'class': 'foo'}, attrs={'class': 'bar'})
+        self.check_html(widget, 'date', datetime(2006, 1, 10, 7, 30), html=html)
+        widget = SplitHiddenDateTimeWidget(time_attrs={'class': 'bar'}, attrs={'class': 'foo'})
+        self.check_html(widget, 'date', datetime(2006, 1, 10, 7, 30), html=html)