Skip to content

Commit 6c7646c

Browse files
committed
MockServiceProvider throw an exception if query does not match.
1 parent 5d3ab17 commit 6c7646c

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

src/test/java/com/intuit/graphql/orchestrator/testhelpers/MockServiceProvider.java

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,7 @@
1212
import java.util.Map;
1313
import java.util.Set;
1414
import java.util.concurrent.CompletableFuture;
15+
import org.apache.commons.collections4.MapUtils;
1516

1617
public class MockServiceProvider implements ServiceProvider {
1718

@@ -52,7 +53,11 @@ public ServiceType getSeviceType() {
5253
@Override
5354
public CompletableFuture<Map<String, Object>> query(ExecutionInput executionInput,
5455
GraphQLContext context) {
55-
return CompletableFuture.completedFuture(this.responseMap.get(executionInput.getQuery()));
56+
Map<String, Object> data = this.responseMap.get(executionInput.getQuery());
57+
if (MapUtils.isEmpty(data)) {
58+
throw new IllegalArgumentException("Mock response not found for query: " + executionInput.getQuery());
59+
}
60+
return CompletableFuture.completedFuture(data);
5661
}
5762

5863
public static Builder builder() {

0 commit comments

Comments
 (0)