Skip to content
Snippets Groups Projects
Select Git revision
  • master default protected
1 result

test_create_movie.py

Blame
    • test_create_movie.py 2.00 KiB
    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    12
    13
    14
    15
    16
    17
    18
    19
    20
    21
    22
    23
    24
    25
    26
    27
    28
    29
    30
    31
    32
    33
    34
    35
    36
    37
    38
    39
    40
    41
    42
    43
    44
    45
    46
    47 
    

    

    

    from unittest import TestCase

from movies import MovieDatabase, Movie
from main import MoviesApp


class TestCreateMovie(TestCase):
    """
    Four versions of the same test case for a database-modifying method, each with a stronger oracle.
    """

    def test_create_movie_does_not_crash(self):
        url = MovieDatabase.construct_in_memory_url()
        movie_database = MovieDatabase(url)
        movie_database.ensure_tables_exist()
        session = movie_database.create_session()
        MoviesApp._create_movie(session, 'Example', 4000000, 290000000)
        # implicit oracle

    def test_create_movie_inserts_movies(self):
        url = MovieDatabase.construct_in_memory_url()
        movie_database = MovieDatabase(url)
        movie_database.ensure_tables_exist()
        session = movie_database.create_session()
        MoviesApp._create_movie(session, 'Example', 4000000, 290000000)
        actual = session.query(Movie).filter(Movie.title == 'Example').one()
        self.assertIsNotNone(actual)  # less-weak oracle (but still weak)

    def test_create_movie_inserts_one_movie(self):
        url = MovieDatabase.construct_in_memory_url()
        movie_database = MovieDatabase(url)
        movie_database.ensure_tables_exist()
        session = movie_database.create_session()
        MoviesApp._create_movie(session, 'Example', 4000000, 290000000)
        actual = session.query(Movie).filter(Movie.title == 'Example').count()
        self.assertEqual(actual, 1)  # weak oracle

    def test_create_movie_inserts_correct_dollar_figures(self):
        url = MovieDatabase.construct_in_memory_url()
        movie_database = MovieDatabase(url)
        movie_database.ensure_tables_exist()
        session = movie_database.create_session()
        MoviesApp._create_movie(session, 'Example', 4000000, 290000000)
        actual = session.query(Movie).filter(Movie.title == 'Example').one()
        self.assertEqual(actual.budget, 4000000)
        self.assertEqual(actual.gross_revenue, 290000000)  # strong oracle (recommended)