def test_single_char(self): buf = StringIO() buf.write("TESTING SOME SEARCHING!\n" * 10000) filename = self._create_temp_file(buf.getvalue()) searcher = ReverseFileSearcher(filename, "C") found_positions = searcher.find_all() self.assertEqual(len(found_positions), 10000)
def setUp(self): self.needle = "THENEEDLE" buf = StringIO() buf.write("TESTING SOME SEARCHING!\n" * 10000) self.positions = [1000, 2000, 2500, 3700, 7034, 8343] for pos in self.positions: buf.seek(pos) buf.write(self.needle) self.filename = self._create_temp_file(buf.getvalue()) self.searcher = ReverseFileSearcher(self.filename, self.needle)
def test_needle_on_chunk_border(self): buf = StringIO() buf.write("TESTING SOME SEARCHING!\n" * 10000) buf.seek(-ReverseFileSearcher.DEFAULT_CHUNK_SIZE, os.SEEK_END) buf.write(self.needle) filename = self._create_temp_file(buf.getvalue()) searcher = ReverseFileSearcher(filename, self.needle) found_positions = searcher.find_all() self.assertEqual(len(found_positions), 1)
def test_lots_of_needles_in_same_chunk(self): buf = StringIO() buf.write("TESTING SOME SEARCHING!\n" * 10000) buf.seek(-(ReverseFileSearcher.DEFAULT_CHUNK_SIZE + 100), os.SEEK_END) for _ in xrange(20): buf.write(self.needle) filename = self._create_temp_file(buf.getvalue()) searcher = ReverseFileSearcher(filename, self.needle) found_positions = searcher.find_all() self.assertEqual(len(found_positions), 20)
def test_unicode(self): encoding = "utf-8" needle = u"Åter till testerna, så låt oss nu testa lite".encode(encoding) buf = StringIO() data = (u"Det är nog bra att ha några konstiga bokstäver här med.\n" * 10000).encode(encoding) buf.write(data) positions = [1000, 2000, 2500, 3700, 7034, 8343] for pos in positions: buf.seek(pos) buf.write(needle) filename = self._create_temp_file(buf.getvalue()) searcher = ReverseFileSearcher(filename, needle) self.assertEqual(searcher.find_all(), tuple(reversed(self.positions)))
def test_unicode(self): encoding = "utf-8" needle = u"Åter till testerna, så låt oss nu testa lite".encode( encoding) buf = StringIO() data = (u"Det är nog bra att ha några konstiga bokstäver här med.\n" * 10000).encode(encoding) buf.write(data) positions = [1000, 2000, 2500, 3700, 7034, 8343] for pos in positions: buf.seek(pos) buf.write(needle) filename = self._create_temp_file(buf.getvalue()) searcher = ReverseFileSearcher(filename, needle) self.assertEqual(searcher.find_all(), tuple(reversed(self.positions)))
class TestFileSearcher(unittest2.TestCase): def _create_temp_file(self, buf): _, filename = tempfile.mkstemp("log") with open(filename, "w") as f: f.write(buf) f.flush() return filename def setUp(self): self.needle = "THENEEDLE" buf = StringIO() buf.write("TESTING SOME SEARCHING!\n" * 10000) self.positions = [1000, 2000, 2500, 3700, 7034, 8343] for pos in self.positions: buf.seek(pos) buf.write(self.needle) self.filename = self._create_temp_file(buf.getvalue()) self.searcher = ReverseFileSearcher(self.filename, self.needle) def test_construction(self): self.assertEqual(self.searcher._filename, self.filename) self.assertEqual(self.searcher._needle, self.needle) def test_find_all(self): positions = self.searcher.find_all() self.assertEqual(positions, tuple(reversed(self.positions))) def test_find_one(self): pos = self.searcher.find() self.assertEqual(pos, self.positions[-1]) def test_unicode(self): encoding = "utf-8" needle = u"Åter till testerna, så låt oss nu testa lite".encode( encoding) buf = StringIO() data = (u"Det är nog bra att ha några konstiga bokstäver här med.\n" * 10000).encode(encoding) buf.write(data) positions = [1000, 2000, 2500, 3700, 7034, 8343] for pos in positions: buf.seek(pos) buf.write(needle) filename = self._create_temp_file(buf.getvalue()) searcher = ReverseFileSearcher(filename, needle) self.assertEqual(searcher.find_all(), tuple(reversed(self.positions))) def test_needle_split_by_chunk(self): buf = StringIO() buf.write("TESTING SOME SEARCHING!\n" * 10000) buf.seek(-(ReverseFileSearcher.DEFAULT_CHUNK_SIZE + 5), os.SEEK_END) buf.write(self.needle) filename = self._create_temp_file(buf.getvalue()) searcher = ReverseFileSearcher(filename, self.needle) found_positions = searcher.find_all() self.assertEqual(len(found_positions), 1) def test_needle_on_chunk_border(self): buf = StringIO() buf.write("TESTING SOME SEARCHING!\n" * 10000) buf.seek(-ReverseFileSearcher.DEFAULT_CHUNK_SIZE, os.SEEK_END) buf.write(self.needle) filename = self._create_temp_file(buf.getvalue()) searcher = ReverseFileSearcher(filename, self.needle) found_positions = searcher.find_all() self.assertEqual(len(found_positions), 1) def test_needle_on_chunk_border_does_not_hide_occurence(self): buf = StringIO() buf.write("TESTING SOME SEARCHING!\n" * 10000) buf.seek(-(ReverseFileSearcher.DEFAULT_CHUNK_SIZE + 100), os.SEEK_END) buf.write(self.needle) buf.seek(-ReverseFileSearcher.DEFAULT_CHUNK_SIZE, os.SEEK_END) buf.write(self.needle) filename = self._create_temp_file(buf.getvalue()) searcher = ReverseFileSearcher(filename, self.needle) found_positions = searcher.find_all() self.assertEqual(len(found_positions), 2) def test_lots_of_needles_in_same_chunk(self): buf = StringIO() buf.write("TESTING SOME SEARCHING!\n" * 10000) buf.seek(-(ReverseFileSearcher.DEFAULT_CHUNK_SIZE + 100), os.SEEK_END) for _ in xrange(20): buf.write(self.needle) filename = self._create_temp_file(buf.getvalue()) searcher = ReverseFileSearcher(filename, self.needle) found_positions = searcher.find_all() self.assertEqual(len(found_positions), 20) def test_single_chunk(self): buf = StringIO() buf.write("TESTING SOME SEARCHING!\n" * 100) for _ in xrange(20): buf.write(self.needle) filename = self._create_temp_file(buf.getvalue()) searcher = ReverseFileSearcher(filename, self.needle) found_positions = searcher.find_all() self.assertEqual(len(found_positions), 20) def test_single_char(self): buf = StringIO() buf.write("TESTING SOME SEARCHING!\n" * 10000) filename = self._create_temp_file(buf.getvalue()) searcher = ReverseFileSearcher(filename, "C") found_positions = searcher.find_all() self.assertEqual(len(found_positions), 10000) def test_empty_needle(self): buf = StringIO() buf.write("TESTING SOME SEARCHING!\n" * 10000) filename = self._create_temp_file(buf.getvalue()) self.assertRaises(ValueError, ReverseFileSearcher, filename, "") def test_needle_larger_than_chunk_size(self): buf = StringIO() buf.write("TESTING SOME SEARCHING!\n" * 10000) needle = "NEEDLE" * ReverseFileSearcher.DEFAULT_CHUNK_SIZE filename = self._create_temp_file(buf.getvalue()) self.assertRaises(ValueError, ReverseFileSearcher, filename, needle)
class TestFileSearcher(unittest2.TestCase): def _create_temp_file(self, buf): _, filename = tempfile.mkstemp("log") with open(filename, "w") as f: f.write(buf) f.flush() return filename def setUp(self): self.needle = "THENEEDLE" buf = StringIO() buf.write("TESTING SOME SEARCHING!\n" * 10000) self.positions = [1000, 2000, 2500, 3700, 7034, 8343] for pos in self.positions: buf.seek(pos) buf.write(self.needle) self.filename = self._create_temp_file(buf.getvalue()) self.searcher = ReverseFileSearcher(self.filename, self.needle) def test_construction(self): self.assertEqual(self.searcher._filename, self.filename) self.assertEqual(self.searcher._needle, self.needle) def test_find_all(self): positions = self.searcher.find_all() self.assertEqual(positions, tuple(reversed(self.positions))) def test_find_one(self): pos = self.searcher.find() self.assertEqual(pos, self.positions[-1]) def test_unicode(self): encoding = "utf-8" needle = u"Åter till testerna, så låt oss nu testa lite".encode(encoding) buf = StringIO() data = (u"Det är nog bra att ha några konstiga bokstäver här med.\n" * 10000).encode(encoding) buf.write(data) positions = [1000, 2000, 2500, 3700, 7034, 8343] for pos in positions: buf.seek(pos) buf.write(needle) filename = self._create_temp_file(buf.getvalue()) searcher = ReverseFileSearcher(filename, needle) self.assertEqual(searcher.find_all(), tuple(reversed(self.positions))) def test_needle_split_by_chunk(self): buf = StringIO() buf.write("TESTING SOME SEARCHING!\n" * 10000) buf.seek(-(ReverseFileSearcher.DEFAULT_CHUNK_SIZE + 5), os.SEEK_END) buf.write(self.needle) filename = self._create_temp_file(buf.getvalue()) searcher = ReverseFileSearcher(filename, self.needle) found_positions = searcher.find_all() self.assertEqual(len(found_positions), 1) def test_needle_on_chunk_border(self): buf = StringIO() buf.write("TESTING SOME SEARCHING!\n" * 10000) buf.seek(-ReverseFileSearcher.DEFAULT_CHUNK_SIZE, os.SEEK_END) buf.write(self.needle) filename = self._create_temp_file(buf.getvalue()) searcher = ReverseFileSearcher(filename, self.needle) found_positions = searcher.find_all() self.assertEqual(len(found_positions), 1) def test_needle_on_chunk_border_does_not_hide_occurence(self): buf = StringIO() buf.write("TESTING SOME SEARCHING!\n" * 10000) buf.seek(-(ReverseFileSearcher.DEFAULT_CHUNK_SIZE + 100), os.SEEK_END) buf.write(self.needle) buf.seek(-ReverseFileSearcher.DEFAULT_CHUNK_SIZE, os.SEEK_END) buf.write(self.needle) filename = self._create_temp_file(buf.getvalue()) searcher = ReverseFileSearcher(filename, self.needle) found_positions = searcher.find_all() self.assertEqual(len(found_positions), 2) def test_lots_of_needles_in_same_chunk(self): buf = StringIO() buf.write("TESTING SOME SEARCHING!\n" * 10000) buf.seek(-(ReverseFileSearcher.DEFAULT_CHUNK_SIZE + 100), os.SEEK_END) for _ in xrange(20): buf.write(self.needle) filename = self._create_temp_file(buf.getvalue()) searcher = ReverseFileSearcher(filename, self.needle) found_positions = searcher.find_all() self.assertEqual(len(found_positions), 20) def test_single_chunk(self): buf = StringIO() buf.write("TESTING SOME SEARCHING!\n" * 100) for _ in xrange(20): buf.write(self.needle) filename = self._create_temp_file(buf.getvalue()) searcher = ReverseFileSearcher(filename, self.needle) found_positions = searcher.find_all() self.assertEqual(len(found_positions), 20) def test_single_char(self): buf = StringIO() buf.write("TESTING SOME SEARCHING!\n" * 10000) filename = self._create_temp_file(buf.getvalue()) searcher = ReverseFileSearcher(filename, "C") found_positions = searcher.find_all() self.assertEqual(len(found_positions), 10000) def test_empty_needle(self): buf = StringIO() buf.write("TESTING SOME SEARCHING!\n" * 10000) filename = self._create_temp_file(buf.getvalue()) self.assertRaises(ValueError, ReverseFileSearcher, filename, "") def test_needle_larger_than_chunk_size(self): buf = StringIO() buf.write("TESTING SOME SEARCHING!\n" * 10000) needle = "NEEDLE" * ReverseFileSearcher.DEFAULT_CHUNK_SIZE filename = self._create_temp_file(buf.getvalue()) self.assertRaises(ValueError, ReverseFileSearcher, filename, needle)