add integration test

jaebradley · Jun 3, 2017 · c818fc2 · c818fc2
1 parent 88746b1
commit c818fc2
Show file tree

Hide file tree

Showing 2 changed files with 41 additions and 3 deletions.
diff --git a/tables/builders.py b/tables/builders.py
@@ -6,7 +6,12 @@
 
 
 class MovieSearchTableBuilder:
-    HEADERS = [formatted_header(text="Film"), formatted_header(text="Score"), formatted_header(text="Year"), formatted_header(text="Cast")]
+    HEADERS = [
+        formatted_header(text="Film"),
+        formatted_header(text="Score"),
+        formatted_header(text="Year"),
+        formatted_header(text="Cast")
+    ]
     COLUMN_JUSTIFICATION = {
         0: "left",
         1: "left",
@@ -24,8 +29,11 @@ def build(self, movies):
         return table.table
 
     def rows(self, movies):
-        sorted_movies = sorted(movies, key=lambda movie: movie.rotten_tomatoes_score, reverse=True)
-        return [self.row_builder.build(movie=movie) for movie in sorted_movies]
+        sorted_movies = sorted(movies, key=lambda m: m.rotten_tomatoes_score, reverse=True)
+        return [
+            self.row_builder.build(movie=movie)
+            for movie in sorted_movies
+        ]
 
 
 class TvShowSearchTableBuilder:

diff --git a/tests/test_movieSearchTableBuilder.py b/tests/test_movieSearchTableBuilder.py
@@ -0,0 +1,30 @@
+from unittest import TestCase
+import logging
+
+from data import MovieSearchResult
+from tables.builders import MovieSearchTableBuilder
+
+logging.basicConfig(level=logging.DEBUG)
+
+
+class TestMovieSearchTableBuilder(TestCase):
+    movie_search_result = MovieSearchResult(name="name", year="year", rotten_tomatoes_score=1, cast=["Jae", "Baebae"])
+    another_movie_search_result = MovieSearchResult(name="another name", year="another year", rotten_tomatoes_score=100, cast=["Another Jae", "Another Baebae"])
+    movies = [another_movie_search_result, movie_search_result]
+    table_builder = MovieSearchTableBuilder()
+
+    def integration_test(self):
+        expected = """(0lqqqqqqqqqqqqqqwqqqqqqqwqqqqqqqqqqqqqqwqqqqqqqqqqqqqqqqk(B
+(0x(B [4m[1mFilm[0m         (0x(B [4m[1mScore[0m (0x(B     [4m[1mYear[0m     (0x(B [4m[1mCast[0m           (0x(B
+(0tqqqqqqqqqqqqqqnqqqqqqqnqqqqqqqqqqqqqqnqqqqqqqqqqqqqqqqu(B
+(0x(B [1manother name[0m (0x(B [31m100% [0m (0x(B another year (0x(B Another Jae    (0x(B
+(0x(B              (0x(B       (0x(B              (0x(B Another Baebae (0x(B
+(0tqqqqqqqqqqqqqqnqqqqqqqnqqqqqqqqqqqqqqnqqqqqqqqqqqqqqqqu(B
+(0x(B [1mname[0m         (0x(B [36m1% [0m   (0x(B     year     (0x(B Jae            (0x(B
+(0x(B              (0x(B       (0x(B              (0x(B Baebae         (0x(B
+(0mqqqqqqqqqqqqqqvqqqqqqqvqqqqqqqqqqqqqqvqqqqqqqqqqqqqqqqj(B"""
+        output = self.table_builder.build(movies=self.movies)
+        self.assertEqual(expected, output)
+
+        logging.debug("Expected Movie Search Table:\n{expected}".format(expected=expected))
+        logging.debug("Actual Movie Search Table:\n{output}".format(output=output))