public Task DispatchAsync(
IList <BufferedRequest> requests,
CancellationToken cancellationToken)
{
if (requests == null)
{
throw new ArgumentNullException(nameof(requests));
}
if (requests.Count == 0)
{
return(Task.CompletedTask);
}
if (requests.Count == 1)
{
return(DispatchSingleRequestsAsync(requests, cancellationToken));
}
var rewriter = new MergeQueryRewriter();
var variableValues = new Dictionary <string, object>();
for (int i = 0; i < requests.Count; i++)
{
MergeRequest(requests[i], rewriter, variableValues, $"__{i}_");
}
return(DispatchRequestsAsync(
requests,
rewriter.Merge(),
variableValues,
cancellationToken));
}
public void DocumentHasNoOperation()
{
// arrange
DocumentNode query = Parser.Default.Parse(
"type Foo { s: String }");
// act
var rewriter = new MergeQueryRewriter(Array.Empty <string>());
Action action = () => rewriter.AddQuery(query, "_a", false);
// assert
Assert.Equal("document",
Assert.Throws <ArgumentException>(action).ParamName);
}
public void DocumentIsNull()
{
// arrange
DocumentNode query = Utf8GraphQLParser.Parse(
"type Foo { s: String }");
// act
var rewriter = new MergeQueryRewriter(Array.Empty <string>());
Action action = () => rewriter.AddQuery(null, "_a", false);
// assert
Assert.Equal("document",
Assert.Throws <ArgumentNullException>(action).ParamName);
}
public void RequestPrefixIsEmpty()
{
// arrange
DocumentNode query = Parser.Default.Parse(
"type Foo { s: String }");
// act
var rewriter = new MergeQueryRewriter(Array.Empty <string>());
Action action = () => rewriter.AddQuery(
query, default(NameString), false);
// assert
Assert.Equal("requestPrefix",
Assert.Throws <ArgumentException>(action).ParamName);
}
public void QueriesAreNotOfTheSameOperationType()
{
// arrange
DocumentNode query_a = Parser.Default.Parse("query a { b }");
DocumentNode query_b = Parser.Default.Parse("mutation a { b }");
// act
var rewriter = new MergeQueryRewriter(Array.Empty <string>());
rewriter.AddQuery(query_a, "abc", false);
Action action = () => rewriter.AddQuery(query_b, "abc", false);
// assert
Assert.Equal("document",
Assert.Throws <ArgumentException>(action).ParamName);
}
public Task DispatchAsync(
IList <BufferedRequest> requests,
CancellationToken cancellationToken)
{
if (requests == null)
{
throw new ArgumentNullException(nameof(requests));
}
if (requests.Count == 0)
{
return(Task.CompletedTask);
}
if (requests.Count == 1)
{
return(DispatchSingleRequestsAsync(requests, cancellationToken));
}
var rewriter = new MergeQueryRewriter();
var variableValues = new Dictionary <string, object>();
var operationNames = requests
.Select(r => r.Request.OperationName)
.Where(n => n != null)
.Distinct()
.ToList();
if (operationNames.Count == 1)
{
rewriter.SetOperationName(new NameNode(operationNames[0]));
}
for (int i = 0; i < requests.Count; i++)
{
MergeRequest(requests[i], rewriter, variableValues, $"__{i}_");
}
return(DispatchRequestsAsync(
requests,
rewriter.Merge(),
rewriter.OperationName,
variableValues,
cancellationToken));
}
public void QueryWithPrivateVariables()
{
// arrange
DocumentNode query_a = Parser.Default.Parse(
FileResource.Open("StitchingQueryWithUnion.graphql"));
DocumentNode query_b = Parser.Default.Parse(
FileResource.Open("StitchingQueryWithVariables.graphql"));
// act
var rewriter = new MergeQueryRewriter(Array.Empty <string>());
rewriter.AddQuery(query_a, "_a", false);
rewriter.AddQuery(query_b, "_b", false);
DocumentNode document = rewriter.Merge();
// assert
QuerySyntaxSerializer.Serialize(document).MatchSnapshot();
}
public void SimpleShortHandQuery()
{
// arrange
string query_a = "{ a { b } }";
string query_b = "{ c { d } }";
string query_c = "{ a { c } }";
// act
var rewriter = new MergeQueryRewriter(Array.Empty <string>());
rewriter.AddQuery(Parser.Default.Parse(query_a), "_a", false);
rewriter.AddQuery(Parser.Default.Parse(query_b), "_b", false);
rewriter.AddQuery(Parser.Default.Parse(query_c), "_c", false);
DocumentNode document = rewriter.Merge();
// assert
QuerySyntaxSerializer.Serialize(document).MatchSnapshot();
}
public void AliasesMapIsCorrect()
{
// arrange
DocumentNode query_a = Parser.Default.Parse(
FileResource.Open("MergeQueryWithVariable.graphql"));
DocumentNode query_b = Parser.Default.Parse(
FileResource.Open("MergeQueryWithVariable.graphql"));
// act
var rewriter = new MergeQueryRewriter(Array.Empty <string>());
IDictionary <string, string> a =
rewriter.AddQuery(query_a, "_a", true);
IDictionary <string, string> b =
rewriter.AddQuery(query_b, "_b", true);
// assert
a.MatchSnapshot("AliasesMapIsCorrect_A");
b.MatchSnapshot("AliasesMapIsCorrect_B");
}
public void QueryWithGlobalVariables()
{
// arrange
DocumentNode query_a = Parser.Default.Parse(
FileResource.Open("MergeQueryWithVariable.graphql"));
DocumentNode query_b = Parser.Default.Parse(
FileResource.Open("MergeQueryWithVariable.graphql"));
// act
var rewriter = new MergeQueryRewriter(
new HashSet <string>(new[] { "global" }));
rewriter.AddQuery(query_a, "_a", true);
rewriter.AddQuery(query_b, "_b", true);
DocumentNode document = rewriter.Merge();
// assert
QuerySyntaxSerializer.Serialize(document).MatchSnapshot();
}
private static void MergeRequest(
BufferedRequest bufferedRequest,
MergeQueryRewriter rewriter,
IDictionary <string, object> variableValues,
NameString requestPrefix)
{
MergeVariables(
bufferedRequest.Request.VariableValues,
variableValues,
requestPrefix);
bool isAutoGenerated = bufferedRequest.Request.Properties != null &&
bufferedRequest.Request.Properties.ContainsKey(
WellKnownProperties.IsAutoGenerated);
bufferedRequest.Aliases = rewriter.AddQuery(
bufferedRequest.Document,
requestPrefix,
isAutoGenerated);
}