[DOP-26403] Set output type based on SQL query type

Author: dolfinus

data_rentgen/consumer/extractors/generic/io.py

@@ -165,7 +165,7 @@ def _extract_output_type(
     def _extract_output_type_from_sql(self, sql: str) -> OutputTypeDTO | None:
         found = SQL_QUERY_SYNTAX.search(sql)
         if found:
-            return QUERY_TYPE_TO_OUTPUT_TYPE[found.group("query_type")]
+            return QUERY_TYPE_TO_OUTPUT_TYPE[found.group("query_type").upper()]
         found = ALTER_SYNTAX.search(sql)
         if found:
             return OutputTypeDTO.ALTER

tests/test_consumer/test_extractors/test_extractors_interaction.py

@@ -261,6 +261,7 @@ def test_extractors_extract_output_batch_with_lifecycle(
     [
         ("CREATE TABLE AS SELECT * FROM mytable", OutputTypeDTO.CREATE),
         ("INSERT INTO mytable SELECT * FROM mytable", OutputTypeDTO.APPEND),
+        ("insert into mytable select * from mytable", OutputTypeDTO.APPEND),
         ("UPDATE mytable SET a=1", OutputTypeDTO.UPDATE),
         ("DELETE FROM mytable", OutputTypeDTO.DELETE),
         ("COPY mytable FROM '...'", OutputTypeDTO.APPEND),