def _generate_test_job_with_file(params, filename):
test_file = f"{settings.PROJECT_LOCATION}/test_files/{filename}"
with open(test_file, "rb") as f:
django_file = File(f)
params["file"] = django_file
params["md5"] = hashlib.md5(django_file.file.read()).hexdigest()
test_job = Job(**params)
test_job.save()
return test_job
def _generate_test_job_with_file(params, filename):
test_file = "{}/test_files/{}".format(settings.PROJECT_LOCATION, filename)
with open(test_file, "rb") as f:
django_file = File(f)
params['file'] = django_file
params['md5'] = hashlib.md5(django_file.file.read()).hexdigest()
test_job = Job(**params)
test_job.save()
return test_job
def setUp(self):
params = self.get_params()
params["md5"] = hashlib.md5(
params["observable_name"].encode("utf-8")).hexdigest()
test_job = Job(**params)
test_job.save()
self.job_id = test_job.id
self.observable_name = test_job.observable_name
self.observable_classification = test_job.observable_classification
class _FileAnalyzersScriptsTestCase(_AbstractAnalyzersScriptTestCase):
# define runtime configs
runtime_configuration = {
"VirusTotal_v2_Scan_File": {"wait_for_scan_anyway": True, "max_tries": 1},
"VirusTotal_v3_Scan_File": {"max_tries": 1, "poll_distance": 1},
"VirusTotal_v3_Get_File": {"max_tries": 1, "poll_distance": 1},
"VirusTotal_v3_Get_File_And_Scan": {
"max_tries": 1,
"poll_distance": 1,
"force_active_scan": True,
"force_active_scan_if_old": False,
},
"Cuckoo_Scan": {"max_poll_tries": 1, "max_post_tries": 1},
"PEframe_Scan": {"max_tries": 1},
"MWDB_Scan": {
"upload_file": True,
"max_tries": 1,
},
"Doc_Info_Experimental": {
"additional_passwords_to_check": ["testpassword"],
"experimental": True,
},
}
@classmethod
def get_params(cls):
return {
**super().get_params(),
"is_sample": True,
}
def setUp(self):
super().setUp()
# get params
params = self.get_params()
# save job instance
self.test_job = Job(**params)
# overwrite if set in env var
if len(self.analyzers_to_test):
self.test_job.analyzers_to_execute = self.analyzers_to_test
self._read_file_save_job(filename=params["file_name"])
def _read_file_save_job(self, filename: str):
test_file = f"{settings.PROJECT_LOCATION}/test_files/{filename}"
with open(test_file, "rb") as f:
self.test_job.file = File(f)
self.test_job.md5 = hashlib.md5(f.read()).hexdigest()
self.test_job.save()
def setUp(self):
params = {
"source": "test",
"is_sample": False,
"observable_name": os.environ.get("TEST_MD5", ""),
"observable_classification": "hash",
"force_privacy": False,
"analyzers_requested": ["test"]
}
params["md5"] = hashlib.md5(
params['observable_name'].encode('utf-8')).hexdigest()
test_job = Job(**params)
test_job.save()
self.job_id = test_job.id
self.observable_name = test_job.observable_name
self.observable_classification = test_job.observable_classification
def setUp(self):
params = {
"source": "test",
"is_sample": False,
"observable_name": os.environ.get("TEST_DOMAIN", "www.google.com"),
"observable_classification": "domain",
"force_privacy": False,
"analyzers_requested": ["test"],
}
params["md5"] = hashlib.md5(
params["observable_name"].encode("utf-8")).hexdigest()
test_job = Job(**params)
test_job.save()
self.job_id = test_job.id
self.observable_name = test_job.observable_name
self.observable_classification = test_job.observable_classification
class _ObservableAnalyzersScriptsTestCase(_AbstractAnalyzersScriptTestCase):
# define runtime configs
runtime_configuration = {
"Triage_Search": {
"max_tries": 1,
},
"VirusTotal_v3_Get_Observable": {
"max_tries": 1,
"poll_distance": 1,
},
}
@classmethod
def get_params(cls):
return {
**super().get_params(),
"is_sample": False,
}
def setUp(self):
super().setUp()
# init job instance
params = self.get_params()
params["md5"] = hashlib.md5(
params["observable_name"].encode("utf-8")
).hexdigest()
self.test_job = Job(**params)
# overwrite if not set in env var
if len(self.analyzers_to_test):
self.test_job.analyzers_to_execute = self.analyzers_to_test
else:
self.test_job.analyzers_to_execute = [
config.name
for config in self.analyzer_configs.values()
if config.is_observable_type_supported(
params["observable_classification"]
)
]
# save job
self.test_job.save()
def setUp(self):
params = {
"source":
"test",
"is_sample":
False,
"observable_name":
os.environ.get("TEST_MD5", "446c5fbb11b9ce058450555c1c27153c"),
"observable_classification":
"hash",
"force_privacy":
False,
"analyzers_requested": ["test"],
}
params["md5"] = hashlib.md5(
params["observable_name"].encode("utf-8")).hexdigest()
test_job = Job(**params)
test_job.save()
self.job_id = test_job.id
self.observable_name = test_job.observable_name
self.observable_classification = test_job.observable_classification