def test_get_element_html_by_class(self):
    html = self.GET_ELEMENT_BY_CLASS_TEST_STRING
    self.assertEqual(get_element_html_by_class('foo', html), html.strip())
    self.assertEqual(get_element_by_class('no-such-class', html), None)
def test_get_element_html_by_class(self):
    html = self.GET_ELEMENT_BY_CLASS_TEST_STRING
    self.assertEqual(get_element_html_by_class('foo', html), html.strip())
    self.assertEqual(get_element_by_class('no-such-class', html), None)