예제 #1
0
 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())
예제 #2
0
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)