mirror of
https://github.com/simon987/Simple-Incremental-Search-Tool.git
synced 2025-04-19 18:16:45 +00:00
35 lines
801 B
Python
35 lines
801 B
Python
from unittest import TestCase
|
|
|
|
from parsing import GenericFileParser, Sha1CheckSumCalculator, ExtensionMimeGuesser
|
|
from crawler import Crawler
|
|
|
|
|
|
class CrawlerTest(TestCase):
|
|
|
|
def test_dir_walk(self):
|
|
|
|
c = Crawler([GenericFileParser([Sha1CheckSumCalculator()], "test_files/")])
|
|
|
|
c.crawl("./test_folder")
|
|
|
|
self.assertEqual(len(c.documents), 31)
|
|
|
|
def test_file_count(self):
|
|
|
|
c = Crawler([])
|
|
|
|
self.assertEqual(c.countFiles("test_folder"), 31)
|
|
|
|
def test_path(self):
|
|
|
|
c = Crawler([GenericFileParser([], "./test_folder")])
|
|
c.crawl("./test_folder")
|
|
|
|
file_count_in_sub2 = 0
|
|
|
|
for doc in c.documents:
|
|
if doc["path"] == "sub2":
|
|
file_count_in_sub2 += 1
|
|
|
|
self.assertEqual(file_count_in_sub2, 2)
|