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 tearDown(self):
self.assertEqual(self.file_sha256, self.get_result.file_sha256)
self.assertEqual(self.file_sha256, self.get_result.file_infos.sha256)
self.get_result = scan_proberesults(self.get_result.result_id)
for tag in self.get_result.file_infos.tags:
file_tag_remove(self.file_sha256, tag.id)
for tagid in self.former_tag:
file_tag_add(self.file_sha256, tagid)
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_file_tag_add_remove(self):
for tag in self.taglist:
file_tag_add(self.file_sha256, tag.id)
get_result = scan_proberesults(self.get_result.result_id)
self.assertIn(tag.id,
[x.id for x in get_result.file_infos.tags])
for tag in self.taglist:
file_tag_remove(self.file_sha256, tag.id)
get_result = scan_proberesults(self.get_result.result_id)
self.assertNotIn(tag.id,
[x.id for x in get_result.file_infos.tags])
