add a lotta tests (cursor)

Author: Kakapio

src/backend.rs

@@ -203,4 +203,329 @@ mod tests {
             ]
         );
     }
+
+    // Testing select with no rows in table
+    #[test]
+    fn execute_select_empty_table() {
+        let mut table = Table::new();
+        let mut statement = Statement::default();
+        prepare_statement("select", &mut statement);
+        let result = execute_statement(statement, &mut table);
+
+        if let ExecuteResult::Success(Some(rows)) = result {
+            assert_eq!(rows.len(), 0);
+        } else {
+            panic!("Expected Success with empty vector");
+        }
+    }
+
+    // Testing select with specific ID that doesn't exist
+    #[test]
+    fn execute_select_nonexistent_id() {
+        let mut table = Table::new();
+        do_sql_cmd(&mut table, "insert 13 rosh [email protected]");
+
+        let mut statement = Statement::default();
+        prepare_statement("select 42", &mut statement);
+        let result = execute_statement(statement, &mut table);
+
+        if let ExecuteResult::Success(Some(rows)) = result {
+            assert_eq!(rows.len(), 0);
+        } else {
+            panic!("Expected Success with empty vector");
+        }
+    }
+
+    // Testing select with specific ID that exists
+    #[test]
+    fn execute_select_existing_id() {
+        let mut table = Table::new();
+        do_sql_cmd(&mut table, "insert 13 rosh [email protected]");
+        do_sql_cmd(&mut table, "insert 42 stefan [email protected]");
+
+        let mut statement = Statement::default();
+        prepare_statement("select 42", &mut statement);
+        let result = execute_statement(statement, &mut table);
+
+        if let ExecuteResult::Success(Some(rows)) = result {
+            assert_eq!(rows.len(), 1);
+            assert_eq!(rows[0].id, 42);
+            assert_eq!(rows[0].username, "stefan");
+            assert_eq!(rows[0].email, "[email protected]");
+        } else {
+            panic!("Expected Success with one row");
+        }
+    }
+
+    // Testing select all with multiple rows
+    #[test]
+    fn execute_select_all_multiple_rows() {
+        let mut table = Table::new();
+        do_sql_cmd(&mut table, "insert 13 rosh [email protected]");
+        do_sql_cmd(&mut table, "insert 42 stefan [email protected]");
+        do_sql_cmd(
+            &mut table,
+            "insert 1699 sniper_penut [email protected]",
+        );
+
+        let mut statement = Statement::default();
+        prepare_statement("select", &mut statement);
+        let result = execute_statement(statement, &mut table);
+
+        if let ExecuteResult::Success(Some(rows)) = result {
+            assert_eq!(rows.len(), 3);
+        } else {
+            panic!("Expected Success with three rows");
+        }
+    }
+
+    // Testing insert with duplicate ID
+    #[test]
+    fn execute_insert_duplicate_id() {
+        let mut table = Table::new();
+        do_sql_cmd(&mut table, "insert 13 rosh [email protected]");
+        do_sql_cmd(&mut table, "insert 13 stefan [email protected]");
+
+        // The second insert should be added as a new row
+        assert_eq!(
+            table.data,
+            vec![
+                Row {
+                    id: 13,
+                    username: "rosh".to_string(),
+                    email: "[email protected]".to_string()
+                },
+                Row {
+                    id: 13,
+                    username: "stefan".to_string(),
+                    email: "[email protected]".to_string()
+                }
+            ]
+        );
+    }
+
+    // Testing insert with zero ID
+    #[test]
+    fn execute_insert_zero_id() {
+        let mut table = Table::new();
+        do_sql_cmd(&mut table, "insert 0 rosh [email protected]");
+
+        assert_eq!(
+            table.data,
+            vec![Row {
+                id: 0,
+                username: "rosh".to_string(),
+                email: "[email protected]".to_string()
+            }]
+        );
+    }
+
+    // Testing insert with maximum u32 ID
+    #[test]
+    fn execute_insert_max_id() {
+        let mut table = Table::new();
+        do_sql_cmd(&mut table, "insert 4294967295 rosh [email protected]");
+
+        assert_eq!(
+            table.data,
+            vec![Row {
+                id: 4294967295,
+                username: "rosh".to_string(),
+                email: "[email protected]".to_string()
+            }]
+        );
+    }
+
+    // Testing select with zero ID
+    #[test]
+    fn execute_select_zero_id() {
+        let mut table = Table::new();
+        do_sql_cmd(&mut table, "insert 0 rosh [email protected]");
+
+        let mut statement = Statement::default();
+        prepare_statement("select 0", &mut statement);
+        let result = execute_statement(statement, &mut table);
+
+        if let ExecuteResult::Success(Some(rows)) = result {
+            assert_eq!(rows.len(), 1);
+            assert_eq!(rows[0].id, 0);
+        } else {
+            panic!("Expected Success with one row");
+        }
+    }
+
+    // Testing select with maximum u32 ID
+    #[test]
+    fn execute_select_max_id() {
+        let mut table = Table::new();
+        do_sql_cmd(&mut table, "insert 4294967295 rosh [email protected]");
+
+        let mut statement = Statement::default();
+        prepare_statement("select 4294967295", &mut statement);
+        let result = execute_statement(statement, &mut table);
+
+        if let ExecuteResult::Success(Some(rows)) = result {
+            assert_eq!(rows.len(), 1);
+            assert_eq!(rows[0].id, 4294967295);
+        } else {
+            panic!("Expected Success with one row");
+        }
+    }
+
+    // Testing insert with special characters in username
+    #[test]
+    fn execute_insert_special_chars_username() {
+        let mut table = Table::new();
+        do_sql_cmd(&mut table, "insert 13 rosh!@#$ [email protected]");
+
+        assert_eq!(
+            table.data,
+            vec![Row {
+                id: 13,
+                username: "rosh!@#$".to_string(),
+                email: "[email protected]".to_string()
+            }]
+        );
+    }
+
+    // Testing insert with special characters in email
+    #[test]
+    fn execute_insert_special_chars_email() {
+        let mut table = Table::new();
+        do_sql_cmd(&mut table, "insert 13 rosh [email protected]");
+
+        assert_eq!(
+            table.data,
+            vec![Row {
+                id: 13,
+                username: "rosh".to_string(),
+                email: "[email protected]".to_string()
+            }]
+        );
+    }
+
+    // Testing insert with very long username
+    #[test]
+    fn execute_insert_long_username() {
+        let mut table = Table::new();
+        let long_username = "a".repeat(1000);
+        do_sql_cmd(
+            &mut table,
+            &format!("insert 13 {} [email protected]", long_username),
+        );
+
+        assert_eq!(
+            table.data,
+            vec![Row {
+                id: 13,
+                username: long_username,
+                email: "[email protected]".to_string()
+            }]
+        );
+    }
+
+    // Testing insert with very long email
+    #[test]
+    fn execute_insert_long_email() {
+        let mut table = Table::new();
+        let long_email = format!("{}@gmail.com", "a".repeat(1000));
+        do_sql_cmd(&mut table, &format!("insert 13 rosh {}", long_email));
+
+        assert_eq!(
+            table.data,
+            vec![Row {
+                id: 13,
+                username: "rosh".to_string(),
+                email: long_email,
+            }]
+        );
+    }
+
+    // Testing multiple selects on the same table
+    #[test]
+    fn execute_multiple_selects() {
+        let mut table = Table::new();
+        do_sql_cmd(&mut table, "insert 13 rosh [email protected]");
+        do_sql_cmd(&mut table, "insert 42 stefan [email protected]");
+        do_sql_cmd(
+            &mut table,
+            "insert 1699 sniper_penut [email protected]",
+        );
+
+        // Select all
+        let mut statement = Statement::default();
+        prepare_statement("select", &mut statement);
+        let result = execute_statement(statement, &mut table);
+
+        if let ExecuteResult::Success(Some(rows)) = result {
+            assert_eq!(rows.len(), 3);
+        } else {
+            panic!("Expected Success with three rows");
+        }
+
+        // Select specific ID
+        let mut statement = Statement::default();
+        prepare_statement("select 42", &mut statement);
+        let result = execute_statement(statement, &mut table);
+
+        if let ExecuteResult::Success(Some(rows)) = result {
+            assert_eq!(rows.len(), 1);
+            assert_eq!(rows[0].id, 42);
+        } else {
+            panic!("Expected Success with one row");
+        }
+    }
+
+    // Testing insert and select with whitespace
+    #[test]
+    fn execute_insert_select_whitespace() {
+        let mut table = Table::new();
+        do_sql_cmd(&mut table, "  insert  13  rosh  [email protected]  ");
+
+        let mut statement = Statement::default();
+        prepare_statement("  select  13  ", &mut statement);
+        let result = execute_statement(statement, &mut table);
+
+        if let ExecuteResult::Success(Some(rows)) = result {
+            assert_eq!(rows.len(), 0);
+        } else {
+            panic!("Expected Success with empty vector");
+        }
+    }
+
+    // Testing insert and select with tab characters
+    #[test]
+    fn execute_insert_select_tab() {
+        let mut table = Table::new();
+        do_sql_cmd(&mut table, "insert\t13\trosh\t[email protected]");
+
+        let mut statement = Statement::default();
+        prepare_statement("select\t13", &mut statement);
+        let result = execute_statement(statement, &mut table);
+
+        if let ExecuteResult::Success(Some(rows)) = result {
+            assert_eq!(rows.len(), 1);
+            assert_eq!(rows[0].id, 13);
+        } else {
+            panic!("Expected Success with one row");
+        }
+    }
+
+    // Testing insert and select with newline characters
+    #[test]
+    fn execute_insert_select_newline() {
+        let mut table = Table::new();
+        do_sql_cmd(&mut table, "insert\n13\nrosh\n[email protected]");
+
+        let mut statement = Statement::default();
+        prepare_statement("select\n13", &mut statement);
+        let result = execute_statement(statement, &mut table);
+
+        if let ExecuteResult::Success(Some(rows)) = result {
+            assert_eq!(rows.len(), 1);
+            assert_eq!(rows[0].id, 13);
+        } else {
+            panic!("Expected Success with one row");
+        }
+    }
 }