dev: add test cases for BEGIN & END statements

Author: riazus

libsql-server/tests/namespaces/dumps.rs

@@ -425,3 +425,171 @@ fn load_dump_with_invalid_sql() {
 
     sim.run().unwrap();
 }
+
+#[test]
+fn load_dump_with_trigger() {
+    const DUMP: &str = r#"
+    BEGIN TRANSACTION;
+    CREATE TABLE test (x);
+    CREATE TRIGGER simple_trigger 
+    AFTER INSERT ON test 
+    BEGIN
+        INSERT INTO test VALUES (999);
+    END;
+    INSERT INTO test VALUES (1);
+    COMMIT;"#;
+
+    let mut sim = Builder::new()
+        .simulation_duration(Duration::from_secs(1000))
+        .build();
+    let tmp = tempdir().unwrap();
+    let tmp_path = tmp.path().to_path_buf();
+
+    std::fs::write(tmp_path.join("dump.sql"), DUMP).unwrap();
+
+    make_primary(&mut sim, tmp.path().to_path_buf());
+
+    sim.client("client", async move {
+        let client = Client::new();
+
+        let resp = client
+            .post(
+                "http://primary:9090/v1/namespaces/debug_test/create",
+                json!({ "dump_url": format!("file:{}", tmp_path.join("dump.sql").display())}),
+            )
+            .await
+            .unwrap();
+        assert_eq!(resp.status(), StatusCode::OK);
+
+        let db = Database::open_remote_with_connector("http://debug_test.primary:8080", "", TurmoilConnector)?;
+        let conn = db.connect()?;
+
+        // Original INSERT: 1, Trigger INSERT: 999 = 2 total rows
+        let mut rows = conn.query("SELECT COUNT(*) FROM test", ()).await?;
+        let row = rows.next().await?.unwrap();
+        assert_eq!(row.get::<i64>(0)?, 2);
+
+        Ok(())
+    });
+
+    sim.run().unwrap();
+}
+
+#[test]
+fn load_dump_with_case_trigger() {
+    const DUMP: &str = r#"
+    BEGIN TRANSACTION;
+    CREATE TABLE test (id INTEGER, rate REAL DEFAULT 0.0);
+    CREATE TRIGGER case_trigger 
+    AFTER INSERT ON test 
+    BEGIN 
+        UPDATE test 
+        SET rate = 
+            CASE 
+                WHEN NEW.id = 1 
+                    THEN 0.1 
+                ELSE 0.0 
+            END 
+        WHERE id = NEW.id; 
+    END;
+
+    INSERT INTO test (id) VALUES (1);
+    COMMIT;"#;
+
+    let mut sim = Builder::new()
+        .simulation_duration(Duration::from_secs(1000))
+        .build();
+    let tmp = tempdir().unwrap();
+    let tmp_path = tmp.path().to_path_buf();
+
+    std::fs::write(tmp_path.join("dump.sql"), DUMP).unwrap();
+
+    make_primary(&mut sim, tmp.path().to_path_buf());
+
+    sim.client("client", async move {
+        let client = Client::new();
+
+        let resp = client
+            .post(
+                "http://primary:9090/v1/namespaces/case_test/create",
+                json!({ "dump_url": format!("file:{}", tmp_path.join("dump.sql").display())}),
+            )
+            .await
+            .unwrap();
+        assert_eq!(resp.status(), StatusCode::OK);
+
+        let db = Database::open_remote_with_connector("http://case_test.primary:8080", "", TurmoilConnector)?;
+        let conn = db.connect()?;
+
+        let mut rows = conn.query("SELECT id, rate FROM test", ()).await?;
+        let row = rows.next().await?.unwrap();
+        assert_eq!(row.get::<i64>(0)?, 1);
+        assert!((row.get::<f64>(1)? - 0.1).abs() < 0.001);
+
+        Ok(())
+    });
+
+    sim.run().unwrap();
+}
+
+#[test]
+fn load_dump_with_nested_case() {
+    const DUMP: &str = r#"
+    BEGIN TRANSACTION;
+    CREATE TABLE orders (id INTEGER, amount REAL, status TEXT);
+    CREATE TRIGGER nested_trigger 
+    AFTER UPDATE ON orders 
+    BEGIN 
+        UPDATE orders 
+        SET amount = 
+            CASE 
+                WHEN NEW.status = 'completed' 
+                    THEN 
+                        CASE
+                            WHEN OLD.id = 1
+                                THEN OLD.amount * 0.9
+                            ELSE OLD.amount * 0.8
+                        END
+                ELSE OLD.amount 
+            END 
+        WHERE id = NEW.id; 
+    END;
+    
+    INSERT INTO orders (id, amount, status) VALUES (1, 100.0, 'pending');
+    COMMIT;"#;
+
+    let mut sim = Builder::new()
+        .simulation_duration(Duration::from_secs(1000))
+        .build();
+    let tmp = tempdir().unwrap();
+    let tmp_path = tmp.path().to_path_buf();
+
+    std::fs::write(tmp_path.join("dump.sql"), DUMP).unwrap();
+
+    make_primary(&mut sim, tmp.path().to_path_buf());
+
+    sim.client("client", async move {
+        let client = Client::new();
+
+        let resp = client
+            .post(
+                "http://primary:9090/v1/namespaces/nested_test/create",
+                json!({ "dump_url": format!("file:{}", tmp_path.join("dump.sql").display())}),
+            )
+            .await
+            .unwrap();
+        assert_eq!(resp.status(), StatusCode::OK);
+
+        let db = Database::open_remote_with_connector("http://nested_test.primary:8080", "", TurmoilConnector)?;
+        let conn = db.connect()?;
+
+        conn.execute("UPDATE orders SET status = 'completed' WHERE id = 1", ()).await?;
+        let mut rows = conn.query("SELECT amount FROM orders WHERE id = 1", ()).await?;
+        let row = rows.next().await?.unwrap();
+        assert!((row.get::<f64>(0)? - 90.0).abs() < 0.001);
+
+        Ok(())
+    });
+
+    sim.run().unwrap();
+}