Module: Cassie::Testing

Extended by:
ActiveSupport::Concern
Defined in:
lib/cassie/testing.rb

Overview

This class provides helper methods for testing.

Defined Under Namespace

Modules: OverrideMethods

Class Method Summary collapse

Class Method Details

.cleanup! ⇒ Object

Wrap test cases as a block in this method. After the test case finishes, all tables that had data inserted into them will be truncated so that the data state will be clean for the next test case.



23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
 
# File 'lib/cassie/testing.rb', line 23

def cleanup!
  begin
    yield
  ensure
    if Thread.current[:cassie_inserted].present?
      Cassie.instance.batch do
        Thread.current[:cassie_inserted].each do |table|
          keyspace, table = table.split('.', 2)
          schema = Cassie::Schema.find(keyspace)
          schema.truncate!(table) if schema
        end
      end
      Thread.current[:cassie_inserted] = nil
    end
  end
end

.prepare! ⇒ Object

Prepare the test environment. This method must be called before running the test suite.



11
12
13
14
15
16
17
18
 
# File 'lib/cassie/testing.rb', line 11

def prepare!
  Cassie.send(:include, Cassie::Testing) unless Cassie.include?(Cassie::Testing)
  Cassie::Schema.all.each do |schema|
    schema.tables.each do |table|
      schema.truncate!(table)
    end
  end
end