def test_file_tag_twice(self):
found = file_search(hash=self.file_sha256)
self.assertNotEqual(len(found), 0)
file_tag_add(self.file_sha256, self.taglist[0].id)
(total, found) = file_search(hash=self.file_sha256)
self.assertGreaterEqual(total, 1)
with self.assertRaises(IrmaError):
file_tag_add(self.file_sha256, self.taglist[0].id)
def test_file_search_limit(self):
(total, _) = file_search()
(total, res) = file_search(limit=total)
self.assertEqual(type(res), list)
self.assertEqual(len(res), total)
offset = total - total / 2
limit = total / 2
(_, res) = file_search(offset=offset, limit=limit)
self.assertEqual(type(res), list)
self.assertEqual(len(res), limit)
def test_file_search_limit(self):
(total, _) = file_search()
if total > 10:
offset = total - 10
limit = 10
else:
offset = 0
limit = total
(_, res) = file_search(offset=offset, limit=limit)
self.assertEqual(type(res), list)
self.assertEqual(len(res), limit)
def test_file_search_name(self):
force = False
probes = probe_list()
scan_files(FILEPATHS, force, probes, blocking=True)
for name in FILENAMES:
data = file_search(name=name)
self.assertEqual(type(data), tuple)
(total, res) = file_search(name, limit=1)
self.assertEqual(type(res), list)
self.assertEqual(type(res[0]), IrmaResults)
self.assertEqual(len(res), 1)
self.assertEqual(type(total), int)
def test_file_search_hash(self):
force = False
probes = probe_list()
scan_files(FILEPATHS, force, probes, blocking=True)
for hash in HASHES:
(_, res) = file_search(hash=hash)
self.assertTrue(len(res) > 0)
self.assertEqual(type(res[0]), IrmaResults)
def test_file_search_tag(self):
self.assertEqual(len(self.get_result.file_infos.tags), 0)
tagged = []
for tag in self.taglist:
file_tag_add(self.file_sha256, tag.id)
tagged.append(tag.id)
(total, found) = file_search(name=self.file_name, tags=tagged)
self.assertGreater(total, 0)
self.assertIn(self.file_name, [x.name for x in found])
def test_utf8(self):
scan = self._make_scan(UTF8_PATHS)
for get_result in scan.results:
res = scan_proberesults(get_result.result_id)
self.assertIn(res.name, UTF8_SAMPLES)
for filename in UTF8_SAMPLES:
(_, res) = file_search(filename, limit=1)
self.assertEqual(type(res), list)
self.assertEqual(len(res), 1)
self.assertEqual(res[0].name, filename)
def test_utf8(self):
force = False
scan = scan_files(UTF8_PATHS, force, blocking=True)
for get_result in scan.results:
res = scan_proberesults(get_result.result_id)
self.assertIn(res.name, UTF8_SAMPLES)
for filename in UTF8_SAMPLES:
(_, res) = file_search(filename, limit=1)
self.assertEqual(type(res), list)
self.assertEqual(len(res), 1)
self.assertEqual(res[0].name, filename)
def test_file_search_not_existing_tag_and_name(self):
invalid_tagid = max([x.id for x in self.taglist]) + 1
with self.assertRaises(IrmaError):
file_search(name=self.file_name, tags=[invalid_tagid])
def test_file_search_hash_name(self):
with self.assertRaises(IrmaError):
file_search(name="name", hash="hash")
