From 0679357fda626d902c877f93d627810364d43935 Mon Sep 17 00:00:00 2001
From: Nikita Sivukhin <sivukhin@turso.tech>
Date: Sun, 16 Feb 2025 12:33:26 +0400
Subject: [PATCH] insert numbers from different categories in order to cover
 cases of different serial type usage

---
 tests/integration/fuzz/grammar_generator.rs |  4 +--
 tests/integration/fuzz/mod.rs               | 37 ++++++++++++++++-----
 2 files changed, 31 insertions(+), 10 deletions(-)

diff --git a/tests/integration/fuzz/grammar_generator.rs b/tests/integration/fuzz/grammar_generator.rs
index e6f4b60bd..8222bdeb2 100644
--- a/tests/integration/fuzz/grammar_generator.rs
+++ b/tests/integration/fuzz/grammar_generator.rs
@@ -26,7 +26,7 @@ pub enum SymbolType {
         random_length: usize,
     },
     Int {
-        range: Range<i32>,
+        range: Range<i64>,
     },
     #[allow(dead_code)]
     Optional {
@@ -63,7 +63,7 @@ pub fn rand_str(fixed_prefix: &str, random_length: usize) -> SymbolType {
     }
 }
 
-pub fn rand_int(range: Range<i32>) -> SymbolType {
+pub fn rand_int(range: Range<i64>) -> SymbolType {
     SymbolType::Int { range }
 }
 
diff --git a/tests/integration/fuzz/mod.rs b/tests/integration/fuzz/mod.rs
index be6fe5e06..fdfcf6d0c 100644
--- a/tests/integration/fuzz/mod.rs
+++ b/tests/integration/fuzz/mod.rs
@@ -4,7 +4,7 @@ pub mod grammar_generator;
 mod tests {
     use std::rc::Rc;
 
-    use rand::{RngCore, SeedableRng};
+    use rand::SeedableRng;
     use rand_chacha::ChaCha8Rng;
     use rusqlite::params;
 
@@ -632,11 +632,18 @@ mod tests {
     pub fn table_logical_expression_fuzz_ex1() {
         let _ = env_logger::try_init();
 
-        for queries in [[
-            "CREATE TABLE t(x)",
-            "INSERT INTO t VALUES (10)",
-            "SELECT * FROM t WHERE  x = 1 AND 1 OR 0",
-        ]] {
+        for queries in [
+            [
+                "CREATE TABLE t(x)",
+                "INSERT INTO t VALUES (10)",
+                "SELECT * FROM t WHERE  x = 1 AND 1 OR 0",
+            ],
+            [
+                "CREATE TABLE t(x)",
+                "INSERT INTO t VALUES (-3258184727)",
+                "SELECT * FROM t",
+            ],
+        ] {
             let db = TempDatabase::new_empty();
             let limbo_conn = db.connect_limbo();
             let sqlite_conn = rusqlite::Connection::open_in_memory().unwrap();
@@ -663,6 +670,15 @@ mod tests {
         let (in_op, in_op_builder) = g.create_handle();
         let (paren, paren_builder) = g.create_handle();
 
+        let number = g
+            .create()
+            .choice()
+            .option_symbol(rand_int(-0xff..0x100))
+            .option_symbol(rand_int(-0xffff..0x10000))
+            .option_symbol(rand_int(-0xffffff..0x1000000))
+            .option_symbol(rand_int(-0xffffffff..0x100000000))
+            .option_symbol(rand_int(-0xffffffffffff..0x1000000000000))
+            .build();
         value_builder
             .choice()
             .option(
@@ -673,7 +689,7 @@ mod tests {
                     .push_str(")")
                     .build(),
             )
-            .option_symbol(rand_int(0..i32::MAX))
+            .option(number)
             .options_str(["x", "y", "z"])
             .option(
                 g.create()
@@ -742,8 +758,13 @@ mod tests {
         log::info!("seed: {}", seed);
 
         for _ in 0..100 {
-            let (x, y, z) = (rng.next_u32(), rng.next_u32(), rng.next_u32());
+            let (x, y, z) = (
+                g.generate(&mut rng, number, 1),
+                g.generate(&mut rng, number, 1),
+                g.generate(&mut rng, number, 1),
+            );
             let query = format!("INSERT INTO t VALUES ({}, {}, {})", x, y, z);
+            log::info!("insert: {}", query);
             assert_eq!(
                 limbo_exec_rows(&db, &limbo_conn, &query),
                 sqlite_exec_rows(&sqlite_conn, &query)