1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44
|
import unittest
from pyramid import testing
class TutorialViewTests(unittest.TestCase):
def setUp(self):
self.config = testing.setUp()
def tearDown(self):
testing.tearDown()
def test_home(self):
from .views import home
request = testing.DummyRequest()
response = home(request)
self.assertEqual(response.status_code, 200)
self.assertIn(b'Visit', response.body)
def test_hello(self):
from .views import hello
request = testing.DummyRequest()
response = hello(request)
self.assertEqual(response.status_code, 200)
self.assertIn(b'Go back', response.body)
class TutorialFunctionalTests(unittest.TestCase):
def setUp(self):
from tutorial import main
app = main({})
from webtest import TestApp
self.testapp = TestApp(app)
def test_home(self):
res = self.testapp.get('/', status=200)
self.assertIn(b'<body>Visit', res.body)
def test_hello(self):
res = self.testapp.get('/howdy', status=200)
self.assertIn(b'<body>Go back', res.body)
|