Ejemplo n.º 1
0
    def test_url_param_strip(self):
        u = 'http://test.com/?foo=bar&abc=def&z=1'

        # Non existent parameter
        self.assertEqual(url_strip_query_param(u, 'xxx'), u)

        # Try some variations
        self.assertEqual(url_strip_query_param(u, 'foo'), 'http://test.com/?abc=def&z=1')
        self.assertEqual(url_strip_query_param(u, 'abc'), 'http://test.com/?foo=bar&z=1')
        self.assertEqual(url_strip_query_param(u, 'z'),   'http://test.com/?foo=bar&abc=def')

        # Check corner cases
        u = 'http://test.com/?'
        self.assertEqual(url_strip_query_param(u, 'xxx'), u)
Ejemplo n.º 2
0
 def get_next_url(self, url):
     """
     Generates URLs for the current website
     """
     npage = int(url_get_query_param(url, 'page', '0'))
     url = url_strip_query_param(url, 'page')
     if url.find('?') == -1:
         url += '?'
     return "{0}&page={1}".format(url, npage + 1)