Как тестировать приложения Flask с помощью unittest?

Question

[Jekson]

Решил поразбиратся с написанием тестов для flask приложений. Нужен совет бывалых.
Вот такой простой тест, который должен проверять доступна ли главная страница.

import unittest
import os
from app import app

class BasicTestCase(unittest.TestCase):
    def setUp(self):
        self.app = app.test_client()

    def test_main_page(self):
        response = self.app.get('/', content_type='html/text')
        self.assertEqual(response.status_code, 200)

Почему я получаю OK даже если локально не запущен сервер. По идее мне этот тест нужен, например если на проде отвалился Nginx и висит 502 ошибка.
И второй вопрос по тестированию взаимодействия с БД. У меня postgresql, мне тестировать уже существующую базу, или каждый раз создавать новую и удалять ее после теста?

Answer 1

[LODIII]

мне тестировать уже существующую базу

не всегда сможете привести в первоначальное состояние, насколько это критично.
Например автоинкрементные ключи итд.

Comments

[Jekson]

LODIII Да про ключи это верно подмечено, на данном этапе не важно, но хотелось бы сразу выбрать правильный способ.

Answer 2

[kdvoryashin]

На существующей базе тестировать никогда ничего не надо, это плохая практика.
Во время теста создается тестовая бд, setUp() метод как раз для этого.
Кажется в новых версиях flask перешел на pytest вместо unittest, но принцип один в общем-то.
Рекомендую подробнее почитать документацию (перевод), там очень хорошие примеры по вашему вопросу.

Что касается проверки доступна ли главная страница. Это немного не про юнит тестирование.. Когда вы пытаетесь проверить не отвалился ли сейчас nginx, это больше про мониторинг.

Тесты всегда должны быть изолированы от рабочей среды. Когда вы запускаете тест, он запускает отдельно ваше приложение, на время работы теста.

Comments

[Jekson]

kdvoryashin Написал такой тетст, как раз с инициализацией новой БД. И сижу голову ломаю, почему удаляется рабочая БД вместо создания новой. Может есть мысли где ошибся?

from flask import Flask
import unittest
from app import db, app
from images.models import Images, ImageGroup


class FlaskTestCase(unittest.TestCase):
    def create_app(self):
        app.config.from_object('config.TestingConfig')
        return app

    def setUp(self):
        db.create_all()
        im_group = ImageGroup()
        db.session.add(im_group)
        db.session.commit()
        image = Images(img_filename='test.jpg', counter=1, img_data='tests', imagegroup_id=im_group.id)
        db.session.add(image)
        db.session.commit()

    def tearDown(self):
        db.session.remove()
        db.drop_all()

    def test_model_exist(self):
        self.assertTrue(db.session.query(Images).filter(Images.img_filename == 'test.jpg').first() is not None)
        pass


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

[Роман]

в setUp() перед db.create_all() надо подоткнуть свою тестовую бд.
Например вот так

app.config["SQLALCHEMY_DATABASE_URI"] = "sqlite:///" + os.path.join(BASEDIR, "test.db")

ну или как она там у тебя цепляется

[Jekson]

Роман, Я пробую создать postgresql базу, для этого перезатираю конфиг основной на тестовый.
app.config.from_object('config.TestingConfig')

config.py

class TestingConfig(Configuration):
    TESTING = True
    SQLALCHEMY_DATABASE_URI = 'postgresql://%(user)s:%(pw)s@%(host)s:%(port)s/%(db)s' % TEST_POSTGRES
    DEBUG = False

Код вот так теперь выглядит

from flask import Flask
import unittest
from app import app
from images.models import Images, ImageGroup
from flask_sqlalchemy import SQLAlchemy

class FlaskTestCase(unittest.TestCase):
   
    def setUp(self):
        app.config.from_object('config.TestingConfig')
        self.db = SQLAlchemy(app)
        self.db.create_all()
        im_group = ImageGroup()
        self.db.session.add(im_group)
        self.db.session.commit()
        image = Images(img_filename='test.jpg', counter=1, img_data='tests', imagegroup_id=im_group.id)
        self.db.session.add(image)
        self.db.session.commit()

    def tearDown(self):
        self.db.session.remove()
        self.db.drop_all()
        pass

    def test_model_exist(self):
        self.assertTrue(self.db.session.query(Images).filter(Images.img_filename == 'test.jpg').first() is not None)
        pass

Но при запуске теста получаю

File "test.py", line 26, in setUp
    self.db.create_all()

File "/home/y700/Env/bee_ml/lib/python3.6/site-packages/psycopg2/__init__.py", line 130, in connect
    conn = _connect(dsn, connection_factory=connection_factory, **kwasync)
sqlalchemy.exc.OperationalError: (psycopg2.OperationalError) FATAL:  database "db" does not exist

Не могу понять что не нравится. Разве команда self.db.create_all() как раз не для создания новой бд?