Implement Retryable Reads

GODRIVER-624

Change-Id: I7a82d23ac449b8f8595ca3df9cfb7b671ec25efb

Author: eppingere

Makefile

@@ -55,19 +55,27 @@ errcheck:
 
 .PHONY: test
 test:
-	go test $(BUILD_TAGS) -timeout $(TEST_TIMEOUT)s $(TEST_PKGS)
+	for TEST in $(TEST_PKGS) ; do \
+		go test $(BUILD_TAGS) -timeout $(TEST_TIMEOUT)s $$TEST ; \
+	done
 
 .PHONY: test-cover
 test-cover:
-	go test $(BUILD_TAGS) -timeout $(TEST_TIMEOUT)s -cover $(COVER_ARGS) $(TEST_PKGS)
+	for TEST in $(TEST_PKGS) ; do \
+    	go test $(BUILD_TAGS) -timeout $(TEST_TIMEOUT)s -cover $(COVER_ARGS) $$TEST ; \
+    done
 
 .PHONY: test-race
 test-race:
-	go test $(BUILD_TAGS) -timeout $(TEST_TIMEOUT)s -race $(TEST_PKGS)
+	for TEST in $(TEST_PKGS) ; do \
+    	go test $(BUILD_TAGS) -timeout $(TEST_TIMEOUT)s -race $(COVER_ARGS) $$TEST ; \
+    done
 
 .PHONY: test-short
 test-short:
-	go test $(BUILD_TAGS) -timeout $(TEST_TIMEOUT)s -short $(TEST_PKGS)
+	for TEST in $(TEST_PKGS) ; do \
+    	go test $(BUILD_TAGS) -timeout $(TEST_TIMEOUT)s -short $(COVER_ARGS) $$TEST ; \
+    done
 
 .PHONY: update-bson-corpus-tests
 update-bson-corpus-tests:
@@ -109,7 +117,9 @@ vet:
 # Evergreen specific targets
 .PHONY: evg-test
 evg-test:
-	go test $(BUILD_TAGS) -v -timeout $(TEST_TIMEOUT)s $(TEST_PKGS) > test.suite
+	for TEST in $(TEST_PKGS) ; do \
+		go test $(BUILD_TAGS) -v -timeout $(TEST_TIMEOUT)s $$TEST >> test.suite ; \
+	done
 
 .PHONY: evg-test-auth
 evg-test-auth:

data/retryable-reads/aggregate-merge.json

@@ -0,0 +1,98 @@
+{
+  "runOn": [
+    {
+      "minServerVersion": "4.2.0"
+    }
+  ],
+  "database_name": "retryable-reads-tests",
+  "collection_name": "coll",
+  "data": [
+    {
+      "_id": 1,
+      "x": 11
+    },
+    {
+      "_id": 2,
+      "x": 22
+    },
+    {
+      "_id": 3,
+      "x": 33
+    }
+  ],
+  "tests": [
+    {
+      "description": "Aggregate with $merge does not retry",
+      "failPoint": {
+        "configureFailPoint": "failCommand",
+        "mode": {
+          "times": 1
+        },
+        "data": {
+          "failCommands": [
+            "aggregate"
+          ],
+          "closeConnection": true
+        }
+      },
+      "operations": [
+        {
+          "object": "collection",
+          "name": "aggregate",
+          "arguments": {
+            "pipeline": [
+              {
+                "$match": {
+                  "_id": {
+                    "$gt": 1
+                  }
+                }
+              },
+              {
+                "$sort": {
+                  "x": 1
+                }
+              },
+              {
+                "$merge": {
+                  "into": "output-collection"
+                }
+              }
+            ]
+          },
+          "error": true
+        }
+      ],
+      "expectations": [
+        {
+          "command_started_event": {
+            "command": {
+              "aggregate": "coll",
+              "pipeline": [
+                {
+                  "$match": {
+                    "_id": {
+                      "$gt": 1
+                    }
+                  }
+                },
+                {
+                  "$sort": {
+                    "x": 1
+                  }
+                },
+                {
+                  "$merge": {
+                    "into": "output-collection"
+                  }
+                }
+              ]
+            },
+            "command_name": "aggregate",
+            "database_name": "retryable-reads-tests"
+          }
+        }
+      ]
+    }
+  ]
+}

data/retryable-reads/aggregate-merge.yml

@@ -0,0 +1,39 @@
+runOn:
+  -
+    minServerVersion: "4.2.0"
+
+database_name: &database_name "retryable-reads-tests"
+collection_name: &collection_name "coll"
+
+data:
+  - {_id: 1, x: 11}
+  - {_id: 2, x: 22}
+  - {_id: 3, x: 33}
+
+tests:
+  -
+    description: "Aggregate with $merge does not retry"
+    failPoint:
+      configureFailPoint: failCommand
+      mode: { times: 1 }
+      data:
+        failCommands: [aggregate]
+        closeConnection: true
+    operations:
+      -
+        object: collection
+        name: aggregate
+        arguments:
+          pipeline: &pipeline
+            - $match: {_id: {$gt: 1}}
+            - $sort: {x: 1}
+            - $merge: { into: "output-collection" }
+        error: true
+    expectations:
+      -
+        command_started_event:
+          command:
+            aggregate: *collection_name
+            pipeline: *pipeline
+          command_name: aggregate
+          database_name: *database_name