def test_simple(self):
oid = self.createObject(RaleighSSet.TYPE)
sset = RaleighSSet(self.client, oid)
sset.insert('key-01', 'value-01')
sset.insert('key-02', 'value-02')
data = sset.get('key-01')
self.assertEquals(data['value'], 'value-01')
data = sset.get('key-02')
self.assertEquals(data['value'], 'value-02')
self.assertRaises(RaleighException, sset.get, 'key-03')
data = sset.scan(2)
self.assertEquals(data['keys'], ['key-01', 'key-02'])
self.assertEquals(data['values'], ['value-01', 'value-02'])
data = sset.pop('key-01')
self.assertEquals(data['value'], 'value-01')
self.assertRaises(RaleighException, sset.get, 'key-01')
data = sset.scan(2)
self.assertEquals(data['keys'], ['key-02'])
self.assertEquals(data['values'], ['value-02'])
data = sset.update('key-02', 'value-02b')
self.assertEquals(data['old_value'], 'value-02')
data = sset.get('key-02')
self.assertEquals(data['value'], 'value-02b')
def execute(self, client):
sset = RaleighSSet(client, self.oid)
nrequests = 0
exception = None
for i in xrange(self.NREQUESTS):
x = randint(0, 9999999999)
try:
nrequests += 1
data = sset.get('Key-%010d' % x)
except RaleighException as e:
pass
except Exception as e:
exception = e
break
return nrequests, exception
def test_txn_multi_obj(self):
def _sset_get(sset, key, txn_id, expected):
try:
data = sset.get(key, txn_id)
print sset, data
assert data['value'] == expected, (data['value'] == expected)
except Exception as e:
print sset, e
assert expected == None, expected
oid_1 = self.createObject(RaleighSSet.TYPE)
sset_1 = RaleighSSet(self.client, oid_1)
oid_2 = self.createObject(RaleighSSet.TYPE)
sset_2 = RaleighSSet(self.client, oid_2)
key = 'key-01'
value = 'value-01'
# Test Rollback
txn = RaleighTransaction(self.client)
txn.begin()
sset_1.insert(key, value, txn.txn_id)
sset_2.insert(key, value, txn.txn_id)
self.assertRaises(RaleighException, sset_1.get, key)
self.assertRaises(RaleighException, sset_2.get, key)
data = sset_1.get(key, txn.txn_id)
self.assertEquals(data['value'], value)
data = sset_2.get(key, txn.txn_id)
self.assertEquals(data['value'], value)
txn.rollback()
self.assertRaises(RaleighException, sset_1.get, key)
self.assertRaises(RaleighException, sset_2.get, key)
# Test Commit
txn = RaleighTransaction(self.client)
txn.begin()
sset_1.insert(key, value, txn.txn_id)
sset_2.insert(key, value, txn.txn_id)
self.assertRaises(RaleighException, sset_1.get, key)
self.assertRaises(RaleighException, sset_2.get, key)
data = sset_1.get(key, txn.txn_id)
self.assertEquals(data['value'], value)
data = sset_2.get(key, txn.txn_id)
self.assertEquals(data['value'], value)
txn.commit()
data = sset_1.get(key)
self.assertEquals(data['value'], value)
data = sset_2.get(key)
self.assertEquals(data['value'], value)
def test_txn_key_locked(self):
oid = self.createObject(RaleighSSet.TYPE)
sset = RaleighSSet(self.client, oid)
txn_1 = RaleighTransaction(self.client)
txn_1.begin()
txn_2 = RaleighTransaction(self.client)
txn_2.begin()
sset.insert('key-01', 'value-01', txn_1.txn_id)
self.assertRaises(RaleighException, sset.insert, 'key-01', 'value-02', txn_2.txn_id)
txn_2.rollback()
txn_1.commit()
data = sset.get('key-01')
self.assertEquals(data['value'], 'value-01')
