Add test

Author: mpkorstanje

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

@@ -301,7 +301,12 @@ public <T> Optional<T> reduceLinageOf(Pickle pickle, LineageReducer<T> reducer)
     }
 
     public Optional<Location> findLocationOf(Pickle pickle) {
-       return reduceLinageOf(pickle, ascending(FirstLocationCollector::new));
+       return findLineageBy(pickle).flatMap(lineage -> {
+           if (lineage.example().isPresent()) {
+               return lineage.example().map(TableRow::getLocation);
+           }
+           return lineage.scenario().map(Scenario::getLocation);
+       });
     }
 
     public Optional<Pickle> findPickleBy(TestCaseStarted testCaseStarted) {

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

@@ -137,6 +137,10 @@ private static Map<String, Object> createQueryResults(Query query) {
                 .map(hook -> hook.map(Hook::getId))
                 .filter(Optional::isPresent)
                 .collect(toList()));
+        results.put("findLocationBy", query.findAllPickles().stream()
+                .map(query::findLocationOf)
+                .filter(Optional::isPresent)
+                .collect(toList()));
         results.put("findMeta", query.findMeta().map(meta -> meta.getImplementation().getName()));
         results.put("findMostSevereTestStepResultBy", query.findAllTestCaseStarted().stream()
                 .map(query::findMostSevereTestStepResultBy)

java/src/test/java/io/cucumber/query/QueryAcceptanceTest.java

@@ -112,6 +112,52 @@
     "Attachments",
     "Attachments"
   ],
+  "findLocationBy" : [
+    {
+      "line" : 12,
+      "column" : 3
+    },
+    {
+      "line" : 18,
+      "column" : 3
+    },
+    {
+      "line" : 21,
+      "column" : 3
+    },
+    {
+      "line" : 24,
+      "column" : 3
+    },
+    {
+      "line" : 30,
+      "column" : 3
+    },
+    {
+      "line" : 33,
+      "column" : 3
+    },
+    {
+      "line" : 36,
+      "column" : 3
+    },
+    {
+      "line" : 44,
+      "column" : 7
+    },
+    {
+      "line" : 45,
+      "column" : 7
+    },
+    {
+      "line" : 47,
+      "column" : 3
+    },
+    {
+      "line" : 50,
+      "column" : 3
+    }
+  ],
   "findMeta" : "fake-cucumber",
   "findMostSevereTestStepResultBy" : [
     "PASSED",

testdata/attachments.feature.query-results.json

@@ -24,6 +24,12 @@
   "findFeatureBy" : [
     "Empty Scenarios"
   ],
+  "findLocationBy" : [
+    {
+      "line" : 7,
+      "column" : 3
+    }
+  ],
   "findMeta" : "fake-cucumber",
   "findMostSevereTestStepResultBy" : [
     null

testdata/empty.feature.query-results.json

@@ -40,6 +40,44 @@
     "Examples Tables",
     "Examples Tables"
   ],
+  "findLocationBy" : [
+    {
+      "line" : 19,
+      "column" : 7
+    },
+    {
+      "line" : 20,
+      "column" : 7
+    },
+    {
+      "line" : 25,
+      "column" : 7
+    },
+    {
+      "line" : 26,
+      "column" : 7
+    },
+    {
+      "line" : 31,
+      "column" : 7
+    },
+    {
+      "line" : 32,
+      "column" : 7
+    },
+    {
+      "line" : 41,
+      "column" : 7
+    },
+    {
+      "line" : 42,
+      "column" : 7
+    },
+    {
+      "line" : 43,
+      "column" : 7
+    }
+  ],
   "findMeta" : "fake-cucumber",
   "findMostSevereTestStepResultBy" : [
     "PASSED",

testdata/examples-tables.feature.query-results.json

@@ -59,6 +59,28 @@
     "6",
     "4"
   ],
+  "findLocationBy" : [
+    {
+      "line" : 6,
+      "column" : 3
+    },
+    {
+      "line" : 9,
+      "column" : 3
+    },
+    {
+      "line" : 12,
+      "column" : 3
+    },
+    {
+      "line" : 16,
+      "column" : 3
+    },
+    {
+      "line" : 20,
+      "column" : 3
+    }
+  ],
   "findMeta" : "fake-cucumber",
   "findMostSevereTestStepResultBy" : [
     "PASSED",

testdata/hooks.feature.query-results.json

@@ -24,6 +24,12 @@
   "findFeatureBy" : [
     "minimal"
   ],
+  "findLocationBy" : [
+    {
+      "line" : 9,
+      "column" : 3
+    }
+  ],
   "findMeta" : "fake-cucumber",
   "findMostSevereTestStepResultBy" : [
     "PASSED"

testdata/minimal.feature.query-results.json

@@ -28,6 +28,20 @@
     "Usage of a `Rule`",
     "Usage of a `Rule`"
   ],
+  "findLocationBy" : [
+    {
+      "line" : 9,
+      "column" : 5
+    },
+    {
+      "line" : 16,
+      "column" : 5
+    },
+    {
+      "line" : 25,
+      "column" : 5
+    }
+  ],
   "findMeta" : "fake-cucumber",
   "findMostSevereTestStepResultBy" : [
     "PASSED",