tests.py

# coding=utf8

import sys
import time
sys.path.insert(0, '.')
import ssdb

c = ssdb.Client()

uk_cursor = time.time() * 1000


def uk(prefix='key'):
    global uk_cursor
    uk_cursor = uk_cursor + 1
    return '%s-%d' % (prefix, uk_cursor)


def test_set():
    assert c.set(uk(), 'v') == 1


def test_setx():
    key = uk()
    assert c.setx(key, 'v', 1.2) == 1
    assert c.ttl(key) <= 1.2
    time.sleep(1.2)
    assert c.exists(key) is False


def test_expire():
    key = uk()
    assert c.set(key, 'v') == 1
    assert c.expire(key, 1.2) == 1
    assert c.ttl(key) <= 1.2
    assert c.expire(uk(), 1.1) == 0


def test_ttl():
    key = uk()
    assert c.setx(key, 'v', 1.2) == 1
    assert 0 <= c.ttl(key) <= 1.2


def test_setnx():
    key = uk()
    assert c.setnx(key, 'v') == 1
    assert c.setnx(key, 'v') == 0


def test_get():
    key = uk()
    assert c.set(key, 'v') == 1
    assert c.get(key) == 'v'


def test_getset():
    key = uk()
    assert c.set(key, 'v') == 1
    assert c.getset(key, 'val') == 'v'


def test_del():
    key = uk()
    assert c.set(key, 'v')
    assert c.delete(key) == 1
    assert c.exists(key) is False


def test_incr():
    key = uk()
    assert c.set(key, 1) == 1
    assert c.incr(key, 2) == 3
    assert c.get(key) == '3'


def test_getbit():
    key = uk()
    assert c.set(key, 'val') == 1
    assert c.getbit(key, 2) == 1


def test_setbit():
    key = uk()
    assert c.set(key, 'val') == 1
    assert c.setbit(key, 2, 0) == 1
    assert c.get(key) == 'ral'


def test_countbit():
    key = uk()
    assert c.set(key, 'val') == 1
    assert c.countbit(key) == 12


def test_exists():
    key = uk()
    assert c.set(key, 'val') == 1
    assert c.exists(key) is True
    assert c.exists(uk()) is False


def test_big_data():
    key = uk()
    val1 = '123456' * 65536
    val2 = 'hello' * 65535 * 3
    assert c.set(key, val1) == 1
    assert c.get(key) == val1
    assert c.set(key, val2) == 1
    assert c.get(key) == val2


def test_substr():
    key = uk()
    assert c.set(key, 'hello world') == 1
    assert c.substr(key) == 'hello world'
    assert c.substr(key, 6, 10) == 'world'


def test_strlen():
    key = uk()
    assert c.set(key, 'hello world') == 1
    assert c.strlen(key) == 11


def test_keys_scan_rscan():
    start = uk()
    a = uk()
    b = uk()
    assert c.set(a, 1) == 1
    assert c.set(b, 1) == 1
    assert c.keys(start, uk(), 2) == [a, b]
    assert c.scan(start, uk(), -1) == [a, '1', b, '1']
    assert c.rscan(uk(), start, -1) == [b, '1', a, '1']


def test_multi_set_get_del():
    k1, k2, k3 = [uk() for i in range(3)]
    assert c.multi_set(k1, 'v1', k2, 'v2', k3, 'v3') == 3
    assert c.multi_get(k1, k2, k3) == [k1, 'v1', k2, 'v2', k3, 'v3']
    assert c.multi_del(k1, k2, k3) == 3 == 3


def test_hset_hget_hincr_hexists():
    hsh = uk('hash')
    field = uk('field')
    assert c.hset(hsh, field, 'v') == 1
    assert c.hget(hsh, field) == 'v'
    assert c.hdel(hsh, field) == 1
    assert c.hincr(hsh, field, 3) == 3
    assert c.hexists(hsh, field) is True
    assert c.hsize(hsh) == 1


def test_hlist_hrlist():
    start = uk('hash')
    a, b = uk('hash'), uk('hash')
    assert c.hset(a, 'field', 'v')
    assert c.hset(b, 'field', 'v')
    lst = c.hlist(start, uk('hash'), -1)
    rlst = c.hrlist(uk('hash'), start, -1)
    assert lst == [a, b] == rlst[::-1]


def test_hkeys_hscan_hrscan_hgetall_hclear():
    h = uk('hash')
    a = uk('field')
    b = uk('field')
    assert c.hset(h, a, 'va') == 1
    assert c.hset(h, b, 'vb') == 1
    assert c.hkeys(h, '', '', -1) == [a, b]
    assert c.hscan(h, '', '', -1) == [a, 'va', b, 'vb']
    assert c.hrscan(h, '', '', -1) == [b, 'vb', a, 'va']
    assert c.hgetall(h, '', '', -1) == [a, 'va', b, 'vb']
    assert c.hclear(h) == 2
    assert c.hsize(h) == 0


def test_multi_hset_hget_hdel():
    h = uk('hash')
    k1, k2, k3 = [uk() for i in range(3)]
    assert c.multi_hset(h, k1, 'v1', k2, 'v2', k3, 'v3')
    assert c.multi_hget(h, k1, k2, k3) == [k1, 'v1', k2, 'v2', k3, 'v3']
    assert c.multi_hdel(h, k1, k2, k3) == 3
    assert c.hsize(h) == 0


def test_zset_zget_zdel_zincr_zexists_zsize():
    z = uk('zset')
    k = uk()
    assert c.zset(z, k, 13) == 1
    assert c.zget(z, k) == 13
    assert c.zincr(z, k, 3) == 16
    assert c.zexists(z, k) is True
    assert c.zdel(z, k) == 1
    assert c.zexists(z, k) is False
    assert c.zsize(z) == 0


def test_zkeys_zscan_zrscan_zclear():
    z = uk('zset')
    a = uk('key')
    b = uk('key')
    assert c.zset(z, a, 12581)
    assert c.zset(z, b, 12582)
    assert c.zkeys(z, '', '', '', -1) == [a, b]
    assert c.zscan(z, '', '', '', -1) == [a, '12581', b, '12582']
    assert c.zrscan(z, '', '', '', -1) == [b, '12582', a, '12581']
    assert c.zclear(z) == 2


def test_chinese_value():
    k = uk()
    assert c.set(k, '你好世界') == 1
    assert c.get(k) == '你好世界'


def test_chinese_key():
    k = uk() + '你好中国人'
    assert c.set(k, '你好世界') == 1
    assert c.get(k) == '你好世界'