class AppTestCase(unittest.TestCase): def setUp(self): self.app = App("blah") def test_get_text_from_element_NodeWithTextOnly(self): e = html.fromstring("<a>abcd</a>") self.assertEquals("abcd", self.app.get_text_from_element(e)) def test_get_text_from_element_NodeWithTextAndOneChild(self): e = html.fromstring("<a>abcd<c1 />efg</a>") self.assertEquals("abcdefg", self.app.get_text_from_element(e)) def test_get_text_from_element_NodeWithTextAndTwoChildren(self): e = html.fromstring("<a>abc<c1/>def</c2>ghi</a>") self.assertEquals("abcdefghi", self.app.get_text_from_element(e)) def test_get_text_from_element_NodeWithTextAndNestedText(self): e = html.fromstring("<a>abc<c1>def</c1></c2>ghi</a>") self.assertEquals("abcdefghi", self.app.get_text_from_element(e)) def test_get_text_from_element_NodeWithTextAndBRElement(self): e = html.fromstring("<a>abc<br />def</a>") self.assertEquals( "abc" + os.linesep + "def", self.app.get_text_from_element(e) ) def test_get_text_from_element_NodeWithTextAndMultiBRElement(self): e = html.fromstring("<a>abc<br /><br />def<br />ghi<br /></a>") self.assertEquals( "abc" + os.linesep + os.linesep + "def" + os.linesep + "ghi" + os.linesep, self.app.get_text_from_element(e) ) def test_get_main_url_and_page_number_PlainURL_PageNumber1(self): url, n = self.app.get_main_url_and_page_number( "http://www.abc.com/product-reviews/" ) self.assertEquals(1, n) self.assertEquals("http://www.abc.com/product-reviews/", url) def test_get_main_url_and_page_number_URLWithNumber2_PageNumber2(self): url, n = self.app.get_main_url_and_page_number( "http://www.abc.com/product-reviews/?pageNumber=2" ) self.assertEquals(2, n) self.assertEquals("http://www.abc.com/product-reviews/", url) def test_get_main_url_and_page_number_URLWithNumber2AndOtherParam_PageNumber2(self): url, n = self.app.get_main_url_and_page_number( "http://www.abc.com/product-reviews/?a=apple&pageNumber=2" ) self.assertEquals(2, n) self.assertEquals("http://www.abc.com/product-reviews/", url)
def setUp(self): self.app = App("blah")