From 5e594fbd9f899e771c9bd3c7c1dcc1c101887900 Mon Sep 17 00:00:00 2001
From: andrew <elkin.andr@gmail.com>
Date: Mon, 11 Nov 2024 11:01:38 +0300
Subject: [PATCH] test single_integration

---
 tests/test_planner/test_integration_select.py | 43 ++++++++++++++++++-
 1 file changed, 42 insertions(+), 1 deletion(-)

diff --git a/tests/test_planner/test_integration_select.py b/tests/test_planner/test_integration_select.py
index 5180465..faa8831 100644
--- a/tests/test_planner/test_integration_select.py
+++ b/tests/test_planner/test_integration_select.py
@@ -554,7 +554,7 @@ def test_select_from_table_subselect_api_integration(self):
         plan = plan_query(
             query,
             integrations=[{'name': 'int1', 'class_type': 'api', 'type': 'data'}],
-            predictor_metadata=[{'name': 'pred', 'integration_name': 'mindsdb'}]
+            predictor_metadata=[{'name': 'pred', 'integration_name': 'mindsdb'}],
         )
 
         assert plan.steps == expected_plan.steps
@@ -583,6 +583,47 @@ def test_select_from_table_subselect_sql_integration(self):
 
         assert plan.steps == expected_plan.steps
 
+    def test_select_from_single_integration(self):
+        sql_parsed = '''
+            with tab2 as (
+              select * from int1.tabl2
+            )
+            select x from tab2
+            join int1.tab1 on 0=0
+            where x1 in (select id from int1.tab1)
+            limit 1
+        '''
+
+        sql_integration = '''
+            with tab2 as (
+              select * from tabl2
+            )
+            select x from tab2
+            join tab1 on 0=0
+            where x1 in (select id as id from tab1)
+            limit 1
+        '''
+        query = parse_sql(sql_parsed, dialect='mindsdb')
+
+        expected_plan = QueryPlan(
+            predictor_namespace='mindsdb',
+            steps=[
+                FetchDataframeStep(
+                    integration='int1',
+                    query=parse_sql(sql_integration),
+                ),
+            ],
+        )
+
+        plan = plan_query(
+            query,
+            integrations=[{'name': 'int1', 'class_type': 'sql', 'type': 'data'}],
+            predictor_metadata=[{'name': 'pred', 'integration_name': 'mindsdb'}],
+            default_namespace='mindsdb',
+        )
+
+        assert plan.steps == expected_plan.steps
+
     def test_delete_from_table_subselect_api_integration(self):
         query = parse_sql('''
             delete from int1.tab1