mgwilliams · September 20, 2011 17:54 · Sep 20, 2011 · Sep 20, 2011
diff --git a/gistfile1.py b/gistfile1.py
@@ -1,10 +1,27 @@
-from twisted.application.internet import TCPClient
-from telephus.protocol import ManagedCassandraClientFactory
-from telephus.client import CassandraClient
+    @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
 
-class Service(TCPClient):
-
-    def __init__(self, keyspace=None, host="localhost", port=9160, delegate=None):
-        f = ManagedCassandraClientFactory(keyspace)
-        delegate.set_client(CassandraClient(f))
-        TCPClient.__init__(self, host, port, f)
+        yield self.client.system_add_keyspace(self.my_keyspace)
+        yield self.client.set_keyspace(KEYSPACE)
diff --git a/gistfile1.py b/gistfile1.py
@@ -0,0 +1,10 @@
+from twisted.application.internet import TCPClient
+from telephus.protocol import ManagedCassandraClientFactory
+from telephus.client import CassandraClient
+
+class Service(TCPClient):
+
+    def __init__(self, keyspace=None, host="localhost", port=9160, delegate=None):
+        f = ManagedCassandraClientFactory(keyspace)
+        delegate.set_client(CassandraClient(f))
+        TCPClient.__init__(self, host, port, f)