increase max containers in cosmos

Signed-off-by: Gerd Oberlechner <[email protected]>

Author: geoberle

test-integration/hack/start-cosmos-emulator-vnext.sh

@@ -39,6 +39,7 @@ ${CONTAINER_RUNTIME} run \
   --publish 8081:8081 \
   --publish 10250-10255:10250-10255 \
   -e AZURE_COSMOS_EMULATOR_IP_ADDRESS_OVERRIDE=127.0.0.1 \
+  -e AZURE_COSMOS_EMULATOR_PARTITION_COUNT=25 \
   -e PROTOCOL=https \
   --name "${CONTAINER_NAME}" \
   --detach \

test-integration/hack/start-cosmos-emulator.sh

@@ -33,24 +33,30 @@ if ${CONTAINER_RUNTIME} ps -q --filter "name=local-cosmos-emulator-*" | grep -q
     ${CONTAINER_RUNTIME} ps -aq --filter "name=local-cosmos-emulator-*" | xargs -r ${CONTAINER_RUNTIME} rm
 fi
 
+# number of partitions to use for the emulator
+# increase if a lot of tests run in parallel and start failing with 503 errors
+PARTITION_COUNT=50
+
 echo "Starting Cosmos DB emulator with container name: ${CONTAINER_NAME}"
 ${CONTAINER_RUNTIME} run \
   --publish 8081:8081 \
   --publish 10250-10255:10250-10255 \
   -e AZURE_COSMOS_EMULATOR_IP_ADDRESS_OVERRIDE=127.0.0.1 \
+  -e AZURE_COSMOS_EMULATOR_PARTITION_COUNT=${PARTITION_COUNT} \
   --name "${CONTAINER_NAME}" \
   --detach \
   mcr.microsoft.com/cosmosdb/linux/azure-cosmos-emulator:latest
 
 # Wait for emulator to be ready by checking logs
 echo "Waiting for Cosmos DB emulator to be ready..."
 for i in {1..60}; do
-    if ${CONTAINER_RUNTIME} logs "${CONTAINER_NAME}" 2>&1 | grep -q "Started 11/11 partitions"; then
+    if ${CONTAINER_RUNTIME} logs "${CONTAINER_NAME}" 2>&1 | grep -q "Started ${PARTITION_COUNT+1}/${PARTITION_COUNT+1} partitions"; then
         echo "Cosmos DB emulator is ready!"
         break
     fi
     if [ "$i" -eq 60 ]; then
         echo "Timeout waiting for Cosmos DB emulator to be ready"
+        ${CONTAINER_RUNTIME} logs "${CONTAINER_NAME}"
         exit 1
     fi
     echo "Attempt $i/60: Waiting for emulator to start all partitions..."