1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55
|
from __future__ import unicode_literals
import pytest
import redis
from redis._compat import unichr, unicode
from .conftest import _get_client
class TestEncoding(object):
@pytest.fixture()
def r(self, request):
return _get_client(redis.Redis, request=request, decode_responses=True)
def test_simple_encoding(self, r):
unicode_string = unichr(3456) + 'abcd' + unichr(3421)
r['unicode-string'] = unicode_string
cached_val = r['unicode-string']
assert isinstance(cached_val, unicode)
assert unicode_string == cached_val
def test_list_encoding(self, r):
unicode_string = unichr(3456) + 'abcd' + unichr(3421)
result = [unicode_string, unicode_string, unicode_string]
r.rpush('a', *result)
assert r.lrange('a', 0, -1) == result
class TestCommandsAndTokensArentEncoded(object):
@pytest.fixture()
def r(self, request):
return _get_client(redis.Redis, request=request, encoding='utf-16')
def test_basic_command(self, r):
r.set('hello', 'world')
class TestInvalidUserInput(object):
def test_boolean_fails(self, r):
with pytest.raises(redis.DataError):
r.set('a', True)
def test_none_fails(self, r):
with pytest.raises(redis.DataError):
r.set('a', None)
def test_user_type_fails(self, r):
class Foo(object):
def __str__(self):
return 'Foo'
def __unicode__(self):
return 'Foo'
with pytest.raises(redis.DataError):
r.set('a', Foo())
|