def __init__(self): GrepPlugin.__init__(self) self._total_count = 0 self._vuln_count = 0 self._vulns = DiskList() self._ids = DiskList()
def __init__(self): ''' Class init ''' GrepPlugin.__init__(self) self._total_count = 0 self._vulns = DiskList() self._urls = DiskList()
def test_thread_safe(self): dl = DiskList() def worker(range_inst): for i in range_inst: dl.append(i) threads = [] _min = 0 for _max in xrange(0, 1100, 100): th = threading.Thread(target=worker, args=(xrange(_min, _max), )) threads.append(th) _min = _max for th in threads: th.start() for th in threads: th.join() for i in xrange(0, 1000): self.assertTrue(i in dl, i) dl_as_list = list(dl) self.assertEqual(len(dl_as_list), len(set(dl_as_list))) dl_as_list.sort() self.assertEqual(dl_as_list, range(1000))
def test_fuzzable_request(self): dl = DiskList() uri = URL('http://w3af.org/?id=2') qsr1 = HTTPQSRequest(uri, method='GET', headers=Headers([('Referer', 'http://w3af.org/') ])) uri = URL('http://w3af.org/?id=3') qsr2 = HTTPQSRequest(uri, method='OPTIONS', headers=Headers([('Referer', 'http://w3af.org/') ])) uri = URL('http://w3af.org/?id=7') qsr3 = HTTPQSRequest(uri, method='FOO', headers=Headers([('Referer', 'http://w3af.org/') ])) dl.append(qsr1) dl.append(qsr2) self.assertEqual(dl[0], qsr1) self.assertEqual(dl[1], qsr2) self.assertFalse(qsr3 in dl) self.assertTrue(qsr2 in dl)
def test_len(self): dl = DiskList() for i in xrange(0, 100): _ = dl.append(i) self.assertEqual(len(dl) == 100, True)
def __init__(self, scroll_bar, active_filter, possible): ''' :param scroll_bar: Gtk Vertical Scrollbar object :param active_filter: the filter active at startup. :param possible: all filter keys ''' gtk.TextView.__init__(self) MessageConsumer.__init__(self) self.set_editable(False) self.set_cursor_visible(False) self.set_wrap_mode(gtk.WRAP_WORD) self.textbuffer = self.get_buffer() self.show() self.possible = set(possible) self.active_filter = active_filter self.text_position = 0 self.all_messages = DiskList() # scroll bar self.freeze_scrollbar = False scroll_bar.connect("value-changed", self.scroll_changed) # colors self.textbuffer.create_tag("red-fg", foreground="red") self.textbuffer.create_tag("blue-fg", foreground="blue") self.textbuffer.create_tag("brown-fg", foreground="brown") self.bg_colors = { "vulnerability": "red-fg", "information": "blue-fg", "error": "brown-fg", }
def __init__(self): AuditPlugin.__init__(self) # Internal variables self._potential_vulns = DiskList() # User configured variables self._diff_ratio = 0.30
def test_slice_all(self): disk_list = DiskList() disk_list.append('1') disk_list.append('2') dl_copy = disk_list[:] self.assertIn('1', dl_copy) self.assertIn('2', dl_copy)
def test_many_instances(self): all_instances = [] amount = 200 for _ in xrange(amount): disk_list = DiskList() all_instances.append(disk_list) self.assertEqual(len(all_instances), amount)
def __init__(self): AuditPlugin.__init__(self) # Internal variables self._expected_res_mutant = DiskDict() self._freq_list = DiskList() re_str = '<!--#exec cmd="echo -n (.*?);echo -n (.*?)" -->' self._extract_results_re = re.compile(re_str)
def test_slice_first_N(self): disk_list = DiskList() disk_list.append('1') disk_list.append('2') disk_list.append('3') dl_copy = disk_list[:1] self.assertIn('1', dl_copy) self.assertNotIn('2', dl_copy) self.assertNotIn('3', dl_copy)
def test_urlobject(self): dl = DiskList() dl.append(URL('http://w3af.org/?id=2')) dl.append(URL('http://w3af.org/?id=3')) self.assertEqual(dl[0], URL('http://w3af.org/?id=2')) self.assertEqual(dl[1], URL('http://w3af.org/?id=3')) self.assertFalse(URL('http://w3af.org/?id=4') in dl) self.assertTrue(URL('http://w3af.org/?id=2') in dl)
def test_remove_table(self): disk_list = DiskList() table_name = disk_list.table_name db = get_default_temp_db_instance() self.assertTrue(db.table_exists(table_name)) disk_list.cleanup() self.assertFalse(db.table_exists(table_name))
def test_sorted(self): dl = DiskList() dl.append('abc') dl.append('def') dl.append('aaa') sorted_dl = sorted(dl) self.assertEqual(['aaa', 'abc', 'def'], sorted_dl)
def __init__(self): GrepPlugin.__init__(self) # Internal variables self._already_added = DiskList() # Compile all regular expressions and store information to avoid # multiple queries to the same function self._common_directories = get_common_directories() self._compiled_regexes = {} self._compile_regex()
def test_getitem(self): dl = DiskList() dl.append('a') dl.append(1) dl.append([3, 2, 1]) self.assertEqual(dl[0] == 'a', True) self.assertEqual(dl[1] == 1, True) self.assertEqual(dl[2] == [3, 2, 1], True) self.assertRaises(IndexError, dl.__getitem__, 3)
def test_islice(self): disk_list = DiskList() disk_list.extend('ABCDEFG') EXPECTED = 'CDEFG' result = '' for c in itertools.islice(disk_list, 2, None, None): result += c self.assertEqual(EXPECTED, result)
def test_getitem_negative(self): dl = DiskList() dl.append('a') dl.append('b') dl.append('c') self.assertEqual(dl[-1], 'c') self.assertEqual(dl[-2], 'b') self.assertEqual(dl[-3], 'a') self.assertRaises(IndexError, dl.__getitem__, -4)
def test_reverse_iteration(self): dl = DiskList() dl.append(1) dl.append(2) dl.append(3) reverse_iter_res = [] for i in reversed(dl): reverse_iter_res.append(i) self.assertEqual(reverse_iter_res, [3, 2, 1])
def test_extend(self): dl = DiskList() dl.append('a') dl.extend([1, 2, 3]) self.assertEqual(len(dl), 4) self.assertEqual(dl[0], 'a') self.assertEqual(dl[1], 1) self.assertEqual(dl[2], 2) self.assertEqual(dl[3], 3)
def test_clear(self): dl = DiskList() dl.append('a') dl.append('b') self.assertEqual(len(dl), 2) dl.clear() self.assertEqual(len(dl), 0)
def test_ordered_iter(self): dl = DiskList() dl.append('abc') dl.append('def') dl.append('aaa') sorted_dl = [] for i in dl.ordered_iter(): sorted_dl.append(i) self.assertEqual(['aaa', 'abc', 'def'], sorted_dl)
def test_string(self): dl = DiskList() for i in xrange(0, 1000): rnd = ''.join(random.choice(string.letters) for i in xrange(40)) _ = dl.append(rnd) self.assertEqual(rnd in dl, True) for i in string.letters: self.assertNotIn(i, dl) self.assertIn(rnd, dl)
def test_int(self): dl = DiskList() for i in xrange(0, 1000): _ = dl.append(i) for i in xrange(0, 1000 / 2): r = random.randint(0, 1000 - 1) self.assertEqual(r in dl, True) for i in xrange(0, 1000 / 2): r = random.randint(1000, 1000 * 2) self.assertEqual(r in dl, False)
def test_pickle(self): dl = DiskList() dl.append('a') dl.append(1) dl.append([3, 2, 1]) values = [] for i in dl: values.append(i) self.assertEqual(values[0] == 'a', True) self.assertEqual(values[1] == 1, True) self.assertEqual(values[2] == [3, 2, 1], True)
def test_unicode(self): dl = DiskList() dl.append(u'à') dl.append(u'המלצת השבוע') dl.append([ u'à', ]) self.assertEqual(dl[0], u'à') self.assertEqual(dl[1], u'המלצת השבוע') self.assertEqual(dl[2], [ u'à', ])
def __init__(self): OutputPlugin.__init__(self) # Internal variables self._initialized = False self._style_output_file = os.path.join('plugins', 'output', 'html_file', 'style.css') # These attributes hold the file pointers self._file = None self._aditional_info = DiskList() # User configured parameters self._verbose = False self._output_file_name = '~/report.html'
def __init__(self, w3af): gtk.DrawingArea.__init__(self) MessageConsumer.__init__(self) self.w3af = w3af self.pangolayout = self.create_pango_layout("") # store all messages to be able to redraw self.all_messages = DiskList() self._need_redraw = 0 # control variables self.alreadyStopped = False self.timeGrouping = 2 self.timeBase = int(time.time() * 1000) self.realLeftMargin = MIZQ self.gc = None self._redraw_gen = None # Go live! self.connect("expose-event", self.area_expose_cb) gobject.timeout_add(500, self.draw_handler) self.show()
def test_not(self): dl = DiskList() self.assertFalse(dl)