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_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])
def setUp(self):
if self.taglist is None:
self.taglist = tag_list()
if len(self.taglist) == 0:
raise unittest.SkipTest("Skipping No tag found (please add some)")
# Insure file is present (Force=False)
scan = scan_files([self.file_path], False, blocking=True)
self.get_result = scan_proberesults(scan.results[0].result_id)
# Insure file got no tags for test
self.former_tag = [x.id for x in self.get_result.file_infos.tags]
if len(self.former_tag) != 0:
for tagid in self.former_tag:
file_tag_remove(self.file_sha256, tagid)
self.get_result = scan_proberesults(scan.results[0].result_id)
