def test_add_httpPostDataRequest(self):
ds = disk_set()
uri = url_object('http://w3af.org/?id=2')
pdr1 = httpPostDataRequest(uri, method='GET', headers={'Referer': 'http://w3af.org/'})
uri = url_object('http://w3af.org/?id=3')
pdr2 = httpPostDataRequest(uri, method='GET', headers={'Referer': 'http://w3af.com/'})
uri = url_object('http://w3af.org/?id=7')
pdr3 = httpPostDataRequest(uri, method='FOO', headers={'Referer': 'http://w3af.com/'})
ds.add( pdr1 )
ds.add( pdr2 )
ds.add( pdr2 )
ds.add( pdr1 )
self.assertEqual( ds[0] , pdr1)
self.assertEqual( ds[1] , pdr2)
self.assertFalse( pdr3 in ds )
self.assertTrue( pdr2 in ds )
self.assertEqual( len(ds) , 2)
# This forces an internal change in the URL object
pdr2.getURL().url_string
self.assertTrue( pdr2 in ds )
def __init__(self):
baseDiscoveryPlugin.__init__(self)
# Internal variables
self._compiled_ignore_re = None
self._compiled_follow_re = None
self._broken_links = disk_set()
self._fuzzable_reqs = disk_set()
self._first_run = True
self._known_variants = variant_db()
self._already_filled_form = scalable_bloomfilter()
# User configured variables
self._ignore_regex = ''
self._follow_regex = '.*'
self._only_forward = False
self._compileRE()
def test_add(self):
ds = disk_set()
ds.add(1)
ds.add(2)
ds.add(3)
ds.add(1)
self.assertEqual( list(ds), [1,2,3])
def test_add_urlobject(self):
ds = disk_set()
ds.add( url_object('http://w3af.org/?id=2') )
ds.add( url_object('http://w3af.org/?id=3') )
ds.add( url_object('http://w3af.org/?id=3') )
self.assertEqual( ds[0] , url_object('http://w3af.org/?id=2'))
self.assertEqual( ds[1] , url_object('http://w3af.org/?id=3'))
self.assertEqual( len(ds) , 2)
self.assertFalse( url_object('http://w3af.org/?id=4') in ds )
self.assertTrue( url_object('http://w3af.org/?id=2') in ds )
def test_update(self):
ds = disk_set()
ds.add(1)
ds.update([2,3,1])
self.assertEqual( list(ds), [1,2,3])
