class RedisworksTestCase(unittest.TestCase):
"""RedisWorks Tests."""
def setUp(self):
self.root = Root(redis=FakeStrictRedis)
self.red = FakeStrictRedis()
def tearDown(self):
# Clear data in fakeredis.
self.red.flushall()
def test_nested_format(self):
value = {1: 1, 2: {"a": "hello"}}
result = Root.doformat(value)
int_str = bTYPE_IDENTIFIER + b'num' + bITEM_DIVIDER + b'int' + bITEM_DIVIDER
expected_result = {int_str + b'1': int_str + b'1',
int_str + b'2': bTYPE_IDENTIFIER +
b'dict' + bITEM_DIVIDER +
b'dict' + bITEM_DIVIDER + b'{"a": "hello"}'}
self.assertEqual(result, expected_result)
def test_numbers(self):
today = datetime.date.today()
now = datetime.datetime.utcnow()
items = (10, 10.1, Decimal("10"), 10+1j, today, now)
for val in items:
self.root.part = val
result = self.red.get('root.part')
expected_result = Root.doformat(val)
self.assertEqual(result, expected_result)
# flushing dotobject local cache
self.root.flush()
self.assertEqual(self.root.part, val)
def test_return_string(self):
root2 = Root(redis=FakeStrictRedis, return_object=False)
val = 11.1
expected_result = b'11.1'
root2.part = val
# flushing dotobject local cache
root2.flush()
self.assertEqual(root2.part, expected_result)
def test_grandchild(self):
string = "for real?"
self.root.haha.wahaha = string
result = self.red.get('root.haha.wahaha')
expected_result = Root.doformat(string)
self.assertEqual(result, expected_result)
self.root.flush()
self.assertEqual(self.root.haha.wahaha, string)
self.assertEqual(self.root['haha.wahaha'], string)
def test_child_set(self):
value = {1, 2, 4}
expected_result = set(Root.doformat(value))
self.root.part_set = value
result = self.red.smembers('root.part_set')
self.assertEqual(result, expected_result)
self.root.flush()
self.assertEqual(self.root.part_set, value)
def test_child_dict(self):
value = {1: 1, 2: 2, 3: 4}
expected_result = Root.doformat(value)
self.root.part_dic = value
result = self.red.hgetall('root.part_dic')
self.assertEqual(result, expected_result)
self.root.flush()
self.assertEqual(self.root.part_dic, value)
def test_child_nested_dict(self):
value = {1: 1, 2: {"a": "hello"}, 3: 4}
expected_result = Root.doformat(value)
self.root.part = value
result = self.red.hgetall('root.part')
self.assertEqual(result, expected_result)
self.root.flush()
self.assertEqual(self.root.part, value)
def test_child_iterable(self):
value = [1, 3, "a"]
expected_result = Root.doformat(value)
self.root.part = value
result = self.red.lrange("root.part", 0, -1)
self.assertEqual(result, expected_result)
self.root.flush()
self.assertEqual(self.root.part, value)
def test_child_nested_iterable(self):
value = [1, 3, ["a", 3]]
expected_result = Root.doformat(value)
self.root.part = value
result = self.red.lrange("root.part", 0, -1)
self.assertEqual(result, expected_result)
self.root.flush()
self.assertEqual(self.root.part, value)
def test_many_different_children_types(self):
set_value = {1, 2, 4}
self.root.part_set = set_value
dict_value = {1: 1, 2: {"a": 1}}
self.root.part_dict = dict_value
list_value = [1, ["b", 3]]
self.root.part_list = list_value
self.root.flush()
self.assertEqual(self.root.part_set, set_value)
self.assertEqual(self.root.part_dict, dict_value)
self.assertEqual(self.root.part_list, list_value)
def test_many_different_children_types2(self):
mylist = [1, 3, 4]
self.root.my.list = mylist
some_date = datetime.datetime(2016, 8, 22, 10, 3, 19)
self.root.time = some_date
mydict = {1: 1, "a": "b"}
self.root.the.mapping.example = mydict
self.root.flush()
self.assertEqual(self.root.my.list, mylist)
self.assertEqual(self.root.time, some_date)
self.assertEqual(self.root.the.mapping.example, mydict)
def test_change_key_type(self):
mylist = [1, 3, 4]
self.root.something = mylist
st = "string"
self.root.something = st
self.assertEqual(self.root.something, st)
def test_number_comparison(self):
self.root.num = 10
num = self.root.num
self.assertIsInstance(num, LazyDot)
self.assertTrue(num > 8)
self.assertTrue(num < 11)
self.assertTrue(num <= 12)
self.assertTrue(num >= 10)
self.assertFalse(num > 10)
def test_number_math(self):
self.root.num = 10
num = self.root.num
self.assertIsInstance(num, LazyDot)
self.assertEqual(num * 2, 20)
def test_saving_set_removed_the_old_one(self):
self.root.myset = {1, 2, 3}
self.root.myset = {4, 5}
self.root.flush()
self.assertEqual(self.root.myset, {4, 5})
def test_saving_list_removed_the_old_one(self):
self.root.myset = [1, 2, 3]
self.root.myset = [4, 5]
self.root.flush()
self.assertEqual(self.root.myset, [4, 5])
