pk/atst
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

.read() should return an ImmutableMultiDict

Browse Source
This commit is contained in:
richard-dds 2018-11-19 11:38:22 -05:00
parent 29f9520dda
commit de12aee163
2 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
atst/utils/form_cache.py
View File

@ -1,5 +1,6 @@
from hashlib import sha256 from hashlib import sha256
import json import json
from werkzeug.datastructures import ImmutableMultiDict
DEFAULT_CACHE_NAME = "formcache" DEFAULT_CACHE_NAME = "formcache"
@ -24,7 +25,8 @@ class FormCache(object):
def read(self, formdata_key, key_prefix=DEFAULT_CACHE_NAME): def read(self, formdata_key, key_prefix=DEFAULT_CACHE_NAME):
data = self.redis.get(self._key(key_prefix, formdata_key)) data = self.redis.get(self._key(key_prefix, formdata_key))
return json.loads(data) if data is not None else {} dict_data = json.loads(data) if data is not None else {}
return ImmutableMultiDict(dict_data)
@staticmethod @staticmethod
def _key(prefix, hash_): def _key(prefix, hash_):

6
tests/utils/test_form_cache.py
View File

@ -1,5 +1,5 @@
import pytest import pytest
from werkzeug.datastructures import ImmutableDict from werkzeug.datastructures import ImmutableMultiDict
from atst.utils.form_cache import DEFAULT_CACHE_NAME, FormCache from atst.utils.form_cache import DEFAULT_CACHE_NAME, FormCache
@ -10,13 +10,13 @@ def form_cache(app):
def test_cache_form_data(app, form_cache): def test_cache_form_data(app, form_cache):
data = ImmutableDict({"kessel_run": "12 parsecs"}) data = ImmutableMultiDict({"kessel_run": "12 parsecs"})
key = form_cache.write(data) key = form_cache.write(data)
assert app.redis.get("{}:{}".format(DEFAULT_CACHE_NAME, key)) assert app.redis.get("{}:{}".format(DEFAULT_CACHE_NAME, key))
def test_retrieve_form_data(form_cache): def test_retrieve_form_data(form_cache):
data = ImmutableDict({"class": "corellian"}) data = ImmutableMultiDict({"class": "corellian"})
key = form_cache.write(data) key = form_cache.write(data)
retrieved = form_cache.read(key) retrieved = form_cache.read(key)
assert retrieved == data assert retrieved == data