mirror of
https://github.com/MISP/misp-galaxy.git
synced 2024-11-30 02:37:17 +00:00
77 lines
2.1 KiB
Python
77 lines
2.1 KiB
Python
#!/usr/bin/env python
|
|
# encoding: utf-8
|
|
|
|
|
|
"""
|
|
@author: william
|
|
@contact: 1342247033@qq.com
|
|
@site: http://www.xiaolewei.com
|
|
@file: test_cache.py
|
|
@time: 22/05/2018 21:17
|
|
"""
|
|
import random
|
|
import string
|
|
import time
|
|
from ddt import ddt, data, unpack
|
|
|
|
|
|
@ddt
|
|
class TestData(object):
|
|
|
|
@staticmethod
|
|
def key(length=5):
|
|
letters = string.ascii_lowercase
|
|
return ''.join(random.choice(letters) for i in range(length))
|
|
|
|
@data(0, 1, 13423, '', 'hello', {}, {'hello': 'world'})
|
|
def test_cache(self, value):
|
|
self.driver.put('key', value)
|
|
self.assertEqual(value, self.driver.get('key'))
|
|
|
|
@data((0, 1), (1, 2), (1321, 1), ('', 2), ('hello', 2), ({}, 1), ({'hello': 'world'}, 2))
|
|
@unpack
|
|
def test_ttl_cache(self, value, ttl):
|
|
key = self.key()
|
|
self.driver.put(key, value, ttl)
|
|
time.sleep(ttl / 2)
|
|
self.assertEqual(value, self.driver.get(key))
|
|
time.sleep(ttl / 2 + 1)
|
|
self.assertEqual(None, self.driver.get(key))
|
|
|
|
@data((0, 1), (3, 0), (-1, 1), (4, -4))
|
|
@unpack
|
|
def test_incr_decr(self, old, amount):
|
|
key = self.key()
|
|
self.driver.put(key, old)
|
|
self.driver.increase(key, amount)
|
|
self.assertEqual(old + amount, self.driver.get(key))
|
|
self.driver.decrease(key, amount)
|
|
self.assertEqual(old, self.driver.get(key))
|
|
self.driver.decrease(key, amount)
|
|
self.assertEqual(old - amount, self.driver.get(key))
|
|
|
|
@data(0, 1, 13423, '', 'hello', {}, {'hello': 'world'})
|
|
def test_forever(self, value):
|
|
self.driver.forever('key', value)
|
|
self.assertEqual(value, self.driver.get('key'))
|
|
|
|
@data(0, 1, 13423, '', 'hello', {}, {'hello': 'world'})
|
|
def test_forget(self, value):
|
|
key = self.key()
|
|
self.driver.put(key, value)
|
|
self.driver.forget(key)
|
|
self.assertEqual(None, self.driver.get(key))
|
|
|
|
@data(0, 1, 13423, '', 'hello', {}, {'hello': 'world'})
|
|
def test_flush(self, value):
|
|
key = self.key()
|
|
self.driver.put(key, value)
|
|
self.driver.flush()
|
|
self.assertEqual(None, self.driver.get(key))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|