diff options
Diffstat (limited to 'django')
| -rw-r--r-- | django/contrib/admin/tests.py | 10 |
1 files changed, 9 insertions, 1 deletions
diff --git a/django/contrib/admin/tests.py b/django/contrib/admin/tests.py index eb04b40d15..9ad817d4a7 100644 --- a/django/contrib/admin/tests.py +++ b/django/contrib/admin/tests.py @@ -94,4 +94,12 @@ class AdminSeleniumWebDriverTestCase(LiveServerTestCase): actual_values = [] for option in options: actual_values.append(option.get_attribute('value')) - self.assertEqual(values, actual_values)
\ No newline at end of file + self.assertEqual(values, actual_values) + + def has_css_class(self, selector, klass): + """ + Returns True if the element identified by `selector` has the CSS class + `klass`. + """ + return (self.selenium.find_element_by_css_selector(selector) + .get_attribute('class').find(klass) != -1)
\ No newline at end of file |