在测试之间擦除Cassandra DB(Rspec)

时间:2012-10-04 20:14:14

标签: ruby testing rspec cassandra

我正在运行涉及Cassandra数据库中的数据的rspec测试。在测试之间擦除/清理数据库的最佳做法是什么?对于我的mongo数据,我正在使用DatabaseCleaner,我正在寻找一个等效的Cassandra。我目前正在我的spec_helper.rb中执行以下操作,但它非常慢,所以我正在寻找更好的解决方案。谢谢!

config.before :each do
    ['column1', 'column2'].each do |name|
      begin
        $cassandra.drop_column_family(name)
      rescue
        next
      ensure
        cf = Cassandra::ColumnFamily.new(keyspace: 'db_name', name: name, comparator_type: 'TimeUUIDType')
        $cassandra.add_column_family(cf)
      end
    end

2 个答案:

答案 0 :(得分:3)

老问题,但我找到了有用的要点,也许它也会帮助其他人(我正在使用cequel gem来访问Cassandra):

https://gist.github.com/elado/c95a4ffa952809865ee8

# in spec_helper.rb

RSpec.configure do |config|
  records = []

  config.before :suite do
    Cequel::Record.descendants.each do |klass|
      klass.after_create {|r| records << r }
    end
  end

  config.after :each do
    records.each(&:destroy)
    records.clear
  end

  def clean_cequel!
    Cequel::Record.descendants.each { |klass| Cequel::Record.connection.schema.truncate_table(klass.table_name) }
  end

  config.before :suite do
    clean_cequel!
  end

  config.after :suite do
    clean_cequel!
  end
end

答案 1 :(得分:1)

尝试截断,禁用autoSnapshot中的cassandra.yaml并禁用密钥空间中的durable_writes。

截断过去很慢,但自1.1.1以来这是固定的:https://issues.apache.org/jira/browse/CASSANDRA-4153