added disable indexes while restoring unit tests

Author: aishek

spec/lib/dump_rake/dump_reader_spec.rb

@@ -490,6 +490,14 @@ def create_entry(rows_count)
           expect(@dump).to receive(:insert_into_table).with('`first`', '(`id`, `name`)', @rows.map(&:inspect))
           @dump.read_table('first', 100)
         end
+
+        it "should remove indexes around reading/writing table" do
+          create_entry(100)
+          allow(@dump).to receive(:clear_table)
+          allow(@dump).to receive(:insert_into_table)
+          expect(@dump).to receive(:with_disabled_indexes).with('first')
+          @dump.read_table('first', 100)
+        end
       end
     end
 

spec/lib/dump_rake/table_manipulation_spec.rb

@@ -48,6 +48,59 @@
     end
   end
 
+  describe "with_disabled_indexes" do
+    it "should call ActiveRecord::Base.connection.indexes for table" do
+      expect(ActiveRecord::Base.connection).to receive(:indexes).with('table').and_return([])
+      with_disabled_indexes 'table' do
+      end
+    end
+
+    it "should remove indexes first add indexes then" do
+      expect(ActiveRecord::Base.connection).to receive(:indexes).with('table').and_return([])
+      expect(self).to receive(:remove_indexes).with([]).ordered
+      expect(self).to receive(:add_indexes).with([]).ordered
+      with_disabled_indexes 'table' do
+      end
+    end
+  end
+
+  describe "remove_indexes" do
+    it "should call ActiveRecord::Base.connection.remove_index for each passed index" do
+      indexes = [
+        OpenStruct.new(:table => 'table1', :name => 'index1'),
+        OpenStruct.new(:table => 'table2', :name => 'index2')
+      ]
+      expect(ActiveRecord::Base.connection).to receive(:remove_index).with('table1', { :name => 'index1' }).ordered
+      expect(ActiveRecord::Base.connection).to receive(:remove_index).with('table2', { :name => 'index2' }).ordered
+      remove_indexes indexes
+    end
+  end
+
+  describe "index_options" do
+    it "should return non-empty valid index options" do
+      index = OpenStruct.new(
+        :unique => nil,
+        :name => 'index1',
+        :unknown_key => 1,
+
+        :members => [:unique, :order, :name, :where, :length, :internal, :using, :algorithm, :type]
+      )
+      expect(index_options(index)).to eq(:name => 'index1')
+    end
+  end
+
+  describe "add_indexes" do
+    it "should call ActiveRecord::Base.connection.add_index for each passed index" do
+      indexes = [
+        OpenStruct.new(:table => 'table1', :columns => [:column1], :name => 'index1', :unknown_key => 1, :members => [:name]),
+        OpenStruct.new(:table => 'table2', :columns => [:column2], :name => 'index2', :unknown_key => 1, :members => [:name])
+      ]
+      expect(ActiveRecord::Base.connection).to receive(:add_index).with('table1', [:column1], { :name => 'index1' }).ordered
+      expect(ActiveRecord::Base.connection).to receive(:add_index).with('table2', [:column2], { :name => 'index2' }).ordered
+      add_indexes indexes
+    end
+  end
+
   describe "insert_into_table" do
     it "should call ActiveRecord::Base.connection.insert with sql for insert if values is string" do
       expect(ActiveRecord::Base.connection).to receive(:insert).with("INSERT INTO `table` (`c1`,`c2`) VALUES (`v1`,`v2`)", anything)