def test_simplest_url(self): u = URL('http://w3af.com/foo/bar.txt') self.assertEqual(u.path, '/foo/bar.txt') self.assertEqual(u.scheme, 'http') self.assertEqual(u.get_file_name(), 'bar.txt') self.assertEqual(u.get_extension(), 'txt')
def test_set_filename(self): u = URL('https://w3af.com:443/xyz/def.html') u.set_file_name('abc.pdf') self.assertEqual(u.url_string, 'https://w3af.com/xyz/abc.pdf') self.assertEqual(u.get_file_name(), 'abc.pdf') u = URL('https://w3af.com/xyz/def.html?id=1') u.set_file_name('abc.pdf') self.assertEqual(u.url_string, 'https://w3af.com/xyz/abc.pdf?id=1') u = URL('https://w3af.com/xyz/def.html?file=/etc/passwd') u.set_file_name('abc.pdf') self.assertEqual(u.url_string, 'https://w3af.com/xyz/abc.pdf?file=/etc/passwd') u = URL('https://w3af.com/') u.set_file_name('abc.pdf') self.assertEqual(u.url_string, 'https://w3af.com/abc.pdf')
def test_set_filename(self): u = URL('https://w3af.com:443/xyz/def.html') u.set_file_name( 'abc.pdf' ) self.assertEqual(u.url_string, 'https://w3af.com/xyz/abc.pdf') self.assertEqual(u.get_file_name(), 'abc.pdf') u = URL('https://w3af.com/xyz/def.html?id=1') u.set_file_name( 'abc.pdf' ) self.assertEqual(u.url_string, 'https://w3af.com/xyz/abc.pdf?id=1') u = URL('https://w3af.com/xyz/def.html?file=/etc/passwd') u.set_file_name( 'abc.pdf' ) self.assertEqual(u.url_string, 'https://w3af.com/xyz/abc.pdf?file=/etc/passwd') u = URL('https://w3af.com/') u.set_file_name( 'abc.pdf' ) self.assertEqual(u.url_string, 'https://w3af.com/abc.pdf')