SERVER-37871 Enforce agreement on shard key across cluster for $out

1 files changed, 27 insertions, 4 deletions

diff --git a/src/mongo/db/pipeline/document_source_out_test.cpp b/src/mongo/db/pipeline/document_source_out_test.cpp
index 44256d1fb4b..60530547165 100644
--- a/src/mongo/db/pipeline/document_source_out_test.cpp
+++ b/src/mongo/db/pipeline/document_source_out_test.cpp
@@ -67,6 +67,12 @@ public:
         OperationContext* opCtx, const NamespaceString& nss) const override {
         return {"_id"};
     }
+
+    void checkRoutingInfoEpochOrThrow(const boost::intrusive_ptr<ExpressionContext>& expCtx,
+                                      const NamespaceString&,
+                                      ChunkVersion) const override {
+        return;  // Pretend it always matches for our tests here.
+    }
 };
 
 class DocumentSourceOutTest : public AggregationContextFixture {
@@ -359,21 +365,38 @@ TEST_F(DocumentSourceOutTest, FailsToParseIfUniqueKeyHasDuplicateFields) {
     ASSERT_THROWS_CODE(createOutStage(spec), AssertionException, ErrorCodes::BadValue);
 }
 
-TEST_F(DocumentSourceOutTest, FailsToParseIfTargetEpochIsSpecifiedOnMongos) {
+TEST_F(DocumentSourceOutTest, FailsToParseIfTargetCollectionVersionIsSpecifiedOnMongos) {
     BSONObj spec = BSON("$out" << BSON("to"
                                        << "test"
                                        << "mode"
                                        << kDefaultMode
                                        << "uniqueKey"
                                        << BSON("_id" << 1)
-                                       << "epoch"
-                                       << OID::gen()));
+                                       << "targetCollectionVersion"
+                                       << ChunkVersion(0, 0, OID::gen()).toBSON()));
     getExpCtx()->inMongos = true;
     ASSERT_THROWS_CODE(createOutStage(spec), AssertionException, 50984);
 
-    // Test that 'targetEpoch' is accepted if not in mongos.
+    // Test that 'targetCollectionVersion' is accepted if _from_ mongos.
     getExpCtx()->inMongos = false;
+    getExpCtx()->fromMongos = true;
     ASSERT(createOutStage(spec) != nullptr);
+
+    // Test that 'targetCollectionVersion' is not accepted if on mongod but not from mongos.
+    getExpCtx()->inMongos = false;
+    getExpCtx()->fromMongos = false;
+    ASSERT_THROWS_CODE(createOutStage(spec), AssertionException, 51018);
+}
+
+TEST_F(DocumentSourceOutTest, FailsToParseifUniqueKeyIsNotSentFromMongos) {
+    BSONObj spec = BSON("$out" << BSON("to"
+                                       << "test"
+                                       << "mode"
+                                       << kDefaultMode
+                                       << "targetCollectionVersion"
+                                       << ChunkVersion(0, 0, OID::gen()).toBSON()));
+    getExpCtx()->fromMongos = true;
+    ASSERT_THROWS_CODE(createOutStage(spec), AssertionException, 51017);
 }
 
 TEST_F(DocumentSourceOutTest, CorrectlyUsesTargetDbThatMatchesAggregationDb) {