from mock import Mock
from tornado.testing import AsyncHTTPTestCase
from candolint.models import database, Project, Change, Check
from viewer import CandolintViewer
class ViewerTestCase(AsyncHTTPTestCase):
    def get_app(self):
        database.init(':memory:')
        database.connect()
        database.create_tables([Project, Change, Check], safe=True)
        database.init = Mock()
        database.connect = Mock()
        database.disconnect = Mock()
        return CandolintViewer()