Fix provider.yaml and test

apache · Dec 4, 2024 · ee43b81 · ee43b81
1 parent 4b965ce
commit ee43b81
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 8 deletions.
diff --git a/providers/src/airflow/providers/elasticsearch/provider.yaml b/providers/src/airflow/providers/elasticsearch/provider.yaml
@@ -86,7 +86,7 @@ hooks:
       - airflow.providers.elasticsearch.hooks.elasticsearch
 
 connection-types:
-  - hook-class-name: airflow.providers.elasticsearch.hooks.elasticsearch.ElasticsearchHook
+  - hook-class-name: airflow.providers.elasticsearch.hooks.elasticsearch.ElasticsearchSQLHook
     connection-type: elasticsearch
 
 logging:

diff --git a/providers/tests/elasticsearch/hooks/test_elasticsearch.py b/providers/tests/elasticsearch/hooks/test_elasticsearch.py
@@ -34,10 +34,10 @@ class TestElasticsearchSQLHookConn:
     def setup_method(self):
         self.connection = Connection(host="localhost", port=9200, schema="http")
 
-        class UnitTestElasticsearchHook(ElasticsearchSQLHook):
+        class UnitTestElasticsearchSQLHook(ElasticsearchSQLHook):
             conn_name_attr = "elasticsearch_conn_id"
 
-        self.db_hook = UnitTestElasticsearchHook()
+        self.db_hook = UnitTestElasticsearchSQLHook()
         self.db_hook.get_connection = mock.Mock()
         self.db_hook.get_connection.return_value = self.connection
 
@@ -55,13 +55,13 @@ def setup_method(self):
         self.conn.cursor.return_value = self.cur
         conn = self.conn
 
-        class UnitTestElasticsearchHook(ElasticsearchSQLHook):
+        class UnitTestElasticsearchSQLHook(ElasticsearchSQLHook):
             conn_name_attr = "test_conn_id"
 
             def get_conn(self):
                 return conn
 
-        self.db_hook = UnitTestElasticsearchHook()
+        self.db_hook = UnitTestElasticsearchSQLHook()
 
     def test_get_first_record(self):
         statement = "SQL"

diff --git a/tests/always/test_project_structure.py b/tests/always/test_project_structure.py
@@ -526,9 +526,6 @@ class TestElasticsearchProviderProjectStructure(ExampleCoverageTest):
     PROVIDER = "elasticsearch"
     CLASS_DIRS = {"hooks"}
     CLASS_SUFFIXES = ["Hook"]
-    DEPRECATED_CLASSES = {
-        "airflow.providers.elasticsearch.hooks.elasticsearch.ElasticsearchHook",
-    }
 
 
 class TestCncfProviderProjectStructure(ExampleCoverageTest):