Function test_row_dml

crates/core/src/sql/execute.rs:1626–1688

()

    // Verify we don't return rows on DML
    #[test]
    fn test_row_dml() -> ResultTest<()> {
        let db = TestDB::durable()?;

        let table_id = db.create_table_for_test("T", &[("a", AlgebraicType::U8)], &[])?;
        with_auto_commit(&db, |tx| -> Result<_, DBError> {
            for i in 0..4u8 {
                insert(&db, tx, table_id, &product!(i))?;
            }
            Ok(())
        })?;

        let rt = db.runtime().expect("runtime should be there");

        let server = Identity::from_claims("issuer", "server");

        let internal_auth = AuthCtx::new(server, server);

        let tmp_vec = Vec::new();
        let run = |db, sql: &str, auth, subs, mut tmp_vec| {
            let sql = sql.to_string();
            async move { run(db, sql, auth, subs, None, &mut tmp_vec).await }
        };

        let check = |db, sql, auth, metrics: ExecutionMetrics| {
            let result = rt.block_on(run(db, sql, auth, None, tmp_vec.clone()))?;
            assert_eq!(result.rows, vec![]);
            assert_eq!(result.metrics.rows_inserted, metrics.rows_inserted);
            assert_eq!(result.metrics.rows_deleted, metrics.rows_deleted);
            assert_eq!(result.metrics.rows_updated, metrics.rows_updated);

            Ok::<(), DBError>(())
        };

        let ins = ExecutionMetrics {
            rows_inserted: 1,
            ..ExecutionMetrics::default()
        };
        let upd = ExecutionMetrics {
            rows_updated: 5,
            ..ExecutionMetrics::default()
        };
        let del = ExecutionMetrics {
            rows_deleted: 1,
            ..ExecutionMetrics::default()
        };

        check(db.clone(), "INSERT INTO T (a) VALUES (5)", internal_auth.clone(), ins)?;
        check(db.clone(), "UPDATE T SET a = 2", internal_auth.clone(), upd)?;
        assert_eq!(
            rt.block_on(run(
                db.clone(),
                "SELECT * FROM T",
                internal_auth.clone(),
                None,
                tmp_vec.clone()
            ))?
            .rows,
            vec![product!(2u8)]