def setUp(self):
self.client = Client(['127.0.0.1:9042'], keyspace=TEST_KEYSPACE, columnfamily=TEST_COLUMNFAMILY)
def setUp(self):
self.client = Client(['127.0.0.1:9042'],
keyspace=TEST_KEYSPACE,
columnfamily=TEST_COLUMNFAMILY)
class CasscacheTests(unittest.TestCase):
def setUp(self):
self.client = Client(['127.0.0.1:9042'], keyspace=TEST_KEYSPACE, columnfamily=TEST_COLUMNFAMILY)
def tearDown(self):
self.client.disconnect_all()
cluster = Cluster()
session = cluster.connect()
session.execute("TRUNCATE %s.%s" % (TEST_KEYSPACE, TEST_COLUMNFAMILY))
cluster.shutdown()
def test_get_not_found(self):
self.assertIsNone(self.client.get('lol'))
def test_set(self):
self.assertEqual(self.client.set('lol', 'derp'), 1)
def test_get_found(self):
self.client.set('lol', 'derp')
self.assertEqual(self.client.get('lol'), 'derp')
def test_set_get_multi(self):
keys = map(lambda n: 'lol%d' % n, range(0, 5))
mapping = dict((key, 'derp') for key in keys)
self.client.set_multi(mapping)
self.assertEqual(self.client.get_multi(keys), mapping)
def test_set_prefix(self):
self.client.set_multi({'lol': 'derp'}, key_prefix='!')
self.assertEqual(self.client.get('!lol'), 'derp')
def test_get_prefix(self):
self.client.set('!lol', 'derp')
self.assertEqual(self.client.get_multi(['lol'], key_prefix='!'), {'lol': 'derp'})
def test_delete(self):
self.client.set('lol', 'derp')
self.client.delete('lol')
self.assertIsNone(self.client.get('lol'))
def test_delete_multi(self):
self.client.set('lol', 'derp')
self.client.set('lol2', 'derp')
self.client.delete_multi(['lol', 'lol2'])
self.assertEqual(self.client.get_multi(['lol', 'lol2']), {})
def test_marshal_int(self):
self.client.set('lol', 1)
res = self.client.get('lol')
self.assertIsInstance(res, int)
self.assertEqual(res, 1)
@unittest.skipIf(sys.version_info[0] == 3, "python3 doesn't have a long type")
def test_marshal_long(self):
self.client.set('lol', long(1))
res = self.client.get('lol')
self.assertIsInstance(res, long)
self.assertEqual(res, long(1))
def test_marshal_pickle(self):
self.client.set('lol', [1, 2])
self.assertEqual(self.client.get('lol'), [1, 2])
def test_flush_all(self):
self.client.set('lol', 'derp')
self.client.flush_all()
self.assertIsNone(self.client.get('lol'))
def test_set_and_expire(self):
self.client.set('lol', 'derp', -1)
self.assertIsNone(self.client.get('lol'))
class CasscacheTests(unittest.TestCase):
def setUp(self):
self.client = Client(['127.0.0.1:9042'],
keyspace=TEST_KEYSPACE,
columnfamily=TEST_COLUMNFAMILY)
def tearDown(self):
self.client.disconnect_all()
cluster = Cluster()
session = cluster.connect()
session.execute("TRUNCATE %s.%s" % (TEST_KEYSPACE, TEST_COLUMNFAMILY))
cluster.shutdown()
def test_get_not_found(self):
self.assertIsNone(self.client.get('lol'))
def test_set(self):
self.assertEqual(self.client.set('lol', 'derp'), 1)
def test_get_found(self):
self.client.set('lol', 'derp')
self.assertEqual(self.client.get('lol'), 'derp')
def test_set_get_multi(self):
keys = map(lambda n: 'lol%d' % n, range(0, 5))
mapping = dict((key, 'derp') for key in keys)
self.client.set_multi(mapping)
self.assertEqual(self.client.get_multi(keys), mapping)
def test_set_prefix(self):
self.client.set_multi({'lol': 'derp'}, key_prefix='!')
self.assertEqual(self.client.get('!lol'), 'derp')
def test_get_prefix(self):
self.client.set('!lol', 'derp')
self.assertEqual(self.client.get_multi(['lol'], key_prefix='!'),
{'lol': 'derp'})
def test_delete(self):
self.client.set('lol', 'derp')
self.client.delete('lol')
self.assertIsNone(self.client.get('lol'))
def test_delete_multi(self):
self.client.set('lol', 'derp')
self.client.set('lol2', 'derp')
self.client.delete_multi(['lol', 'lol2'])
self.assertEqual(self.client.get_multi(['lol', 'lol2']), {})
def test_marshal_int(self):
self.client.set('lol', 1)
res = self.client.get('lol')
self.assertIsInstance(res, int)
self.assertEqual(res, 1)
@unittest.skipIf(sys.version_info[0] == 3,
"python3 doesn't have a long type")
def test_marshal_long(self):
self.client.set('lol', long(1))
res = self.client.get('lol')
self.assertIsInstance(res, long)
self.assertEqual(res, long(1))
def test_marshal_pickle(self):
self.client.set('lol', [1, 2])
self.assertEqual(self.client.get('lol'), [1, 2])
def test_flush_all(self):
self.client.set('lol', 'derp')
self.client.flush_all()
self.assertIsNone(self.client.get('lol'))
def test_set_and_expire(self):
self.client.set('lol', 'derp', -1)
self.assertIsNone(self.client.get('lol'))
