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)
def test_parser_no_default(self):
parser = create_parser()
args = parser.parse_args()
self.assertEqual(args.add_default, None)
def test_parser_default_argument(self):
parser = create_parser()
args = parser.parse_args(["--add-default", "constant"])
self.assertEqual(args.add_default, "constant")
def test_parser_default(self):
parser = create_parser()
args = parser.parse_args(["--add-default"])
self.assertEqual(args.add_default, '{msg[malware.name]}')
