def test_find(self): find_id = random.randint(1, 499) url = url_object('http://w3af.org/a/b/foobar.php?foo=123') tag_value = createRandAlNum(10) for i in xrange(0, 500): fr = FuzzReq(url, dc={'a': ['1']}) code = 200 if i == find_id: code = 302 res = httpResponse(code, '<html>',{'Content-Type':'text/html'}, url, url) h1 = HistoryItem() h1.request = fr res.setId(i) h1.response = res if i == find_id: h1.toggleMark() h1.updateTag(tag_value) h1.save() h2 = HistoryItem() self.assertEqual(len(h2.find([('tag', "%"+tag_value+"%", 'like')])), 1) self.assertEqual(len(h2.find([('code', 302, '=')])), 1) self.assertEqual(len(h2.find([('mark', 1, '=')])), 1) self.assertEqual(len(h2.find([('has_qs', 1, '=')])), 500) self.assertEqual(len(h2.find([('has_qs', 1, '=')], resultLimit=10)), 10) results = h2.find([('has_qs', 1, '=')], resultLimit=1, orderData=[('id','desc')]) self.assertEqual(results[0].id, 499) search_data = [] search_data.append(('id', find_id + 1, "<")) search_data.append(('id', find_id - 1, ">")) self.assertEqual(len(h2.find(search_data)), 1)
def test_find(self): find_id = random.randint(1, 499) url = URL("http://w3af.org/a/b/foobar.php?foo=123") tag_value = rand_alnum(10) for i in xrange(0, 500): request = HTTPRequest(url, data="a=1") code = 200 if i == find_id: code = 302 hdr = Headers([("Content-Type", "text/html")]) res = HTTPResponse(code, "<html>", hdr, url, url) h1 = HistoryItem() h1.request = request res.set_id(i) h1.response = res if i == find_id: h1.toggle_mark() h1.update_tag(tag_value) h1.save() h2 = HistoryItem() self.assertEqual(len(h2.find([("tag", "%" + tag_value + "%", "like")])), 1) self.assertEqual(len(h2.find([("code", 302, "=")])), 1) self.assertEqual(len(h2.find([("mark", 1, "=")])), 1) self.assertEqual(len(h2.find([("has_qs", 1, "=")])), 500) self.assertEqual(len(h2.find([("has_qs", 1, "=")], result_limit=10)), 10) results = h2.find([("has_qs", 1, "=")], result_limit=1, orderData=[("id", "desc")]) self.assertEqual(results[0].id, 499) search_data = [] search_data.append(("id", find_id + 1, "<")) search_data.append(("id", find_id - 1, ">")) self.assertEqual(len(h2.find(search_data)), 1)