Handle ROW and triple quotes

Author: felixbarny

x-pack/plugin/esql/qa/testFixtures/src/main/java/org/elasticsearch/xpack/esql/EsqlTestUtils.java

@@ -1298,8 +1298,8 @@ public static String addRemoteIndices(String query, Set<String> lookupIndices, b
         String[] commandParts = afterSetStatements.trim().split("\\s+", 2);
 
         String command = commandParts[0].trim();
-        String commandArgs = commandParts[1].trim();
         if (SourceCommand.isSourceCommand(command)) {
+            String commandArgs = commandParts[1].trim();
             String[] indices = new EsqlParser().createStatement(afterSetStatements)
                 .collect(UnresolvedRelation.class)
                 .getFirst()
@@ -1332,7 +1332,7 @@ private static String getStuffAfterIndices(String commandArgs, String lastIndex)
         } else {
             stuffAfterIndices = commandArgs.trim();
         }
-        if (stuffAfterIndices.startsWith("\"")) {
+        while (stuffAfterIndices.startsWith("\"")) {
             stuffAfterIndices = stuffAfterIndices.substring(1).trim();
         }
         return stuffAfterIndices;

x-pack/plugin/esql/qa/testFixtures/src/test/java/org/elasticsearch/xpack/esql/EsqlTestUtilsTests.java

@@ -88,4 +88,24 @@ public void testSubquery() {
                 | KEEP _index,  emp_no, languages, language_name""")
         );
     }
+
+    public void testTripleQuotes() {
+        assertThat(
+            EsqlTestUtils.addRemoteIndices("from \"\"\"employees\"\"\" | limit 2", Set.of(), false),
+            equalTo("from *:employees,employees | limit 2")
+        );
+    }
+
+    public void testRow() {
+        assertThat(
+            EsqlTestUtils.addRemoteIndices("""
+                ROW a = "1953-01-23T12:15:00Z - some text - 127.0.0.1;"\s
+                 | DISSECT a "%{Y}-%{M}-%{D}T%{h}:%{m}:%{s}Z - %{msg} - %{ip};"\s
+                 | KEEP Y, M, D, h, m, s, msg, ip""", Set.of(), false),
+            equalTo("""
+                ROW a = "1953-01-23T12:15:00Z - some text - 127.0.0.1;"\s
+                 | DISSECT a "%{Y}-%{M}-%{D}T%{h}:%{m}:%{s}Z - %{msg} - %{ip};"\s
+                 | KEEP Y, M, D, h, m, s, msg, ip""")
+        );
+    }
 }