add an integration test

Author: idegtiarenko

x-pack/plugin/esql/src/internalClusterTest/java/org/elasticsearch/xpack/esql/action/CrossClusterLookupJoinIT.java

@@ -11,6 +11,9 @@
 import org.elasticsearch.action.admin.indices.alias.IndicesAliasesRequestBuilder;
 import org.elasticsearch.client.internal.Client;
 import org.elasticsearch.common.settings.Settings;
+import org.elasticsearch.index.IndexMode;
+import org.elasticsearch.index.IndexSettings;
+import org.elasticsearch.index.query.TermQueryBuilder;
 import org.elasticsearch.xpack.core.enrich.EnrichPolicy;
 import org.elasticsearch.xpack.core.enrich.action.ExecuteEnrichPolicyAction;
 import org.elasticsearch.xpack.core.enrich.action.PutEnrichPolicyAction;
@@ -28,6 +31,7 @@
 
 import static org.elasticsearch.test.hamcrest.ElasticsearchAssertions.assertAcked;
 import static org.elasticsearch.xpack.esql.EsqlTestUtils.getValuesList;
+import static org.elasticsearch.xpack.esql.action.EsqlQueryRequest.syncEsqlQueryRequest;
 import static org.hamcrest.Matchers.containsString;
 import static org.hamcrest.Matchers.empty;
 import static org.hamcrest.Matchers.equalTo;
@@ -557,6 +561,42 @@ public void testLookupJoinFieldTypes() throws IOException {
         }
     }
 
+    public void testLookupJoinRetryAnalysis() throws IOException {
+        setupClusters(3);
+        setSkipUnavailable(REMOTE_CLUSTER_1, false);
+        setSkipUnavailable(REMOTE_CLUSTER_2, false);
+
+        var defaultSettings = Settings.builder();
+        createIndexWithDocument(LOCAL_CLUSTER, "data", defaultSettings, Map.of("key", 1, "f1", 1));
+        createIndexWithDocument(REMOTE_CLUSTER_1, "data", defaultSettings, Map.of("key", 2, "f2", 2));
+        createIndexWithDocument(REMOTE_CLUSTER_2, "data", defaultSettings, Map.of("key", 3, "f3", 3));
+
+        var lookupSettings = Settings.builder().put(IndexSettings.MODE.getKey(), IndexMode.LOOKUP);
+        createIndexWithDocument(LOCAL_CLUSTER, "lookup", lookupSettings, Map.of("key", 1, "location", "local"));
+        createIndexWithDocument(REMOTE_CLUSTER_1, "lookup", lookupSettings, Map.of("key", 2, "location", "remote-1"));
+        // lookup is intentionally absent on REMOTE_CLUSTER_2
+
+        // The following query uses filter f2=2 that narrows down execution only to REMOTE_CLUSTER_1 index however,
+        // later it uses `WHERE f1 == 1` esql condition that to an attribute present only on the local cluster index.
+        // This causes analysis to fail and retry the entire query without a filter.
+        // The second analysis executes against all cluster indices and should discover that lookup is absent on REMOTE_CLUSTER_2.
+        expectThrows(
+            VerificationException.class,
+            containsString("lookup index [lookup] is not available in remote cluster [remote-b]"),
+            () -> runQuery(
+                syncEsqlQueryRequest().query("FROM data,*:data | LOOKUP JOIN lookup ON key | WHERE f1 == 1")
+                    .filter(new TermQueryBuilder("f2", 2))
+            )
+        );
+    }
+
+    private void createIndexWithDocument(String clusterAlias, String indexName, Settings.Builder settings, Map<String, Object> source) {
+        var client = client(clusterAlias);
+        client.admin().indices().prepareCreate(indexName).setSettings(settings).get();
+        client.prepareIndex(indexName).setSource(source).get();
+        client.admin().indices().prepareRefresh(indexName).get();
+    }
+
     protected Map<String, Object> setupClustersAndLookups() throws IOException {
         var setupData = setupClusters(2);
         populateLookupIndex(LOCAL_CLUSTER, "values_lookup", 10);