Add unit test for SERVER-4745.

1 files changed, 22 insertions, 0 deletions

diff --git a/src/mongo/dbtests/chunktests.cpp b/src/mongo/dbtests/chunktests.cpp
index 7943e38d078..822046a6a15 100644
--- a/src/mongo/dbtests/chunktests.cpp
+++ b/src/mongo/dbtests/chunktests.cpp
@@ -214,6 +214,27 @@ namespace ChunkTests {
             virtual BSONArray expectedShardNames() const { return BSON_ARRAY( "1" << "2" << "3" ); }
         };
 
+        class CompoundKeyBase : public Base {
+            virtual BSONObj shardKey() const {
+                return BSON( "a" << 1 << "b" << 1 );
+            }
+            virtual BSONArray splitPoints() const {
+                return BSON_ARRAY( BSON( "a" << 5 << "b" << 10 ) << BSON ( "a" << 5 << "b" << 20 ) );
+            }
+        };
+
+        class InMultiShard : public CompoundKeyBase {
+            virtual BSONObj query() const {
+                return BSON( "a" << BSON( "$in" << BSON_ARRAY( 0 << 5 << 10 ) ) <<
+                             "b" << BSON( "$in" << BSON_ARRAY( 0 << 5 << 25 ) ) );
+            }
+            // If we were to send this query to just the shards it actually needed to hit, it would only hit shards 0 and 2
+            // Because of the optimization from SERVER-4745, however, we'll also hit shard 1.
+            virtual BSONArray expectedShardNames() const {
+                return BSON_ARRAY( "0" << "1" << "2" );
+            }
+        };
+
     } // namespace ChunkManagerTests
     
     class All : public Suite {
@@ -241,6 +262,7 @@ namespace ChunkTests {
             add<ChunkManagerTests::EqualityThenUnsatisfiable>();
             add<ChunkManagerTests::InequalityThenUnsatisfiable>();
             add<ChunkManagerTests::OrEqualityUnsatisfiableInequality>();
+            add<ChunkManagerTests::InMultiShard>();
         }
     } myall;