Replace with_multi_batch_size with stub_const

test/models/solid_cache/entry_test.rb

@@ -71,9 +71,7 @@ class EntryTest < ActiveSupport::TestCase
     end
 
     test "batching multi queries" do
-      with_multi_batch_size(2) do
-        Entry.stubs(:const_get).with(:MULTI_BATCH_SIZE).returns("stubbed_value")
-
+      stub_const(Entry, :MULTI_BATCH_SIZE, 2) do
         assert_queries_count(2) do
           Entry.write_multi([ { key: "hello".b, value: "there" }, { key: "foo".b, value: "bar" }, { key: "baz".b, value: "zab" } ])
         end
@@ -88,15 +86,5 @@ class EntryTest < ActiveSupport::TestCase
       def write_entries(count = 20)
         Entry.write_multi(count.times.map { |i| { key: "key#{i}", value: "value#{i}" } })
       end
-
-      def with_multi_batch_size(value)
-        old_value = Entry::MULTI_BATCH_SIZE
-        Entry.send(:remove_const, :MULTI_BATCH_SIZE)
-        Entry.const_set(:MULTI_BATCH_SIZE, value)
-        yield
-      ensure
-        Entry.send(:remove_const, :MULTI_BATCH_SIZE)
-        Entry.const_set(:MULTI_BATCH_SIZE, old_value)
-      end
   end
 end