1、首先在app目录下创建单元测试脚本tests.py内容如下:
import unittest
import identidock
class TestCase(unittest.TestCase):
def setUp(self):
identidock.app.config["TESTING"] = True
self.app = identidock.app.test_client()
def test_get_mainpage(self):
page = self.app.post("/", data=dict(name="Moby Dock"))
assert page.status_code == 200
assert 'Hello' in str(page.data)
assert 'Moby Dock' in str(page.data)
def test_html_escaping(self):
page = self.app.post("/", data=dict(name='"><b>TEST</b><!--'))
assert '<b>' not in str(page.data)
if __name__ == '__main__':
unittest.main()
2、修改identidock.py如下使用html.escape 方法处理转义
3、构建运行测试脚本
4、修改cmd.py把测试脚本的路由加进去
5、重新构建容器,通过更改变量来执行测试