Contents

require 'test_helper'

class CassandraObject::BatchesTest < CassandraObject::TestCase
  test 'find_each' do
    Issue.create 
    Issue.create 

    issues = []
    Issue.find_each do |issue|
      issues << issue
    end

    assert_equal Issue.all.to_set, issues.to_set
  end

  test 'find_in_batches' do
    Issue.create 
    Issue.create
    Issue.create
    
    issue_batches = []
    Issue.find_in_batches(batch_size: 2) do |issues|
      issue_batches << issues
    end
    
    assert_equal 2, issue_batches.size
    assert issue_batches.any? { |issues| issues.size == 2 }
    assert issue_batches.any? { |issues| issues.size == 1 }
  end

  test 'batch' do
    Issue.batch do
      Issue.create
      Issue.create
      assert_equal 0, Issue.count
    end

    assert_equal 2, Issue.count
  end
end

Version data entries

4 entries across 4 versions & 1 rubygems

Version	Path
gotime-cassandra_object-2.8.3	test/batches_test.rb
gotime-cassandra_object-2.8.2	test/batches_test.rb
gotime-cassandra_object-2.8.1	test/batches_test.rb
gotime-cassandra_object-2.8.0	test/batches_test.rb