tests.py

#!/usr/bin/env python3

# -------------------------------
# Copyright (C) 2016
# IBDB
# -------------------------------

# -------
# Imports
# -------

from unittest import main, TestCase
from models import *
from app import app
# ----------
# TestModels
# ----------


class TestModels (TestCase):

    # ------
    # Author
    # ------

    def test_model_author_1(self):
        with app.app_context():
            testauthor = Author.query.get(1)
            self.assertEqual(testauthor.id, 1)

    def test_model_author_2(self):
        with app.app_context():
            testauthor = Author.query.filter_by(
                first_name='Anthony', last_name='Doerr')
            self.assertEqual(testauthor[0].first_name, 'Anthony')
            self.assertEqual(testauthor[0].last_name, 'Doerr')

    def test_model_author_3(self):
        with app.app_context():
            testauthor = Author.query.filter_by(
                first_name='Paula', last_name='Hawkins')
            self.assertEqual(
                testauthor[0].Books[0].title, 'THE GIRL ON THE TRAIN')

    # ----------------
    # Author.to_dict()
    # ----------------

    def test_author_to_dict_1(self):
        with app.app_context():
            testauthor = Author(id=1, first_name='Anthony', last_name='Doerr')
            dictauthor = testauthor.to_dict()
            self.assertEqual(testauthor.first_name, dictauthor['first_name'])

    def test_author_to_dict_2(self):
        with app.app_context():
            testauthor = Author(id=1, first_name='Paula', last_name='Hawkins')
            dictauthor = testauthor.to_dict()
            self.assertEqual(testauthor.last_name, dictauthor['last_name'])

    def test_author_to_dict_3(self):
        with app.app_context():
            testauthor = Author(
                id=1, first_name='Harper', last_name='Lee', bio='hi')
            dictauthor = testauthor.to_dict()
            self.assertEqual(testauthor.bio, dictauthor['bio'])

    # -----------------
    # Author.get_html()
    # -----------------

    def test_author_get_html_1(self):
        with app.app_context():
            testauthor = Author(id=1, first_name='Anthony', last_name='Doerr')
            htmlauthor = testauthor.get_html('anthony')
            self.assertEqual(
                "<td>Anthony Doerr</td><td><b><i>anthony</i></b> </td>", htmlauthor)

    def test_author_get_html_2(self):
        with app.app_context():
            testauthor = Author(id=1, first_name='Anthony', last_name='Doerr')
            htmlauthor = testauthor.get_html('doerr')
            self.assertEqual(
                "<td>Anthony Doerr</td><td><b><i>doerr</i></b> </td>", htmlauthor)

    def test_author_get_html_3(self):
        with app.app_context():
            testauthor = Author(id=2, first_name='John', last_name='Green')
            htmlauthor = testauthor.get_html('john')
            self.assertEqual(
                "<td>John Green</td><td><b><i>john</i></b> </td>", htmlauthor)

    # -----------------
    # Author.get_link()
    # -----------------
    def test_author_get_link_1(self):
        with app.app_context():
            testauthor = Author(id=1, first_name='Anthony', last_name='Doerr')
            linkauthor = testauthor.get_link()
            self.assertEqual("/author/1", linkauthor)

    def test_author_get_link_2(self):
        with app.app_context():
            testauthor = Author(id=10000, first_name='John', last_name='Green')
            linkauthor = testauthor.get_link()
            self.assertEqual("/author/10000", linkauthor)

    def test_author_get_link_3(self):
        with app.app_context():
            testauthor = Author(id=9999999999, first_name='Rick', last_name='Riordan')
            linkauthor = testauthor.get_link()
            self.assertEqual("/author/9999999999", linkauthor)


    # ----
    # Book
    # ----

    def test_model_book_1(self):
        with app.app_context():
            testbook = Book.query.get(1)
            self.assertEqual(testbook.id, 1)

    def test_model_book_2(self):
        with app.app_context():
            testbook = Book.query.filter_by(
                title='ALL THE LIGHT WE CANNOT SEE').all()
            self.assertEqual(testbook[0].title, 'ALL THE LIGHT WE CANNOT SEE')

    def test_model_book_3(self):
        with app.app_context():
            testbook = Book.query.filter_by(
                title='THE GIRL ON THE TRAIN').all()
            self.assertEqual(testbook[0].author.first_name, 'Paula')

    # --------------
    # Book.to_dict()
    # --------------

    def test_book_to_dict_1(self):
        with app.app_context():
            testbook = Book(
                id=1, title='ALL THE LIGHT WE CANNOT SEE', isbn='1', summary='hi')
            dictbook = testbook.to_dict()
            self.assertEqual(testbook.title, dictbook['title'])

    def test_book_to_dict_2(self):
        with app.app_context():
            testbook = Book(
                id=1, title='ALL THE LIGHT WE CANNOT SEE', isbn='1', summary='hi')
            dictbook = testbook.to_dict()
            self.assertEqual(testbook.isbn, dictbook['isbn'])

    def test_book_to_dict_3(self):
        with app.app_context():
            testbook = Book(
                id=1, title='ALL THE LIGHT WE CANNOT SEE', isbn='1', summary='hi')
            dictbook = testbook.to_dict()
            self.assertEqual(testbook.summary, dictbook['summary'])

    # -----------------
    # Book.get_html()
    # -----------------
    def test_book_get_html_1(self):
        with app.app_context():
            testbook = Book(title="moonshinning for dummies")
            htmlbook = testbook.get_html('dummies')
            self.assertEqual(
                "<td>moonshinning for dummies</td><td>moonshinning for <b><i>dummies</i></b> </td>",
                htmlbook)

    def test_book_get_html_2(self):
        with app.app_context():
            testbook = Book(title="Going Bananas")
            htmlbook = testbook.get_html('Bananas')
            self.assertEqual(
                "<td>Going Bananas</td><td>going <b><i>bananas</i></b> </td>",
                htmlbook)

    def test_book_get_html_3(self):
        with app.app_context():
            testbook = Book(title="Moonshine with Bananas")
            htmlbook = testbook.get_html('Bananas')
            self.assertEqual(
                "<td>Moonshine with Bananas</td><td>moonshine with <b><i>bananas</i></b> </td>",
                htmlbook)

    # -----------------
    # Book.get_link()
    # -----------------
    def test_book_get_link_1(self):
        with app.app_context():
            testbook = Book(id=5, title="The art of distilling good whiskey")
            linkbook = testbook.get_link()
            self.assertEqual("/book/5", linkbook)

    def test_book_get_link_2(self):
        with app.app_context():
            testbook = Book(id=999999999999, title="Jim and Jack and Hank")
            linkbook = testbook.get_link()
            self.assertEqual("/book/999999999999", linkbook)

    def test_book_get_link_3(self):
        with app.app_context():
            testbook = Book(id=1010101010, title="Daddy's Gone a Hunting")
            linkbook = testbook.get_link()
            self.assertEqual("/book/1010101010", linkbook)

# ----
# Main
# ----

if __name__ == '__main__':
    main()