26 lines
569 B
Python
26 lines
569 B
Python
import unittest
|
|
|
|
class UnitTest(unittest.TestCase):
|
|
def setUp(self):
|
|
from app import app
|
|
app.testing = True
|
|
self.app = app.test_client()
|
|
|
|
def test_homepage(self):
|
|
rv = self.app.get('/')
|
|
self.assertEqual(rv.status, '200 OK')
|
|
|
|
def test_play(self):
|
|
rv = self.app.get('/play/TEST')
|
|
self.assertEqual(rv.status, '200 OK')
|
|
|
|
def test_end(self):
|
|
rv = self.app.get('/end/TEST')
|
|
self.assertEqual(rv.status, '200 OK')
|
|
|
|
def test_lost(self):
|
|
rv = self.app.get('/lost/TEST')
|
|
self.assertEqual(rv.status,'200 OK')
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|