First tests for imdb module

This commit is contained in:
2023-01-20 13:02:32 -05:00
parent 13bdc201f0
commit cbe4abfb5f
3 changed files with 16 additions and 2 deletions

View File

@ -3,3 +3,5 @@ omit=
vrobbler/wsgi.py vrobbler/wsgi.py
vrobbler/asgi.py vrobbler/asgi.py
vrobbler/cli.py vrobbler/cli.py
*admin.py
migrations/*

View File

@ -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"

View File

@ -2,10 +2,10 @@ import logging
from typing import Optional from typing import Optional
from django.utils import timezone from django.utils import timezone
import imdb from imdb import Cinemagoer
from videos.models import Video from videos.models import Video
imdb_client = imdb.Cinemagoer() imdb_client = Cinemagoer()
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)