test_unittest.py

from server import app
from model import db, connect_to_db
import unittest
import test_db
import testing.postgresql
from sqlalchemy import create_engine
import os


class ServerTests(unittest.TestCase):
    """Runs tests on routes/page render"""

    def setUp(self):
        """Code to run before every test."""

        self.client = app.test_client()  # test_client from Werkzeug library returns a "browser" to "run" app
        app.config['TESTING'] = True

    def test_homepage(self):
        """Does homepage load?"""

        result = self.client.get("/")
        self.assertEqual(result.status_code, 200)
        self.assertIn(b'<h1 class="text-center main-title">Cello<br>Tree</h1>', result.data)

    # TODO: test what renders based on user login 


class TestDb(unittest.TestCase):
    """Runs tests on database"""

    def setUp(self):
        """Code to run before every test."""

        self.client = app.test_client()
        os.system('dropdb testdb')
        os.system('createdb testdb')
        self.postgresql = testing.postgresql.Postgresql(name="testdb", port=7654)

        create_engine(self.postgresql.url())

        app.config['TESTING'] = True
        connect_to_db(app, db_uri="postgresql:///testdb")

        db.create_all()
        test_db.test_all()

    def test_homepage(self):
        """Can I add everything to my db?"""

        result = self.client.get('/')
        self.assertEqual(result.status_code, 200)
        self.assertIn(b'<h1 class="text-center main-title">Cello<br>Tree</h1>', result.data)

    def tearDown(self):
        """Code to run after every test"""

        db.session.remove()
        db.drop_all()
        db.engine.dispose()

        self.postgresql.stop()


if __name__ == "__main__":
    unittest.main(verbosity=2)