带redis的Flask - 在pytest期间禁用缓存?

时间:2018-03-23 14:21:57

标签: python unit-testing caching flask

我使用Flask-Caching==1.3.3在我的Flask应用程序中实现了一个Redis缓存,但显然我的一些端点单元测试现在失败了,因为响应被缓存,使得一些POST / PUT测试失败。

在单元测试期间是否有一些好方法可以禁用缓存?我正在使用pytest==3.5.0

EG。这会失败,因为从缓存返回旧条目:

   def test_updating_biography(self):
        """Should update the current newest entry with the data in the JSON."""
        response = self.app.put(
            "/api/1.0/biography/",
            data=json.dumps(
                dict(
                    short="UnitTest Updated newest short",
                    full="UnitTest Updated newest full",
                )
            ),
            content_type="application/json"
        )

        get_bio = self.app.get("/api/1.0/biography/")
        biodata = json.loads(get_bio.get_data().decode())

        self.assertEquals(200, response.status_code)
        self.assertEquals(200, get_bio.status_code)
>       self.assertEquals("UnitTest Updated newest short", biodata["biography"][0]["short"])
E       AssertionError: 'UnitTest Updated newest short' != 'UnitTest fourth short'
E       - UnitTest Updated newest short
E       + UnitTest fourth short

tests/biography/test_biography_views.py:100: AssertionError

我试过例如:

def setUp(self):
    app.config["CACHE_TYPE"] = None
    app.config["CACHE_REDIS_URL"] = ""
    self.app = app.test_client()

还有app.config["CACHE_TYPE"] = "null"app.config["CACHE_TYPE"] = "",但它仍然在单元测试中使用缓存。

我试过这个,但它当然不属于app环境:

@cache.cached(timeout=0)
def test_updating_biography(self):

1 个答案:

答案 0 :(得分:1)

正如评论中所提到的,sytech的想法对我有用,因为我只用这个redis测试一个应用程序。显然,如果您为多个应用程序使用共享redis,这可能对您不起作用。但就我的情况而言,它完美无缺,可以毫无问题地重复:

import unittest

from flask_caching import Cache
from app import app, db


class TestBiographyViews(unittest.TestCase):
    def setUp(self):
        """Add some test entries to the database, so we can test getting the latest one."""

        # Clear redis cache completely
        cache = Cache()
        cache.init_app(app, config={"CACHE_TYPE": "redis"})
        with app.app_context():
            cache.clear()

        self.app = app.test_client()

以上就是你所需要的。其余测试用例可以正常进行。适合我。