def do_perform_remote_test(self, caplog, url, expected): thug = ThugAPI() thug.set_useragent('win7ie90') thug.set_image_processing() thug.set_threshold(2) thug.disable_cert_logging() thug.set_features_logging() thug.set_ssl_verify() thug.log_init(url) thug.add_htmlclassifier( os.path.join(self.signatures_path, "html_signature_12.yar")) thug.add_imageclassifier( os.path.join(self.signatures_path, "image_signature_14.yar")) thug.add_imageclassifier( os.path.join(self.signatures_path, "image_signature_15.yar")) thug.run_remote(url) records = [r.message for r in caplog.records] matches = 0 for e in expected: for record in records: if e in record: matches += 1 assert matches >= len(expected)
def do_perform_test(self, caplog, url, expected): thug = ThugAPI() thug.set_useragent('win7ie90') thug.set_broken_url() thug.log_init(url) thug.run_remote(url) records = [r.message for r in caplog.records] matches = 0 for e in expected: for record in records: if e in record: matches += 1 assert matches >= len(expected)
def do_perform_test(self, caplog, url, expected): thug = ThugAPI() thug.set_useragent('win7ie90') thug.set_vt_query() thug.set_vt_submit() thug.disable_cert_logging() thug.set_features_logging() thug.log_init(url) thug.run_remote(url) records = [r.message for r in caplog.records] matches = 0 for e in expected: for record in records: if e in record: matches += 1 assert matches >= len(expected)