findMostSevereTestStepResultBy TestCaseFinished

Author: mpkorstanje

java/src/main/java/io/cucumber/query/Query.java

@@ -173,6 +173,14 @@ public Optional<TestStepResult> findMostSevereTestStepResultBy(TestCaseStarted t
                 .stream()
                 .map(TestStepFinished::getTestStepResult)
                 .max(testStepResultComparator);
+    }   
+    
+    public Optional<TestStepResult> findMostSevereTestStepResultBy(TestCaseFinished testCaseFinished) {
+        requireNonNull(testCaseFinished);
+        return findTestStepsFinishedBy(testCaseFinished)
+                .stream()
+                .map(TestStepFinished::getTestStepResult)
+                .max(testStepResultComparator);
     }
 
     public String findNameOf(GherkinDocument element, NamingStrategy namingStrategy) {
@@ -249,6 +257,12 @@ public Optional<Pickle> findPickleBy(TestCaseStarted testCaseStarted) {
         return findTestCaseBy(testCaseStarted)
                 .flatMap(this::findPickleBy);
     }
+    
+    public Optional<Pickle> findPickleBy(TestCaseFinished testCaseFinished) {
+        requireNonNull(testCaseFinished);
+        return findTestCaseBy(testCaseFinished)
+                .flatMap(this::findPickleBy);
+    }
 
     public Optional<Pickle> findPickleBy(TestCase testCase) {
         requireNonNull(testCase);
@@ -403,6 +417,14 @@ public List<TestStepFinished> findTestStepsFinishedBy(TestCaseStarted testCaseSt
         return new ArrayList<>(testStepsFinished);
     }
 
+    public List<TestStepFinished> findTestStepsFinishedBy(TestCaseFinished testCaseFinished) {
+        requireNonNull(testCaseFinished);
+        List<TestStepFinished> testStepsFinished = testStepsFinishedByTestCaseStartedId.
+                getOrDefault(testCaseFinished.getTestCaseStartedId(), emptyList());
+        // Concurrency
+        return new ArrayList<>(testStepsFinished);
+    }
+
     public List<Entry<TestStepFinished, TestStep>> findTestStepFinishedAndTestStepBy(TestCaseStarted testCaseStarted) {
         return findTestStepsFinishedBy(testCaseStarted).stream()
                 .map(testStepFinished -> findTestStepBy(testStepFinished).map(testStep -> new SimpleEntry<>(testStepFinished, testStep)))