from mock import Mock
from tornado.testing import AsyncHTTPTestCase
from candolint.models import database, Project, Change, Check
from viewer import CandolintViewer
def setup_module():
    database.init(':memory:')
    database.connect()
    database.create_tables([Project, Change, Check], safe=True)
    database.init = Mock()
    database.connect = Mock()
    database.close = Mock()
class ViewerTestCase(AsyncHTTPTestCase):
    def get_app(self):
        return CandolintViewer()
    def test_index(self):
        response = self.fetch('/')
        assert response.code == 200
    def test_404(self):
        response = self.fetch('/nobodyhere')
        assert response.code == 404