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/asgi.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 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__)