__init__.py

import logging
from unittest.mock import patch

from flask_testing import TestCase

from aura_engine_api.rest.encoder import JSONEncoder

with patch("sys.argv", ["config=tests/config/engine-api.ini"]):
    from aura_engine_api.app import app


class BaseTestCase(TestCase):
    def create_app(self):
        logging.getLogger("connexion.operation").setLevel("ERROR")
        app.json_encoder = JSONEncoder
        return app

    def assert204(self, response, message=None):
        """Based on assert200 in parent class"""
        self.assertStatus(response, 204, message)