internal static INameValueCollection ExtractResponseHeaders(HttpResponseMessage responseMessage)
{
INameValueCollection headers = new StoreResponseNameValueCollection();
foreach (KeyValuePair <string, IEnumerable <string> > headerPair in responseMessage.Headers)
{
if (string.Compare(headerPair.Key, HttpConstants.HttpHeaders.OwnerFullName, StringComparison.Ordinal) == 0)
{
foreach (string val in headerPair.Value)
{
headers.Add(headerPair.Key, Uri.UnescapeDataString(val));
}
}
else
{
foreach (string val in headerPair.Value)
{
headers.Add(headerPair.Key, val);
}
}
}
if (responseMessage.Content != null)
{
foreach (KeyValuePair <string, IEnumerable <string> > headerPair in responseMessage.Content.Headers)
{
if (string.Compare(headerPair.Key, HttpConstants.HttpHeaders.OwnerFullName, StringComparison.Ordinal) == 0)
{
foreach (string val in headerPair.Value)
{
headers.Add(headerPair.Key, Uri.UnescapeDataString(val));
}
}
else
{
foreach (string val in headerPair.Value)
{
headers.Add(headerPair.Key, val);
}
}
}
}
return(headers);
}
/// <summary>
/// For mocking a TransportClient response/
/// </summary>
/// <param name="request">The <see cref="DocumentServiceRequest"/> instance.</param>
/// <returns>A <see cref="StoreResponse"/> instance.</returns>
public static StoreResponse GetStoreResponse(DocumentServiceRequest request)
{
StoreResponseNameValueCollection headers = MockRequestHelper.GenerateTestHeaders();
if (request.OperationType == OperationType.Read)
{
headers.Add(WFConstants.BackendHeaders.LSN, "1");
if (request.ResourceAddress.EndsWith(MockedItemBenchmarkHelper.ExistingItemId))
{
return(new StoreResponse()
{
ResponseBody = new MemoryStream(MockRequestHelper.testItemResponsePayload, 0, MockRequestHelper.testItemResponsePayload.Length, writable: false, publiclyVisible: true),
Status = (int)System.Net.HttpStatusCode.OK,
Headers = headers,
});
}
return(new StoreResponse()
{
ResponseBody = Stream.Null,
Status = (int)System.Net.HttpStatusCode.NotFound,
Headers = headers,
});
}
if (request.OperationType == OperationType.Delete)
{
if (request.ResourceAddress.EndsWith(MockedItemBenchmarkHelper.ExistingItemId))
{
return(new StoreResponse()
{
ResponseBody = new MemoryStream(MockRequestHelper.testItemResponsePayload, 0, MockRequestHelper.testItemResponsePayload.Length, writable: false, publiclyVisible: true),
Status = (int)System.Net.HttpStatusCode.OK,
Headers = headers,
});
}
return(new StoreResponse()
{
ResponseBody = Stream.Null,
Status = (int)System.Net.HttpStatusCode.NotFound,
Headers = headers,
});
}
if (request.OperationType == OperationType.Create ||
request.OperationType == OperationType.Replace ||
request.OperationType == OperationType.Upsert ||
request.OperationType == OperationType.Patch)
{
return(new StoreResponse()
{
ResponseBody = new MemoryStream(MockRequestHelper.testItemResponsePayload, 0, MockRequestHelper.testItemResponsePayload.Length, writable: false, publiclyVisible: true),
Status = (int)System.Net.HttpStatusCode.OK,
Headers = headers,
});
}
if (request.ResourceType == ResourceType.Document &&
request.OperationType == OperationType.ReadFeed)
{
return(new StoreResponse()
{
ResponseBody = new MemoryStream(MockRequestHelper.testItemFeedResponsePayload, 0, MockRequestHelper.testItemFeedResponsePayload.Length, writable: false, publiclyVisible: true),
Status = (int)System.Net.HttpStatusCode.OK,
Headers = headers,
});
}
if (request.OperationType == OperationType.Batch)
{
MemoryStream responseContent = batchResponsePayloadWriter.GeneratePayload();
return(new StoreResponse()
{
ResponseBody = responseContent,
Status = (int)System.Net.HttpStatusCode.OK,
Headers = headers,
});
}
return(null);
}
/// <summary>
/// For mocking a TransportClient response/
/// </summary>
/// <param name="request">The <see cref="DocumentServiceRequest"/> instance.</param>
/// <returns>A <see cref="StoreResponse"/> instance.</returns>
public static StoreResponse GetStoreResponse(DocumentServiceRequest request)
{
if (request.ResourceType == ResourceType.Document &&
request.OperationType == OperationType.Query)
{
StoreResponseNameValueCollection queryHeaders = new StoreResponseNameValueCollection()
{
ActivityId = Guid.NewGuid().ToString(),
BackendRequestDurationMilliseconds = "1.42",
CurrentReplicaSetSize = "1",
CurrentWriteQuorum = "1",
CurrentResourceQuotaUsage = "documentSize=0;documentsSize=1;documentsCount=1;collectionSize=1;",
GlobalCommittedLSN = "-1",
LSN = "2540",
LocalLSN = "2540",
LastStateChangeUtc = "Wed, 18 Aug 2021 20:30:05.117 GMT",
MaxResourceQuota = "documentSize=10240;documentsSize=10485760;documentsCount=-1;collectionSize=10485760;",
NumberOfReadRegions = "0",
OwnerFullName = "dbs/f4ac3cfd-dd38-4adb-b2d2-be97b3efbd1b/colls/2a926112-a26e-4935-ac6a-66df269c890d",
OwnerId = "GHRtAJahWQ4=",
PartitionKeyRangeId = "0",
PendingPKDelete = "false",
QueryExecutionInfo = "{\"reverseRidEnabled\":false,\"reverseIndexScan\":false}",
QueryMetrics = "totalExecutionTimeInMs=0.78;queryCompileTimeInMs=0.26;queryLogicalPlanBuildTimeInMs=0.04;queryPhysicalPlanBuildTimeInMs=0.18;queryOptimizationTimeInMs=0.01;VMExecutionTimeInMs=0.04;indexLookupTimeInMs=0.00;documentLoadTimeInMs=0.02;systemFunctionExecuteTimeInMs=0.00;userFunctionExecuteTimeInMs=0.00;retrievedDocumentCount=1;retrievedDocumentSize=671;outputDocumentCount=1;outputDocumentSize=720;writeOutputTimeInMs=0.00;indexUtilizationRatio=1.00",
QuorumAckedLSN = "2540",
QuorumAckedLocalLSN = "2540",
RequestCharge = "2.27",
SchemaVersion = "1.12",
ServerVersion = " version=2.14.0.0",
SessionToken = "0:-1#2540",
TransportRequestID = "2",
XPRole = "0",
};
return(new StoreResponse()
{
ResponseBody = new MemoryStream(MockRequestHelper.testItemFeedResponsePayload, 0, MockRequestHelper.testItemFeedResponsePayload.Length, writable: false, publiclyVisible: true),
Status = (int)System.Net.HttpStatusCode.OK,
Headers = queryHeaders,
});
}
StoreResponseNameValueCollection headers = MockRequestHelper.GenerateTestHeaders();
if (request.OperationType == OperationType.Read)
{
headers.Add(WFConstants.BackendHeaders.LSN, "1");
if (request.ResourceAddress.EndsWith(MockedItemBenchmarkHelper.ExistingItemId))
{
return(new StoreResponse()
{
ResponseBody = new MemoryStream(MockRequestHelper.testItemResponsePayload, 0, MockRequestHelper.testItemResponsePayload.Length, writable: false, publiclyVisible: true),
Status = (int)System.Net.HttpStatusCode.OK,
Headers = headers,
});
}
return(new StoreResponse()
{
ResponseBody = Stream.Null,
Status = (int)System.Net.HttpStatusCode.NotFound,
Headers = headers,
});
}
if (request.OperationType == OperationType.Delete)
{
if (request.ResourceAddress.EndsWith(MockedItemBenchmarkHelper.ExistingItemId))
{
return(new StoreResponse()
{
ResponseBody = new MemoryStream(MockRequestHelper.testItemResponsePayload, 0, MockRequestHelper.testItemResponsePayload.Length, writable: false, publiclyVisible: true),
Status = (int)System.Net.HttpStatusCode.OK,
Headers = headers,
});
}
return(new StoreResponse()
{
ResponseBody = Stream.Null,
Status = (int)System.Net.HttpStatusCode.NotFound,
Headers = headers,
});
}
if (request.OperationType == OperationType.Create ||
request.OperationType == OperationType.Replace ||
request.OperationType == OperationType.Upsert ||
request.OperationType == OperationType.Patch)
{
return(new StoreResponse()
{
ResponseBody = new MemoryStream(MockRequestHelper.testItemResponsePayload, 0, MockRequestHelper.testItemResponsePayload.Length, writable: false, publiclyVisible: true),
Status = (int)System.Net.HttpStatusCode.OK,
Headers = headers,
});
}
if (request.ResourceType == ResourceType.Document &&
request.OperationType == OperationType.ReadFeed)
{
return(new StoreResponse()
{
ResponseBody = new MemoryStream(MockRequestHelper.testItemFeedResponsePayload, 0, MockRequestHelper.testItemFeedResponsePayload.Length, writable: false, publiclyVisible: true),
Status = (int)System.Net.HttpStatusCode.OK,
Headers = headers,
});
}
if (request.OperationType == OperationType.Batch)
{
MemoryStream responseContent = batchResponsePayloadWriter.GeneratePayload();
return(new StoreResponse()
{
ResponseBody = responseContent,
Status = (int)System.Net.HttpStatusCode.OK,
Headers = headers,
});
}
return(null);
}
