39:81920cb8753b
Anton Shestakov <av6@dwimlabs.net>, Sat, 18 Jun 2016 22:45:43 +0800
tests: create the in-memory database once in setup_module() Also mock the correct method, close() instead of disconnect().
next change 41:8d6436a284d2
previous change 37:bbce1e45da1e

tests/test_viewer.py

Permissions: -rw-r--r--

File Latest Revisions Annotate Diff Comparison
Other formats:
JSON Raw
Feeds:
Atom RSS 
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