feat: current state

tests/core_contract.rs

@@ -0,0 +1,80 @@
+use azure_database_tui::{
+    domain::{
+        identifiers::{quote_identifier, quote_qualified},
+        profile::{ConnectionDraft, TlsPolicy},
+        sql::{CopyDirection, SqlLexer},
+        table::{DropTableConfirmation, InsertFieldValue, InsertForm},
+    },
+    services::tables::TableService,
+};
+
+#[test]
+fn identifiers_are_quoted_per_segment() {
+    assert_eq!(
+        quote_identifier("Order\"Items").unwrap(),
+        "\"Order\"\"Items\""
+    );
+    assert_eq!(
+        quote_qualified("Sales", "Order Items").unwrap().0,
+        "\"Sales\".\"Order Items\""
+    );
+}
+
+#[test]
+fn profile_draft_rejects_empty_values() {
+    let draft = ConnectionDraft {
+        azure_resource_id: " ".into(),
+        server_host: "host".into(),
+        port: 5432,
+        database: "db".into(),
+        username: "user".into(),
+        tls: TlsPolicy::SystemTrust,
+        save_secret: false,
+    };
+    assert!(draft.validate().is_err());
+}
+
+#[test]
+fn lexer_keeps_semicolons_inside_lexical_regions() {
+    let sql = "select ';'; -- ;\n select $$;$$; /* outer /* ; */ */ select 3;";
+    assert_eq!(SqlLexer::split(sql).len(), 3);
+    assert_eq!(
+        SqlLexer::classify_copy("COPY t FROM STDIN"),
+        Some(CopyDirection::FromStdin)
+    );
+    assert_eq!(
+        SqlLexer::classify_copy("copy t to stdout"),
+        Some(CopyDirection::ToStdout)
+    );
+}
+
+#[test]
+fn insert_values_are_not_interpolated_and_drop_needs_exact_name() {
+    let form = InsertForm {
+        schema: "public".into(),
+        relation: "orders".into(),
+        values: vec![
+            (
+                "amount".into(),
+                InsertFieldValue::Text("1'); drop table x; --".into()),
+            ),
+            ("note".into(), InsertFieldValue::Null),
+            ("created".into(), InsertFieldValue::UseDefault),
+        ],
+    };
+    let (sql, values) = TableService::insert_sql(&form).unwrap();
+    assert_eq!(
+        sql,
+        "INSERT INTO \"public\".\"orders\" (\"amount\", \"note\") VALUES ($1, NULL)"
+    );
+    assert_eq!(values.len(), 1);
+    assert!(!sql.contains("drop table"));
+    let confirmation = DropTableConfirmation {
+        canonical_name: "\"public\".\"orders\"".into(),
+        typed_name: "\"public\".\"orders\"".into(),
+    };
+    assert_eq!(
+        TableService::drop_sql("public", "orders", &confirmation).unwrap(),
+        "DROP TABLE \"public\".\"orders\""
+    );
+}