Adding more tests

Azure · Jan 4, 2022 · 85049c4 · 85049c4
1 parent ec8f4ce
commit 85049c4
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/Microsoft.Azure.Cosmos/tests/Microsoft.Azure.Cosmos.Tests/GatewayStoreModelTest.cs b/Microsoft.Azure.Cosmos/tests/Microsoft.Azure.Cosmos.Tests/GatewayStoreModelTest.cs
@@ -414,14 +414,18 @@ await GatewayStoreModel.ApplySessionTokenAsync(
  });
  }
 
- [TestMethod]
- public async Task TestRequestOverloadRemovesSessionToken()
+ [DataTestMethod]
+ [DataRow(false, false, DisplayName = "Single master - Read")]
+ [DataRow(true, false, DisplayName = "Multi master - Read")]
+ [DataRow(false, true, DisplayName = "Single master - Write")]
+ [DataRow(true, true, DisplayName = "Multi master - Write")]
+ public async Task TestRequestOverloadRemovesSessionToken(bool multiMaster, bool isWriteRequest)
  {
  INameValueCollection headers = new StoreRequestNameValueCollection();
  headers.Set(HttpConstants.HttpHeaders.ConsistencyLevel, ConsistencyLevel.Eventual.ToString());
 
  DocumentServiceRequest dsrNoSessionToken = DocumentServiceRequest.CreateFromName(
- OperationType.Read,
+ isWriteRequest ? OperationType.Create : OperationType.Read,
  "Test",
  ResourceType.Document,
  AuthorizationTokenType.PrimaryMasterKey,
@@ -432,6 +436,7 @@ public async Task TestRequestOverloadRemovesSessionToken()
  sMock.Setup(x => x.ResolveGlobalSessionToken(dsrNoSessionToken)).Returns(dsrSessionToken);
 
  Mock<IGlobalEndpointManager> globalEndpointManager = new Mock<IGlobalEndpointManager>();
+ globalEndpointManager.Setup(gem => gem.CanUseMultipleWriteLocations(It.Is<DocumentServiceRequest>(drs => drs == dsrNoSessionToken))).Returns(multiMaster);
  await this.GetGatewayStoreModelForConsistencyTest(async (gatewayStoreModel) =>
  {
  await GatewayStoreModel.ApplySessionTokenAsync(