flask_test.py

import unittest

from flask05 import app


class FlaskTest(unittest.TestCase):
    def setUp(self) -> None:
        self.client = app.test_client()

    def test_index(self):
        response = self.client.get("/")
        statuscode = response.status_code
        self.assertEqual(statuscode, 200)
        self.assertEqual(
            "<h2>Use this site to maintain and organize your notes.</h2>"
            in response.get_data(as_text=True),
            True,
        )

    def test_notes(self):
        response = self.client.get("/notes")
        statuscode = response.status_code
        self.assertEqual(statuscode, 200)
        self.assertEqual("Title" and "Date" in response.get_data(as_text=True), True)

    def test_note(self):
        response = self.client.get("/notes/1")
        statuscode = response.status_code
        self.assertEqual(statuscode, 200)
        self.assertEqual("Note Details" in response.get_data(as_text=True), True)

    def test_new(self):
        response = self.client.get("/notes/new")
        statuscode = response.status_code
        self.assertEqual(statuscode, 200)
        self.assertEqual(
            '<form action="new" method="post">' in response.get_data(as_text=True), True
        )

    def test_delete(self):
        response = self.client.get("/notes/delete")
        statuscode = response.status_code
        self.assertEqual(statuscode, 500)


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