def setUp(self):
logging.basicConfig(level=logging.WARN, \
format='%(asctime)s %(levelname)s [%(funcName)s] %(message)s')
clist = ['amoeba.ucsd.edu:9160']
self.ds = CassandraStore(cass_host_list=clist)
self.key = self._mkey()
self.value = self._mkey()
self.dict = {'column1': 'val3', 'column2': 'val4'}
self.set = set()
self.set.add(self._mkey())
self.set.add(self._mkey())
self.set.add(self._mkey())
class TechInterfaceTest(unittest.TestCase):
def setUp(self):
logging.basicConfig(level=logging.WARN, \
format='%(asctime)s %(levelname)s [%(funcName)s] %(message)s')
clist = ['amoeba.ucsd.edu:9160']
self.ds = CassandraStore(cass_host_list=clist)
self.key = self._mkey()
self.value = self._mkey()
self.dict = {'column1': 'val3', 'column2': 'val4'}
self.set = set()
self.set.add(self._mkey())
self.set.add(self._mkey())
self.set.add(self._mkey())
def tearDown(self):
self.ds.delete(self.key)
del self.ds
def _mkey(self):
# Generate a pseudo-random string. handy, that.
return str(uuid4())
def test_get_404(self):
# Make sure we can't read the not-written
rc = self.ds.get(self.key)
self.failUnlessEqual(rc, None)
def test_write_and_delete(self):
# Hmm, simplest op, just looking for exceptions
self.ds.put(self.key, self.value)
def test_delete(self):
self.ds.put(self.key, self.value)
self.ds.delete(self.key)
rc = self.ds.get(self.key)
self.failUnlessEqual(rc, None)
def test_val_put_get_delete(self):
# Write, then read to verify same
self.ds.put(self.key, self.value)
b = self.ds.get(self.key)
self.failUnlessEqual(self.value, b)
def test_dict_put_get_delete(self):
# Write the dict, then read to verify the same
self.ds.put(self.key,self.dict)
b = self.ds.get(self.key)
self.failUnlessEqual(self.dict, b)
def test_set_put_get_delete(self):
# Write the dict, then read to verify the same
self.ds.put(self.key,self.set)
b = self.ds.get(self.key)
self.failUnlessEqual(self.set, b)
def test_incr(self):
a=self.ds.incr(self.key)
self.failUnlessEqual(1, a)
a=self.ds.incr(self.key)
self.failUnlessEqual(2, a)
def test_query(self):
# Write a key, query for it, verify contents
self.ds.put(self.key, self.value)
rl = self.ds.query(self.key)
self.failUnlessEqual(rl[0][0], self.key)