spec/mongo/retryable_spec.rb in mongo-2.18.3 vs spec/mongo/retryable_spec.rb in mongo-2.19.0

@@ -1,7 +1,7 @@
 # frozen_string_literal: true
-# encoding: utf-8
+# rubocop:todo all
 
 require 'spec_helper'
 
 class RetryableTestConsumer
   include Mongo::Retryable
@@ -44,11 +44,11 @@
       operation.execute
     end
   end
 
   def retry_write_allowed_as_configured?
-    retry_write_allowed?(session, write_concern)
+    write_worker.retry_write_allowed?(session, write_concern)
   end
 end
 
 class LegacyRetryableTestConsumer < RetryableTestConsumer
   def session
@@ -85,10 +85,12 @@
 end
 
 class RetryableHost
   include Mongo::Retryable
 
-  public :retry_write_allowed?
+  def retry_write_allowed?(*args)
+    write_worker.retry_write_allowed?(*args)
+  end
 end
 
 describe Mongo::Retryable do
 
   let(:operation) do