#!/usr/bin/env python3 import unittest from tmdb.tmdb_api import Movie, TvShow from tmdb.tmdb import TmdbBot from tmdb.database import Database from sqlalchemy import create_engine class TestTmdbMethods(unittest.TestCase): ### TMDB API def test_search_item(self): movie = Movie() id = movie.search_title('Breakfast Club') self.assertEqual(id, 2108) def test_cast(self): movie = Movie() movie.search_title('Breakfast Club') self.assertEqual('Anthony Michael Hall', movie.cast[0]) def test_title(self): movie = Movie() movie.search_title('Breakfast Club') self.assertEqual('The Breakfast Club', movie.title) def test_overview(self): movie = Movie() movie.search_title('Breakfast Club') description = 'Samstag morgen in einer amerikanischen High-School' self.assertEqual(description, movie.overview[:len(description)]) def test_change_language(self): movie = Movie() movie.set_language('en') movie.search_title('Breakfast Club') description = 'Five high school students from different walks of life endure a Saturday detention' self.assertEqual(description, movie.overview[:len(description)]) def test_html_construction(self): movie = Movie() tmdb = TmdbBot("","" ,"" ,"" ,"" ,"" ,"" ,"" ,"" ) movie.query_details('550') message = tmdb.construct_html_message(movie, overview_length = 10) self.assertEqual(message, """

Fight Club

Ein Yuppie [...]

Acting: Edward Norton, Brad Pitt, Helena Bonham Carter

taken from www.themoviedb.org

""") def test_database_language(self): engine = create_engine('sqlite:///test.db', echo = True) db = Database(engine) db.set_language('@testuser:example.com', 'de') self.assertEqual(str(db.get_language('@testuser:example.com')), 'de') db.set_language('@testuser:example.com', 'en') self.assertEqual(str(db.get_language('@testuser:example.com')), 'en') def test_id_lookup(self): movie = Movie() movie.query_details('2108') self.assertEqual('The Breakfast Club', movie.title) def test_search_fails(self): movie = Movie() id = movie.search_title('Breakfast Club 2019') self.assertEqual(id, None) self.assertEqual(None, movie.title) # TV Shows def test_search_tvshow(self): movie = TvShow() id = movie.search_title('The Flash') self.assertEqual(id, 60735) def test_tv_title(self): movie = TvShow() movie.search_title('The Flash') self.assertEqual('The Flash', movie.title) def test_cast(self): movie = TvShow() movie.search_title('The Flash') self.assertEqual('Grant Gustin', movie.cast[0]) self.assertEqual('Carlos Valdes', movie.cast[2]) if __name__ == '__main__': unittest.main()