SERVER-29107 move shardCollection logic into new _configsvrShardCollection command on config server

1 files changed, 18 insertions, 0 deletions

diff --git a/src/mongo/s/config_server_test_fixture.h b/src/mongo/s/config_server_test_fixture.h
index 9eeb0fa7398..790902664cb 100644
--- a/src/mongo/s/config_server_test_fixture.h
+++ b/src/mongo/s/config_server_test_fixture.h
@@ -34,11 +34,13 @@ namespace mongo {
 
 class BSONObj;
 class ChunkType;
+struct ChunkVersion;
 class KeysCollectionDocument;
 class NamespaceString;
 class Shard;
 class ShardingCatalogClient;
 class ShardingCatalogManager;
+class ShardId;
 class ShardRegistry;
 class ShardType;
 template <typename T>
@@ -91,11 +93,27 @@ public:
     StatusWith<ChunkType> getChunkDoc(OperationContext* opCtx, const BSONObj& minKey);
 
     /**
+     * Inserts a document for the database into the config.databases collection.
+     */
+    void setupDatabase(const std::string& dbName, const ShardId primaryShard, const bool sharded);
+
+    /**
      * Returns the indexes definitions defined on a given collection.
      */
     StatusWith<std::vector<BSONObj>> getIndexes(OperationContext* opCtx, const NamespaceString& ns);
 
     /**
+     * Expects a setShardVersion command to be executed on the specified shard.
+     *
+     * The expectedChunkVersion is optional, because in some cases it may not be possible to know
+     * the OID of a ChunkVersion generated by some internal code. (See SERVER-29451).
+     */
+    void expectSetShardVersion(const HostAndPort& expectedHost,
+                               const ShardType& expectedShard,
+                               const NamespaceString& expectedNs,
+                               boost::optional<ChunkVersion> expectedChunkVersion);
+
+    /**
      * Returns the stored raw pointer to the addShard TaskExecutor's NetworkInterface.
      */
     executor::NetworkInterfaceMock* networkForAddShard() const;