Added missing assertions in widget_attrs test

jrwagz · Jan 28, 2014 · 80e8e5c · 80e8e5c
1 parent 12fca19
commit 80e8e5c
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/autocomplete_light/tests/widgets.py b/autocomplete_light/tests/widgets.py
@@ -36,10 +36,15 @@ def test_init_without_registry(self):
         widget = self.widget_class('FkModelAutocomplete')
         self.assertEqual(widget.autocomplete.model, FkModel)
 
-    def test_widget_data_attributes(self):
+    def test_widget_attrs(self):
         widget = self.widget_class('FkModelAutocomplete',
             widget_attrs={'data-widget-foo': 'bar', 'class':'foobar'})
         html = widget.render('somewidget', None)
+        et = etree.fromstring(html)
+
+        self.assertEquals(et.attrib['data-widget-foo'], 'bar')
+        self.assertIn('foobar', et.attrib['class'])
+        self.assertIn('autocomplete-light-widget', et.attrib['class'])
 
     def test_widget_js_attributes_deprecation(self):
         with self.assertRaises(PendingDeprecationWarning) as context: