test_agg_db_load.py

import unittest
import os
import json
from app import create_app, db


class AggDBTestCase(unittest.TestCase):
    """This class represents the aggdb test case"""

    def setUp(self):
        """Define test variables and initialize app."""
        self.app = create_app(config_name="development")
        self.client = self.app.test_client
        self.aggdb = {'name': 'Go to Borabora for vacation'}

        # binds the app to the current context
        with self.app.app_context():
            # create all tables
            db.create_all()

    def test_aggdb_creation(self):
        """Test API can create a aggdb (POST request)"""
        res = self.client().post('/aggdbs/', data=self.aggdb)
        self.assertEqual(res.status_code, 201)
        self.assertIn('Go to Borabora', str(res.data))

    def test_api_can_get_all_aggdbs(self):
        """Test API can get a aggdb (GET request)."""
        res = self.client().post('/aggdbs/', data=self.aggdb)
        self.assertEqual(res.status_code, 201)
        res = self.client().get('/aggdbs/')
        self.assertEqual(res.status_code, 200)
        self.assertIn('Go to Borabora', str(res.data))

    def test_api_can_get_aggdb_by_id(self):
        """Test API can get a single aggdb by using it's id."""
        rv = self.client().post('/aggdbs/', data=self.aggdb)
        self.assertEqual(rv.status_code, 201)
        result_in_json = json.loads(rv.data.decode('utf-8').replace("'", "\""))
        result = self.client().get(
            '/aggdbs/{}'.format(result_in_json['id']))
        self.assertEqual(result.status_code, 200)
        self.assertIn('Go to Borabora', str(result.data))

    def test_aggdb_can_be_edited(self):
        """Test API can edit an existing aggdb. (PUT request)"""
        rv = self.client().post(
            '/aggdbs/',
            data={'name': 'Eat, pray and love'})
        self.assertEqual(rv.status_code, 201)
        rv = self.client().put(
            '/aggdbs/1',
            data={
                "name": "Dont just eat, but also pray and love :-)"
            })
        self.assertEqual(rv.status_code, 200)
        results = self.client().get('/aggdbs/1')
        self.assertIn('Dont just eat', str(results.data))

    def test_aggdb_deletion(self):
        """Test API can delete an existing aggdb. (DELETE request)."""
        rv = self.client().post(
            '/aggdbs/',
            data={'name': 'Eat, pray and love'})
        self.assertEqual(rv.status_code, 201)
        res = self.client().delete('/aggdbs/1')
        self.assertEqual(res.status_code, 200)
        # Test to see if it exists, should return a 404
        result = self.client().get('/aggdbs/1')
        self.assertEqual(result.status_code, 404)

    def tearDown(self):
        """teardown all initialized variables."""
        with self.app.app_context():
            # drop all tables
            db.session.remove()
            db.drop_all()


# Make the tests conveniently executable
if __name__ == "__main__":
    unittest.main()