Function test_row_limit

crates/core/src/sql/execute.rs:1517–1622

()

    #[test]
    fn test_row_limit() -> 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..5u8 {
                insert(&db, tx, table_id, &product!(i))?;
            }
            Ok(())
        })?;

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

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

        let tmp_vec = Vec::new();

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

        let run = |db, sql, auth, subs, mut tmp_vec| rt.block_on(run(db, sql, auth, subs, None, &mut tmp_vec));
        // No row limit, both queries pass.
        assert!(run(
            db.clone(),
            "SELECT * FROM T".to_string(),
            internal_auth.clone(),
            None,
            tmp_vec.clone()
        )
        .is_ok());
        assert!(run(
            db.clone(),
            "SELECT * FROM T".to_string(),
            external_auth.clone(),
            None,
            tmp_vec.clone()
        )
        .is_ok());

        // Set row limit.
        assert!(run(
            db.clone(),
            "SET row_limit = 4".to_string(),
            internal_auth.clone(),
            None,
            tmp_vec.clone()
        )
        .is_ok());

        // External query fails.
        assert!(run(
            db.clone(),
            "SELECT * FROM T".to_string(),
            internal_auth.clone(),
            None,
            tmp_vec.clone()
        )