Esempio n. 1
0
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
Esempio n. 2
0
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
Esempio n. 3
0
 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
Esempio n. 4
0
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()
Esempio n. 5
0
 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
Esempio n. 6
0
 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
Esempio n. 7
0
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()
Esempio n. 8
0
 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