Function ensureMongoPrimary

packages/core/fumadb/test/shared.ts:404–433

(mongodb: MongoClient)

}

async function ensureMongoPrimary(mongodb: MongoClient) {
  const admin = mongodb.db("admin");

  try {
    await admin.command({ replSetGetStatus: 1 });
  } catch (error) {
    if (isMongoError(error, "NotYetInitialized")) {
      await admin.command({ replSetInitiate: mongoReplicaSetConfig });
    } else if (
      isMongoError(error, "InvalidReplicaSetConfig") ||
      (error instanceof Error && error.message.includes("not a member"))
    ) {
      await admin.command({
        replSetReconfig: mongoReplicaSetConfig,
        force: true,
      });
    } else {
      throw error;
    }
  }

  for (let i = 0; i < 30; i++) {
    const hello = await admin.command({ hello: 1 });
    if (hello.isWritablePrimary) return;

    await wait(500);
  }

  throw new Error("Timed out waiting for MongoDB replica set primary.");
}

export async function resetMongoDB(mongodb: MongoClient) {
  await ensureMongoPrimary(mongodb);