モブプロだった人のブログ

モブプログラミングに魅了されたエンジニアのブログ

Flask + Flask-SQLAlchemy のチュートリアルを日本語で (unittest 編)

チュートリアル関連記事

はじめに

今回は、unittest によるテストを書いてみたいと思います。

本格的にやるには、もっと色々なロジックを追加する必要がありますが、今回は軽くトップページを表示するところまで進めます。

各ファイルの作成

Python ファイル作成

hello_tests.py ファイルを以下のように作成します。

import hello
import unittest

class HelloTestCase(unittest.TestCase):

    def setUp(self):
        hello.app.testing = True
        self.app = hello.app.test_client()

    def tearDown(self):
        pass

    def test_top(self):
        response = self.app.get('/')
        html = response.data.decode('utf-8')
        self.assertIn('ユーザ一覧', html)
        self.assertIn('admin', html)
        self.assertIn('user', html)


if __name__ == '__main__':
    unittest.main()

トップページにアクセスし、

  • ユーザ一覧 という文字が含まれていること
  • admin という文字が含まれていること
  • user という文字が含まれていること

を確認しています。

実行確認

コンソールから python hello_tests.py を実行すると、

.
----------------------------------------------------------------------
Ran 1 test in 0.034s

OK

問題なく、テストをパスしましたね。

おわりに

今回は、本当に簡単なレベルのテストを作成しました。

次回以降で、もう少し実践的なテストコードを追加していきたいと思います。