forked from mirror/ledisdb
157 lines
4.5 KiB
Python
157 lines
4.5 KiB
Python
# coding: utf-8
|
|
# Test set commands
|
|
|
|
import unittest
|
|
import sys
|
|
sys.path.append('..')
|
|
|
|
import pytest
|
|
|
|
import ledis
|
|
from ledis._compat import b
|
|
from ledis import ResponseError
|
|
from util import expire_at, expire_at_seconds
|
|
|
|
l = ledis.Ledis(port=6380)
|
|
|
|
|
|
class TestCmdSet(unittest.TestCase):
|
|
def setUp(self):
|
|
pass
|
|
|
|
def tearDown(self):
|
|
l.flushdb()
|
|
|
|
def test_sadd(self):
|
|
members = set([b('1'), b('2'), b('3')])
|
|
l.sadd('a', *members)
|
|
assert l.smembers('a') == members
|
|
|
|
def test_scard(self):
|
|
l.sadd('a', '1', '2', '3')
|
|
assert l.scard('a') == 3
|
|
|
|
def test_sdiff(self):
|
|
l.sadd('a', '1', '2', '3')
|
|
assert l.sdiff('a', 'b') == set([b('1'), b('2'), b('3')])
|
|
l.sadd('b', '2', '3')
|
|
assert l.sdiff('a', 'b') == set([b('1')])
|
|
|
|
def test_sdiffstore(self):
|
|
l.sadd('a', '1', '2', '3')
|
|
assert l.sdiffstore('c', 'a', 'b') == 3
|
|
assert l.smembers('c') == set([b('1'), b('2'), b('3')])
|
|
l.sadd('b', '2', '3')
|
|
print l.smembers('c')
|
|
print "before"
|
|
assert l.sdiffstore('c', 'a', 'b') == 1
|
|
print l.smembers('c')
|
|
assert l.smembers('c') == set([b('1')])
|
|
|
|
def test_sinter(self):
|
|
l.sadd('a', '1', '2', '3')
|
|
assert l.sinter('a', 'b') == set()
|
|
l.sadd('b', '2', '3')
|
|
assert l.sinter('a', 'b') == set([b('2'), b('3')])
|
|
|
|
def test_sinterstore(self):
|
|
l.sadd('a', '1', '2', '3')
|
|
assert l.sinterstore('c', 'a', 'b') == 0
|
|
assert l.smembers('c') == set()
|
|
l.sadd('b', '2', '3')
|
|
assert l.sinterstore('c', 'a', 'b') == 2
|
|
assert l.smembers('c') == set([b('2'), b('3')])
|
|
|
|
def test_sismember(self):
|
|
l.sadd('a', '1', '2', '3')
|
|
assert l.sismember('a', '1')
|
|
assert l.sismember('a', '2')
|
|
assert l.sismember('a', '3')
|
|
assert not l.sismember('a', '4')
|
|
|
|
def test_smembers(self):
|
|
l.sadd('a', '1', '2', '3')
|
|
assert l.smembers('a') == set([b('1'), b('2'), b('3')])
|
|
|
|
def test_srem(self):
|
|
l.sadd('a', '1', '2', '3', '4')
|
|
assert l.srem('a', '5') == 0
|
|
assert l.srem('a', '2', '4') == 2
|
|
assert l.smembers('a') == set([b('1'), b('3')])
|
|
|
|
def test_sunion(self):
|
|
l.sadd('a', '1', '2')
|
|
l.sadd('b', '2', '3')
|
|
assert l.sunion('a', 'b') == set([b('1'), b('2'), b('3')])
|
|
|
|
def test_sunionstore(self):
|
|
l.sadd('a', '1', '2')
|
|
l.sadd('b', '2', '3')
|
|
assert l.sunionstore('c', 'a', 'b') == 3
|
|
assert l.smembers('c') == set([b('1'), b('2'), b('3')])
|
|
|
|
def test_sclear(self):
|
|
members = set([b('1'), b('2'), b('3')])
|
|
l.sadd('a', *members)
|
|
assert l.sclear('a') == 3
|
|
assert l.sclear('a') == 0
|
|
|
|
def test_smclear(self):
|
|
members = set([b('1'), b('2'), b('3')])
|
|
l.sadd('a', *members)
|
|
l.sadd('b', *members)
|
|
assert l.smclear('a', 'b') == 2
|
|
|
|
def test_sexpire(self):
|
|
members = set([b('1'), b('2'), b('3')])
|
|
assert l.sexpire('a', 100) == 0
|
|
l.sadd('a', *members)
|
|
assert l.sexpire('a', 100) == 1
|
|
assert l.sttl('a') <= 100
|
|
|
|
def test_sexpireat_datetime(self):
|
|
members = set([b('1'), b('2'), b('3')])
|
|
l.sadd('a', *members)
|
|
assert l.sexpireat('a', expire_at())
|
|
assert 0 < l.sttl('a') <= 61
|
|
|
|
def test_sexpireat_unixtime(self):
|
|
members = set([b('1'), b('2'), b('3')])
|
|
l.sadd('a', *members)
|
|
assert l.sexpireat('a', expire_at_seconds())
|
|
assert 0 < l.sttl('a') <= 61
|
|
|
|
def test_sexpireat_no_key(self):
|
|
assert not l.sexpireat('a', expire_at())
|
|
|
|
def test_sexpireat(self):
|
|
assert l.sexpireat('a', 1577808000) == 0
|
|
members = set([b('1'), b('2'), b('3')])
|
|
l.sadd('a', *members)
|
|
assert l.sexpireat('a', 1577808000) == 1
|
|
|
|
def test_sttl(self):
|
|
members = set([b('1'), b('2'), b('3')])
|
|
l.sadd('a', *members)
|
|
assert l.sexpire('a', 100)
|
|
assert l.sttl('a') <= 100
|
|
|
|
def test_spersist(self):
|
|
members = set([b('1'), b('2'), b('3')])
|
|
l.sadd('a', *members)
|
|
l.sexpire('a', 100)
|
|
assert l.sttl('a') <= 100
|
|
assert l.spersist('a')
|
|
assert l.sttl('a') == -1
|
|
|
|
def test_invalid_params(self):
|
|
with pytest.raises(ResponseError) as excinfo:
|
|
l.sadd("a")
|
|
assert excinfo.value.message == "invalid command param"
|
|
|
|
def test_invalid_value(self):
|
|
members = set([b('1'), b('2'), b('3')])
|
|
l.sadd('a', *members)
|
|
self.assertRaises(ResponseError, lambda: l.sexpire('a', 'a'))
|
|
|