From 27cf3deaa67301d55513322dbbdfdc0dfa1c08a7 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Marko=20Milenkovi=C4=87?=
 <milenkovicm@users.noreply.github.com>
Date: Sun, 8 Sep 2024 07:53:53 +0100
Subject: [PATCH 1/2] Add dabricks dialect to `dialect_from_str`

---
 src/dialect/mod.rs | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/dialect/mod.rs b/src/dialect/mod.rs
index 0be8c17c7..ace190507 100644
--- a/src/dialect/mod.rs
+++ b/src/dialect/mod.rs
@@ -587,6 +587,7 @@ pub fn dialect_from_str(dialect_name: impl AsRef<str>) -> Option<Box<dyn Dialect
         "bigquery" => Some(Box::new(BigQueryDialect)),
         "ansi" => Some(Box::new(AnsiDialect {})),
         "duckdb" => Some(Box::new(DuckDbDialect {})),
+        "databricks" => Some(Box::new(DatabricksDialect {})),
         _ => None,
     }
 }

From 9d3a37bc9d8ae54a438d207834eb875a0d1c5ace Mon Sep 17 00:00:00 2001
From: Andrew Lamb <andrew@nerdnetworks.org>
Date: Tue, 10 Sep 2024 15:52:13 -0400
Subject: [PATCH 2/2] Add test

---
 src/dialect/mod.rs | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/dialect/mod.rs b/src/dialect/mod.rs
index ace190507..6b80243ff 100644
--- a/src/dialect/mod.rs
+++ b/src/dialect/mod.rs
@@ -639,6 +639,8 @@ mod tests {
         assert!(parse_dialect("ANSI").is::<AnsiDialect>());
         assert!(parse_dialect("duckdb").is::<DuckDbDialect>());
         assert!(parse_dialect("DuckDb").is::<DuckDbDialect>());
+        assert!(parse_dialect("DataBricks").is::<DatabricksDialect>());
+        assert!(parse_dialect("databricks").is::<DatabricksDialect>());
 
         // error cases
         assert!(dialect_from_str("Unknown").is_none());