From 8dbcbb3c277d81968b0843b03545de958ac9d2ef Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Marko=20Milenkovi=C4=87?=
 <milenkovicm@users.noreply.github.com>
Date: Tue, 10 Sep 2024 20:56:26 +0100
Subject: [PATCH] minor: Add `databricks` dialect to `dialect_from_str` (#1416)

Co-authored-by: Andrew Lamb <andrew@nerdnetworks.org>
---
 src/dialect/mod.rs | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/dialect/mod.rs b/src/dialect/mod.rs
index 0be8c17c..6b80243f 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,
     }
 }
@@ -638,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());