test/test_helper.rb in cassilds-0.9.2 vs test/test_helper.rb in cassilds-0.12.1

@@ -1,14 +1,15 @@
+CASSANDRA_VERSION = ENV['CASSANDRA_VERSION'] || '0.8' unless defined?(CASSANDRA_VERSION)
 
 require 'test/unit'
-require "#{File.expand_path(File.dirname(__FILE__))}/../lib/cassandra"
+require "#{File.expand_path(File.dirname(__FILE__))}/../lib/cassandra/#{CASSANDRA_VERSION}"
 begin; require 'ruby-debug'; rescue LoadError; end
 
 begin
   @test_client = Cassandra.new('Twitter', 'localhost:9160', {:exception_classes => []})
 rescue Thrift::TransportException => e
   #FIXME Make server automatically start if not running
   if e.message =~ /Could not connect/
-    puts "*** Please start the Cassandra server by running 'rake cassandra'. ***" 
+    puts "*** Please start the Cassandra server by running 'rake cassandra'. ***"
     exit 1
   end
 end