1 files changed, 22 insertions, 0 deletions

diff --git a/subprojects/store-query-viatra/src/test/java/tools/refinery/store/query/viatra/tests/QueryBackendHint.java b/subprojects/store-query-viatra/src/test/java/tools/refinery/store/query/viatra/tests/QueryBackendHint.java
new file mode 100644
index 00000000..b1818a17
--- /dev/null
+++ b/subprojects/store-query-viatra/src/test/java/tools/refinery/store/query/viatra/tests/QueryBackendHint.java
@@ -0,0 +1,22 @@
+package tools.refinery.store.query.viatra.tests;
+
+import org.eclipse.viatra.query.runtime.matchers.backend.QueryEvaluationHint;
+
+/**
+ * Overrides {@link QueryEvaluationHint#toString()} for pretty names in parametric test names.
+ */
+class QueryBackendHint extends QueryEvaluationHint {
+        public QueryBackendHint(BackendRequirement backendRequirementType) {
+                super(null, backendRequirementType);
+        }
+
+        @Override
+        public String toString() {
+                return switch (getQueryBackendRequirementType()) {
+                        case UNSPECIFIED -> "default";
+                        case DEFAULT_CACHING -> "incremental";
+                        case DEFAULT_SEARCH -> "localSearch";
+                        default -> throw new IllegalStateException("Unknown BackendRequirement");
+                };
+        }
+}