36 lines
909 B
Python
36 lines
909 B
Python
import unittest
|
|
|
|
from werkzeug.test import Client
|
|
from werkzeug.wrappers import BaseResponse
|
|
|
|
from lodgeit.application import db, make_app
|
|
from lodgeit.models import Paste
|
|
from json import loads
|
|
|
|
|
|
def is_json(response):
|
|
"""True if the response is JSON and the HTTP status was 200."""
|
|
return (response.status_code == 200 and
|
|
response.headers.get('Content-Type', '') == 'application/json')
|
|
|
|
|
|
def json(response):
|
|
"""The response parsed as JSON.
|
|
|
|
No attempt is made to ensure the response is valid or even looks
|
|
like JSON before parsing.
|
|
"""
|
|
return loads(response.data)
|
|
|
|
|
|
class BaseTestCase(unittest.TestCase):
|
|
def setUp(self):
|
|
self.client = Client(
|
|
make_app('sqlite://', 'NONE', False, True), BaseResponse)
|
|
|
|
def teardown(self):
|
|
Paste.query.delete()
|
|
db.session.commit()
|
|
db.session.remove()
|
|
super().teardown()
|