def setUp(self):
logging.basicConfig(level=logging.WARN, \
format='%(asctime)s %(levelname)s [%(funcName)s] %(message)s')
clist = ['amoeba.ucsd.edu:9160']
self.ds = SetCassandraStore(cass_host_list=clist)
self.key = self._mkey()
self.value = self._mkey()
self.value2 = self._mkey()
class SetDataStoreInterfaceTest(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 = SetCassandraStore(cass_host_list=clist)
self.key = self._mkey()
self.value = self._mkey()
self.value2 = 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
storeset = self.ds.smembers(self.key)
self.failUnlessEqual(storeset, None)
def test_sadd(self):
# Hmm, simplest op, just looking for exceptions
self.ds.sadd(self.key, self.value)
def test_sremove(self):
self.ds.sadd(self.key, self.value)
self.ds.sremove(self.key, self.value)
storeset = self.ds.smembers(self.key)
self.failUnlessEqual(storeset, None)
def test_scard(self):
self.ds.sadd(self.key, self.value)
card = self.ds.scard(self.key)
self.failUnlessEqual(card, 1)
def test_sadd_scard(self):
"""
Ensures that the set is a collection of unique elements
Add the same element twice to the set, check to see that
cardinality is one.
"""
self.ds.sadd(self.key, self.value)
self.ds.sadd(self.key, self.value)
card = self.ds.scard(self.key)
self.failUnlessEqual(card, 1)
def test_smembers(self):
"""
Adds two elements to the set then retrieves them.
"""
self.ds.sadd(self.key, self.value)
self.ds.sadd(self.key, self.value2)
members = self.ds.smembers(self.key)
self.failUnlessEqual(members, set([self.value, self.value2]))
