def test_delete_version(self):
store = MCStore(self.proxy.addr)
key = 'key1'
store.set(key, 'aaa')
self.update_pos(256)
self.assertEqual(self.get_meta(store, key), (1, 0, self.last_pos))
store.delete(key)
self.update_pos(256)
self.assertEqual(store.get(key), None)
self.assertEqual(self.get_meta(store, key), (-2, 0, self.last_pos))
self.checkCounterZero()
store.set(key, 'bbb')
self.update_pos(256)
self.assertEqual(store.get(key), 'bbb')
self.assertEqual(self.get_meta(store, key), (3, 0, self.last_pos))
self.checkCounterZero()
class AbnormalCmdTest(BaseTest):
def setUp(self):
BaseTest.setUp(self)
self.store = MCStore(self.db.addr)
self.invalid_key = '/this/is/a/bad/key/%s' % chr(15)
def run_cmd_by_telnet(self, cmd, expected, timeout=2):
addr, port = self.db.addr.split(':')
t = telnetlib.Telnet(addr, port)
t.write('%s\r\n' % cmd)
out = t.read_until('\n', timeout=timeout)
t.write('quit\n')
t.close()
r = out.strip('\r\n')
self.assertEqual(r, expected)
def test_get(self):
# get not exist key
cmd = 'get /test/get'
self.run_cmd_by_telnet(cmd, 'END')
# invalid key
cmd = 'get %s' % self.invalid_key
self.run_cmd_by_telnet(cmd, 'END')
self.checkCounterZero()
def test_set(self):
# invalid key
cmd = 'set %s 0 0 3\r\naaa' % self.invalid_key
self.run_cmd_by_telnet(cmd, 'NOT_STORED')
cmd = 'set /test/set 0 0 3\r\naaaa'
self.run_cmd_by_telnet(cmd, 'CLIENT_ERROR bad data chunk')
self.checkCounterZero()
def test_incr(self):
key = '/test/incr'
self.assertEqual(self.store.delete(key), True)
cmd = 'incr %s 10' % key
self.run_cmd_by_telnet(cmd, '10')
self.assertEqual(self.store.get(key), 10)
# incr 一个 value 为字符串的 key
key = '/test/incr2'
self.assertEqual(self.store.set(key, 'aaa'), True)
cmd = 'incr %s 10' % key
self.run_cmd_by_telnet(cmd, '0')
self.assertEqual(self.store.get(key), 'aaa')
self.checkCounterZero()
def test_delete(self):
key = '/delete/not/exist/key'
cmd = 'delete %s' % key
self.run_cmd_by_telnet(cmd, 'NOT_FOUND')
cmd = 'delete %s' % self.invalid_key
self.run_cmd_by_telnet(cmd, 'NOT_FOUND')
self.checkCounterZero()
def test_get_meta_by_key(self):
key = '/get_meta_by_key/not/exist/key'
cmd = 'get ?%s' % key
self.run_cmd_by_telnet(cmd, 'END')
cmd = 'get ?%s' % self.invalid_key
self.run_cmd_by_telnet(cmd, 'END')
self.checkCounterZero()