Skip to content

Instantly share code, notes, and snippets.

@mgwilliams
Created September 20, 2011 17:54
Show Gist options
  • Save mgwilliams/1229789 to your computer and use it in GitHub Desktop.
Save mgwilliams/1229789 to your computer and use it in GitHub Desktop.
@defer.inlineCallbacks
def setUp(self):
self.cmanager = ManagedCassandraClientFactory(keyspace='system')
self.client = CassandraClient(self.cmanager)
for i in xrange(CONNS):
reactor.connectTCP(HOST, PORT, self.cmanager)
yield self.cmanager.deferred
self.my_keyspace = KsDef(
name=KEYSPACE,
strategy_class='org.apache.cassandra.locator.SimpleStrategy',
replication_factor=1,
cf_defs=[
CfDef(
keyspace=KEYSPACE,
name=CF,
column_type='Standard'
),
]
)
try:
yield self.client.system_drop_keyspace(self.my_keyspace.name)
except:
pass
yield self.client.system_add_keyspace(self.my_keyspace)
yield self.client.set_keyspace(KEYSPACE)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment