From 07bfb1286cb6d01d8080ab3b39a1c76dd123e002 Mon Sep 17 00:00:00 2001
From: Geert Bosch <geert@mongodb.com>
Date: Thu, 16 Nov 2017 14:37:20 -0500
Subject: [PATCH] SERVER-32001 Fix unindexing of rollback in unique partial
 indexes

---
 .../noPassthrough/partial_unique_indexes.js   | 47 +++++++++++++++++++
 .../storage/wiredtiger/wiredtiger_index.cpp   |  1 +
 2 files changed, 48 insertions(+)
 create mode 100644 jstests/noPassthrough/partial_unique_indexes.js

diff --git a/jstests/noPassthrough/partial_unique_indexes.js b/jstests/noPassthrough/partial_unique_indexes.js
new file mode 100644
index 00000000000..7c376872ade
--- /dev/null
+++ b/jstests/noPassthrough/partial_unique_indexes.js
@@ -0,0 +1,47 @@
+/**
+ * SERVER-32001: Test that indexing paths for non-unique, partial, unique, partial&unique
+ * crud operations correctly handle WriteConflictExceptions.
+ */
+(function() {
+    "strict";
+
+    let conn = MongoRunner.runMongod();
+    let testDB = conn.getDB("test");
+
+    let t = testDB.jstests_parallel_allops;
+    t.drop();
+
+    t.createIndex({x: 1, _id: 1}, {partialFilterExpression: {_id: {$lt: 500}}, unique: true});
+    t.createIndex({y: -1, _id: 1}, {unique: true});
+    t.createIndex({x: -1}, {partialFilterExpression: {_id: {$gte: 500}}, unique: false});
+    t.createIndex({y: 1}, {unique: false});
+
+    let _id = {"#RAND_INT": [0, 1000]};
+    let ops = [
+        {op: "remove", ns: t.getFullName(), query: {_id}},
+        {op: "update", ns: t.getFullName(), query: {_id}, update: {$inc: {x: 1}}, upsert: true},
+        {op: "update", ns: t.getFullName(), query: {_id}, update: {$inc: {y: 1}}, upsert: true},
+    ];
+
+    let seconds = 5;
+    let parallel = 5;
+    let host = testDB.getMongo().host;
+
+    let benchArgs = {ops, seconds, parallel, host};
+
+    assert.commandWorked(testDB.adminCommand({
+        configureFailPoint: 'WTWriteConflictExceptionForReads',
+        mode: {activationProbability: 0.01}
+    }));
+    assert.commandWorked(testDB.adminCommand(
+        {configureFailPoint: 'WTWriteConflictException', mode: {activationProbability: 0.01}}));
+    res = benchRun(benchArgs);
+    printjson({res});
+
+    assert.commandWorked(
+        testDB.adminCommand({configureFailPoint: 'WTWriteConflictException', mode: "off"}));
+    assert.commandWorked(
+        testDB.adminCommand({configureFailPoint: 'WTWriteConflictExceptionForReads', mode: "off"}));
+    res = t.validate();
+    assert(res.valid, tojson(res));
+})();
diff --git a/src/mongo/db/storage/wiredtiger/wiredtiger_index.cpp b/src/mongo/db/storage/wiredtiger/wiredtiger_index.cpp
index 68e0728dc75..a65ce5eff8a 100644
--- a/src/mongo/db/storage/wiredtiger/wiredtiger_index.cpp
+++ b/src/mongo/db/storage/wiredtiger/wiredtiger_index.cpp
@@ -1213,6 +1213,7 @@ void WiredTigerIndexUnique::_unindex(WT_CURSOR* c,
                 triggerWriteConflictAtPoint(c);
                 return;
             }
+            invariantWTOK(ret);
             WT_ITEM value;
             invariantWTOK(c->get_value(c, &value));
             BufReader br(value.data, value.size);