RUBY-1954 Integration test for exceptions referencing second attempt's server for retryable operations

Author: p

spec/integration/retryable_errors_spec.rb

@@ -0,0 +1,80 @@
+require 'spec_helper'
+
+describe 'Retryable writes tests' do
+
+  let(:client) do
+    authorized_client.with(retry_writes: true)
+  end
+
+  let(:collection) do
+    client['retryable-writes-error-spec']
+  end
+
+  context 'when retry fails' do
+    require_topology :replica_set
+
+    let(:fail_point_command) do
+      {
+        configureFailPoint: 'failCommand',
+        mode: {times: 1},
+        data: {
+          failCommands: ['find'],
+          errorCode: 11600,
+        },
+      }
+    end
+
+    let(:clear_fail_point_command) do
+      {
+        configureFailPoint: 'failCommand',
+        mode: 'off',
+      }
+    end
+
+    after do
+      ClusterTools.instance.direct_client_for_each_server do |client|
+        client.use(:admin).database.command(clear_fail_point_command)
+      end
+    end
+
+    let(:collection) do
+      client['retryable-writes-error-spec', read: {mode: :secondary_preferred}]
+    end
+
+    it 'is reported on the server of the second attempt' do
+      first_primary = client.cluster.next_primary
+
+      client.cluster.servers_list.each do |server|
+        server.monitor.stop!
+      end
+
+      ClusterTools.instance.direct_client_for_each_server do |client|
+        client.use(:admin).database.command(fail_point_command)
+      end
+
+      # Retry should happen on a server other than the one used for
+      # initial attempt
+      expected_hosts = client.cluster.servers.reject do |server|
+        server.address == first_primary.address
+      end.map(&:address).map(&:host)
+
+      begin
+        collection.find(a: 1).to_a
+      rescue Mongo::Error::OperationFailure => e
+      else
+        fail('Expected operation to fail')
+      end
+
+      puts e.message
+
+      found = expected_hosts.any? do |host|
+        e.message.include?(host)
+      end
+      expect(found).to be true
+
+      expect(e.message).not_to include("on #{first_primary.address.seed}")
+
+      current_primary = client.cluster.next_primary
+    end
+  end
+end

spec/integration/retryable_writes_errors_spec.rb

@@ -1,7 +1,16 @@
 require 'spec_helper'
 
 describe 'Retryable writes errors tests' do
-  describe 'when the storage engine does not support retryable writes but the server does' do
+
+  let(:client) do
+    authorized_client.with(retry_writes: true)
+  end
+
+  let(:collection) do
+    client['retryable-writes-error-spec']
+  end
+
+  context 'when the storage engine does not support retryable writes but the server does' do
     require_mmapv1
     min_server_fcv '3.6'
     require_topology :replica_set, :sharded
@@ -10,14 +19,6 @@
       collection.delete_many
     end
 
-    let(:collection) do
-      client[authorized_collection.name]
-    end
-
-    let(:client) do
-      authorized_client.with(retry_writes: true)
-    end
-
     context 'when a retryable write is attempted' do
       it 'raises an actionable error message' do
         expect {
@@ -27,4 +28,4 @@
       end
     end
   end
-end
+end

spec/support/cluster_tools.rb

@@ -349,8 +349,6 @@ def close_clients
     end
   end
 
-  private
-
   def each_server(&block)
     admin_client.cluster.servers_list.each(&block)
   end
@@ -361,6 +359,8 @@ def direct_client_for_each_server(&block)
     end
   end
 
+  private
+
   def reset_server_states
     each_server do |server|
       server.unknown!