SERVER-44409 Add FSM and nice-to-have collection and index test cases

1 files changed, 17 insertions, 2 deletions

diff --git a/jstests/libs/create_collection_txn_helpers.js b/jstests/libs/create_collection_txn_helpers.js
index 9b6ba15eac8..08d7093a5c7 100644
--- a/jstests/libs/create_collection_txn_helpers.js
+++ b/jstests/libs/create_collection_txn_helpers.js
@@ -1,12 +1,27 @@
 /**
  * Helper function shared by createCollection inside txns tests.
  */
-const createCollAndCRUDInTxn = function(sessionDB, collName, explicitCreate) {
+const createCollAndCRUDInTxn = function(sessionDB, collName, explicitCreate, upsert) {
+    if (undefined === explicitCreate) {
+        doassert('createCollAndCRUDInTxn called with undefined explicitCreate');
+    }
+    if (undefined === upsert) {
+        doassert('createCollAndCRUDInTxn called with undefined upsert');
+    }
     if (explicitCreate) {
         assert.commandWorked(sessionDB.runCommand({create: collName}));
     }
     let sessionColl = sessionDB[collName];
-    assert.commandWorked(sessionColl.insert({a: 1}));
+    if (upsert) {
+        assert.commandWorked(sessionColl.update({_id: 1}, {$inc: {a: 1}}, {upsert: true}));
+    } else {
+        assert.commandWorked(sessionColl.insert({a: 1}));
+    }
     assert.eq(sessionColl.find({a: 1}).itcount(), 1);
+    assert.commandWorked(sessionColl.insert({_id: 2}));
+    let resDoc = sessionColl.findAndModify({query: {_id: 2}, update: {$inc: {a: 1}}});
+    assert.eq(resDoc._id, 2);
+    assert.commandWorked(sessionColl.update({_id: 2}, {$inc: {a: 1}}));
+    assert.commandWorked(sessionColl.deleteOne({_id: 2}));
     assert.eq(sessionColl.find({}).itcount(), 1);
 };