class TestNotFlask(unittest.TestCase):
def setUp(self):
self.app = NotFlask()
def test_valid_route(self):
@self.app.route('/')
def index():
return 'Hello World'
self.assertEqual(self.app.serve('/'), 'Hello World')
def test_url_parameter(self):
@self.app.route('/hello/<username>')
def hello_user(username):
return 'Hello {}!'.format(username)
self.assertEqual(self.app.serve('/hello/ains'), 'Hello ains!')
def test_invalid_route(self):
with self.assertRaises(ValueError):
self.app.serve('/invalid')
def setUp(self):
self.app = NotFlask()
from not_flask import NotFlask
if __name__ == '__main__':
app = NotFlask()
@app.route('/')
def index():
return 'Hello World'
print(app.serve('/'))
