class TestHistory(unittest.TestCase): @mock_urls def setUp(self): self.browser = RoboBrowser(history=True) self.browser.open('http://robobrowser.com/page1/') self.browser.open('http://robobrowser.com/page2/') self.browser.open('http://robobrowser.com/page3/') def test_back(self): self.browser.back() assert_equal(self.browser.url, 'http://robobrowser.com/page2/') def test_back_n(self): self.browser.back(n=2) assert_equal(self.browser.url, 'http://robobrowser.com/page1/') def test_forward(self): self.browser.back() self.browser.forward() assert_equal(self.browser.url, 'http://robobrowser.com/page3/') def test_forward_n(self): self.browser.back(n=2) self.browser.forward(n=2) assert_equal(self.browser.url, 'http://robobrowser.com/page3/') @mock_urls def test_open_clears_forward(self): self.browser.back(n=2) self.browser.open('http://robobrowser.com/page4/') assert_equal(self.browser._cursor, len(self.browser._states) - 1) assert_raises(exceptions.RoboError, self.browser.forward) def test_back_error(self): assert_raises(exceptions.RoboError, self.browser.back, 5)
class TestHistory(unittest.TestCase): @mock_urls def setUp(self): self.browser = RoboBrowser(history=True) self.browser.open('http://robobrowser.com/page1/') self.browser.open('http://robobrowser.com/page2/') self.browser.open('http://robobrowser.com/page3/') def test_back(self): self.browser.back() assert_equal( self.browser.url, 'http://robobrowser.com/page2/' ) def test_back_n(self): self.browser.back(n=2) assert_equal( self.browser.url, 'http://robobrowser.com/page1/' ) def test_forward(self): self.browser.back() self.browser.forward() assert_equal( self.browser.url, 'http://robobrowser.com/page3/' ) def test_forward_n(self): self.browser.back(n=2) self.browser.forward(n=2) assert_equal( self.browser.url, 'http://robobrowser.com/page3/' ) @mock_urls def test_open_clears_forward(self): self.browser.back(n=2) self.browser.open('http://robobrowser.com/page4/') assert_equal( self.browser._cursor, len(self.browser._states) - 1 ) assert_raises( exceptions.RoboError, self.browser.forward ) def test_back_error(self): assert_raises( exceptions.RoboError, self.browser.back, 5 )