diff --git a/.coveragerc b/.coveragerc index dae8ddc..bd4dabb 100644 --- a/.coveragerc +++ b/.coveragerc @@ -3,3 +3,5 @@ omit= vrobbler/wsgi.py vrobbler/asgi.py vrobbler/cli.py + *admin.py + migrations/* diff --git a/tests/scrobbles_tests/test_imdb.py b/tests/scrobbles_tests/test_imdb.py new file mode 100644 index 0000000..c34a809 --- /dev/null +++ b/tests/scrobbles_tests/test_imdb.py @@ -0,0 +1,12 @@ +import pytest +import imdb +from mock import patch + +from vrobbler.apps.scrobbles.imdb import lookup_video_from_imdb + + +def test_lookup_imdb_bad_id(caplog): + data = lookup_video_from_imdb('3409324') + assert data is None + assert caplog.records[0].levelname == "WARNING" + assert caplog.records[0].msg == "IMDB ID should begin with 'tt' 3409324" diff --git a/vrobbler/apps/scrobbles/imdb.py b/vrobbler/apps/scrobbles/imdb.py index 3ee4f86..2c29fea 100644 --- a/vrobbler/apps/scrobbles/imdb.py +++ b/vrobbler/apps/scrobbles/imdb.py @@ -2,10 +2,10 @@ import logging from typing import Optional from django.utils import timezone -import imdb +from imdb import Cinemagoer from videos.models import Video -imdb_client = imdb.Cinemagoer() +imdb_client = Cinemagoer() logger = logging.getLogger(__name__)