spec/integration/chunker_spec.rb in lhm-1.0.0.rc2 vs spec/integration/chunker_spec.rb in lhm-1.0.0.rc3

@@ -1,31 +1,32 @@
-#
-#  Copyright (c) 2011, SoundCloud Ltd., Rany Keddo, Tobias Bielohlawek, Tobias
-#  Schmidt
-#
+# Copyright (c) 2011, SoundCloud Ltd., Rany Keddo, Tobias Bielohlawek, Tobias
+# Schmidt
 
 require File.expand_path(File.dirname(__FILE__)) + '/integration_helper'
 
 require 'lhm'
 require 'lhm/table'
 require 'lhm/migration'
 
 describe Lhm::Chunker do
   include IntegrationHelper
 
-  before(:each) { connect! }
+  before(:each) { connect_master! }
 
   describe "copying" do
     before(:each) do
       @origin = table_create(:origin)
       @destination = table_create(:destination)
       @migration = Lhm::Migration.new(@origin, @destination)
     end
 
     it "should copy 23 rows from origin to destination" do
-      23.times { |n| execute("insert into origin set common = '#{ n }'") }
-      Lhm::Chunker.new(@migration, limit = 23, connection).run
-      count_all(@destination.name).must_equal(23)
+      23.times { |n| execute("insert into origin set id = '#{ n * n + 23 }'") }
+
+      Lhm::Chunker.new(@migration, connection, { :stride => 100 }).run
+
+      slave do
+        count_all(@destination.name).must_equal(23)
+      end
     end
   end
 end
-