Test saving malware locally

maltrieve.py

@@ -190,7 +190,7 @@ def save_malware(response, cfg):
     url = response.url
     data = response.content
     mime_type = magic.from_buffer(data, mime=True)
-    if mime_type in cfg.black_list:
+    if cfg.black_list and mime_type in cfg.black_list:
         logging.info('{mtype} in ignore list for {url}'.format(mtype=mime_type, url=url))
         return
     if cfg.white_list:

test.py

@@ -81,3 +81,10 @@ def test_get_sources():
     cfg = maltrieve.config(args, filename='maltrieve-test.cfg')
     source_lists = maltrieve.process_source_lists(cfg)
     assert len(source_lists) == len(cfg.source_urls)
+
+
+def test_save_malware():
+    args = maltrieve.setup_args([])
+    cfg = maltrieve.config(args, filename='maltrieve-test.cfg')
+    r = requests.get('http://maltrieve.org')
+    assert maltrieve.save_malware(r, cfg)