async def test_create_client(event_loop):
c = Client(loop=event_loop)
assert c._pool is None
pool = await c.get_pool()
assert pool is c._pool
assert c._pool.maxsize == c.max_connection
await c.close()
assert c._pool is None
async def test_execute_command(event_loop):
c = Client(loop=event_loop)
await c.set('a', 1)
res = await c.get('a')
assert res[0] == '1'
assert c._pool is not None
await c.close()
assert c._pool is None
async def test_aiossdb_kv(event_loop):
"""
"""
logger = logging.getLogger("aiossdb_kv")
aio_cli = Client(host="0.0.0.0", port=38888, loop=event_loop)
await aio_cli.set("test_kv", 1234567890)
res = await aio_cli.get("test_kv")
logger.debug(res)
await aio_cli.close()
assert res == b"1234567890"
async def test_aiossdb_dbsize(event_loop):
"""
:Keyword Arguments:
event_loop --
:return: None
"""
aio_cli = Client(host="0.0.0.0", port=38888, loop=event_loop)
res = await aio_cli.dbsize()
assert type(res) == int
await aio_cli.close()
async def test_aiossdb_qpop_front(event_loop):
"""
:Keyword Arguments:
event_loop --
:return: None
"""
aio_cli = Client(host="0.0.0.0", port=38888, loop=event_loop)
res = await aio_cli.qpop_front("non-exists", 5)
assert res == []
await aio_cli.close()
async def test_aiossdb_zset(event_loop):
"""
Keyword Arguments:
event_loop --
"""
logger = logging.getLogger("aiossdb_zset")
aio_cli = Client(host="0.0.0.0", port=38888, loop=event_loop)
await aio_cli.zset("aiossdb_zset", "aiossdb_test", 123)
res = await aio_cli.zget("aiossdb_zset", "aiossdb_test")
logger.debug(res)
await aio_cli.close()
assert res == 123
async def test_aiossdb_queue_mult(event_loop):
"""
Keyword Arguments:
event_loop --
"""
logger = logging.getLogger("aiossdb_queue")
aio_cli = Client(host="0.0.0.0", port=38888, loop=event_loop)
await aio_cli.qpush_front("aiossdb_queue", 123, 456, 789)
res = await aio_cli.qpop_front("aiossdb_queue", 3)
logger.debug(res)
await aio_cli.close()
assert res[0].decode() == "789"
async def test_aiossdb_keys(event_loop):
"""
:Keyword Arguments:
event_loop --
:return: None
"""
aio_cli = Client(host="0.0.0.0", port=38888, loop=event_loop)
res = await aio_cli.keys("A", "F", 1)
# res = await aio_cli.keys("A", "F", 5)
# assert res == [b'B', b'C', b'D', b'E', b'F']
assert res == [b'B']
await aio_cli.close()
async def test_aiossdb_queue(event_loop):
"""
Keyword Arguments:
event_loop --
"""
aio_cli = Client(host="0.0.0.0", port=38888, loop=event_loop)
await aio_cli.qpush_front("aiossdb_queue", 123)
await aio_cli.qpush_back("aiossdb_queue", 456)
qsize = await aio_cli.qsize("aiossdb_queue")
assert qsize == 2
res = await aio_cli.qpop_front("aiossdb_queue", 1)
assert res == b"123"
await aio_cli.qpop_front("aiossdb_queue", 1)
await aio_cli.close()
async def test_aiossdb_hashmd5(event_loop):
"""
:Keyword Arguments:
event_loop --
:return: None
"""
aio_cli = Client(host="0.0.0.0", port=38888, loop=event_loop)
src = "pretend_to_be_function"
m1 = hashlib.md5()
m1.update(src.encode())
md5_str = m1.hexdigest()
await aio_cli.hset(md5_str + "%$#@_xxx", md5_str + "%$#@_xxx" + "123", 123)
res = await aio_cli.hget(md5_str + "%$#@_xxx",
md5_str + "%$#@_xxx" + "123")
assert res == b'123'
await aio_cli.close()
async def test_aiossdb_hash(event_loop):
"""
Keyword Arguments:
event_loop --
"""
logger = logging.getLogger("aiossdb_hash")
aio_cli = Client(host="0.0.0.0", port=38888, loop=event_loop)
await aio_cli.hset("aiossdb_hash", "aiossdb_test", 123)
res = await aio_cli.hget("aiossdb_hash", "aiossdb_test")
logger.debug(res)
assert res == b"123"
await aio_cli.hset("aiossdb_hash", "aiossdb_test1", 123)
res1 = await aio_cli.hgetall("aiossdb_hash")
assert res1 == [b'aiossdb_test', b'123', b'aiossdb_test1', b'123']
hsize = await aio_cli.hsize("aiossdb_hash")
assert hsize == 2
await aio_cli.close()
async def test_aiossdb_multi_zset(event_loop):
"""
:Keyword Arguments:
event_loop --
:return: None
"""
logger = logging.getLogger("aiossdb_multi_zset")
aio_cli = Client(host="0.0.0.0", port=38888, loop=event_loop)
test_list = ['key1', 1, 'key2', 2]
await aio_cli.multi_zset("aiossdb_multi", *test_list)
get_list = ['key1', 'key2']
res = await aio_cli.multi_zget("aiossdb_multi", *get_list)
assert res == [b'key1', b'1', b'key2', b'2']
zsize = await aio_cli.zsize("aiossdb_multi")
assert zsize == 2
await aio_cli.close()