def test_download_not_ignore_not_adware(self): parser = create_parser() args = parser.parse_args(["--mwnmp-ignore-adware"]) with requests_mock.Mocker() as mocker: mocker.get(URL, text='"^foobar$",foobar,not adware,barfoo') result = json.loads(main(args)) self.assertEqual(len(result), 1) self.assertIsInstance(result, list)
def test_download_add_default_constant(self): " Test --add-default parameter with an argument" parser = create_parser() args = parser.parse_args(["--add-default", "constant"]) with requests_mock.Mocker() as mocker: mocker.get(URL, text='') result = json.loads(main(args)) self.assertEqual(result, [{'if': {'classification.taxonomy': 'malicious-code', 'malware.name': '.*'}, 'rulename': 'default', 'then': {'classification.identifier': 'constant'}}] )
def test_download(self): parser = create_parser() args = parser.parse_args("") result = json.loads(main(args)) self.assertGreater(len(result), 100) self.assertIsInstance(result, list)
def test_download_identifier_expression(self): parser = create_parser() args = parser.parse_args('-e "^foobar$" -i foobar'.split(' ')) result = json.loads(main(args)) self.assertGreater(len(result), 100) self.assertIsInstance(result, list)