def test_has_price_tag_not_number(self): tag_mock = Mock() tag_mock.string = "$Test String" self.link_html.find = Mock(return_value=tag_mock) price = listing.extract_price( self.link_html, self.page_html, self.unhidden_page_html) self.assertIsNone(price)
def test_has_price_tag(self): tag_mock = Mock() tag_mock.string = "$345" self.link_html.find = Mock(return_value=tag_mock) price = listing.extract_price( self.link_html, self.page_html, self.unhidden_page_html) self.assertEqual(price, 345)
def test_has_no_price(self): self.link_html.find = Mock(return_value=None) self.link_html.text = "Not a price" price = listing.extract_price( self.link_html, self.page_html, self.unhidden_page_html) self.assertIsNone(price)
def test_has_price_in_link(self): self.link_html.find = Mock(return_value=None) self.link_html.text = "Rent is $345 a month" price = listing.extract_price( self.link_html, self.page_html, self.unhidden_page_html) self.assertEqual(price, 345)