libbmc/libbmc/tests/test_fetcher.py

16 lines
483 B
Python

import unittest
from libbmc.fetcher import *
class TestFetcher(unittest.TestCase):
def test_download(self):
dl, contenttype = download('http://arxiv.org/pdf/1312.4006.pdf')
self.assertIn(contenttype, ['pdf', 'djvu'])
self.assertNotEqual(dl, '')
def test_download_invalid_type(self):
self.assertEqual(download('http://phyks.me/'), (None, None))
def test_download_invalid_url(self):
self.assertEqual(download('a'), (None, None))