Function test_jsonb_insert_function

src/functions/json_functions.rs:1912–2047

()

    
    #[test]
    fn test_jsonb_insert_function() {
        let conn = Connection::open_in_memory().unwrap();
        register_json_functions(&conn).unwrap();
        
        // Test inserting into object
        let test_json = r#"{"name": "John", "age": 30}"#;
        let result: Option<String> = conn.query_row(
            "SELECT jsonb_insert(?, ?, ?)",
            [test_json, "{email}", "\"john@example.com\""],
            |row| row.get(0)
        ).unwrap();
        
        assert!(result.is_some());
        let parsed: JsonValue = serde_json::from_str(&result.unwrap()).unwrap();
        match parsed {
            JsonValue::Object(map) => {
                assert_eq!(map.get("name"), Some(&JsonValue::String("John".to_string())));
                assert_eq!(map.get("age"), Some(&JsonValue::Number(serde_json::Number::from(30))));
                assert_eq!(map.get("email"), Some(&JsonValue::String("john@example.com".to_string())));
            }
            _ => panic!("Expected JSON object"),
        }
        
        // Test inserting into nested object
        let nested_json = r#"{"user": {"name": "Alice"}, "active": true}"#;
        let result: Option<String> = conn.query_row(
            "SELECT jsonb_insert(?, ?, ?)",
            [nested_json, "{user,email}", "\"alice@example.com\""],
            |row| row.get(0)
        ).unwrap();
        
        assert!(result.is_some());
        let parsed: JsonValue = serde_json::from_str(&result.unwrap()).unwrap();
        match parsed {
            JsonValue::Object(map) => {
                if let Some(JsonValue::Object(user_map)) = map.get("user") {
                    assert_eq!(user_map.get("name"), Some(&JsonValue::String("Alice".to_string())));
                    assert_eq!(user_map.get("email"), Some(&JsonValue::String("alice@example.com".to_string())));
                } else {
                    panic!("Expected nested user object");
                }
            }
            _ => panic!("Expected JSON object"),
        }
        
        // Test inserting into array (before index)
        let array_json = r#"["apple", "banana", "cherry"]"#;
        let result: Option<String> = conn.query_row(
            "SELECT jsonb_insert(?, ?, ?)",
            [array_json, "{1}", "\"orange\""],
            |row| row.get(0)
        ).unwrap();
        
        assert!(result.is_some());
        let parsed: JsonValue = serde_json::from_str(&result.unwrap()).unwrap();
        match parsed {
            JsonValue::Array(arr) => {
                assert_eq!(arr.len(), 4);