add tests

Author: pashagolub

src/lib.rs

@@ -771,4 +771,62 @@ mod tests {
 
         assert_eq!(Err(spi::SpiError::InvalidPosition), query_result);
     }
+
+    #[pg_test]
+    fn test_select_value_only_with_key_filter() {
+        // Test for issue #26: selecting only value column with WHERE clause on key
+        let (_container, url) = create_container();
+
+        create_fdt(url);
+
+        // Insert test data
+        Spi::run("INSERT INTO test (key, value) VALUES ('key1','value1'),('key2','value2'),('key3','value3')")
+            .expect("INSERT should work");
+
+        // Test 1: SELECT value WHERE key = 'key2' should return the value
+        let query_result = Spi::get_one::<String>("SELECT value FROM test WHERE key = 'key2'")
+            .expect("SELECT value with key filter should work");
+
+        assert_eq!(Some(format!("value2")), query_result);
+
+        // Test 2: SELECT key WHERE key = 'key1' should also work
+        let query_result = Spi::get_one::<String>("SELECT key FROM test WHERE key = 'key1'")
+            .expect("SELECT key with key filter should work");
+
+        assert_eq!(Some(format!("key1")), query_result);
+
+        // Test 3: SELECT * WHERE key = 'key3' should work (baseline)
+        let query_result = Spi::get_two::<String, String>("SELECT * FROM test WHERE key = 'key3'")
+            .expect("SELECT * with key filter should work");
+
+        assert_eq!((Some(format!("key3")), Some(format!("value3"))), query_result);
+    }
+
+    #[pg_test]
+    fn test_update_value_only_with_key_filter() {
+        // Test for issue #26: UPDATE with only value column when key is in WHERE clause
+        let (_container, url) = create_container();
+
+        create_fdt(url);
+
+        // Insert test data
+        Spi::run("INSERT INTO test (key, value) VALUES ('gather/78','original_value'),('gather/84','other_value')")
+            .expect("INSERT should work");
+
+        // Update without including key column in SET clause
+        Spi::run("UPDATE test SET value = 'updated_value' WHERE key = 'gather/84'")
+            .expect("UPDATE with key filter should work");
+
+        // Verify the update worked
+        let query_result = Spi::get_one::<String>("SELECT value FROM test WHERE key = 'gather/84'")
+            .expect("SELECT after UPDATE should work");
+
+        assert_eq!(Some(format!("updated_value")), query_result);
+
+        // Verify other row was not affected
+        let query_result = Spi::get_one::<String>("SELECT value FROM test WHERE key = 'gather/78'")
+            .expect("SELECT other row should work");
+
+        assert_eq!(Some(format!("original_value")), query_result);
+    }
 }