Implement RandValue trait, import SerialArray and use in tests.

risingwavelabs · Mar 6, 2023 · f37ef4c · f37ef4c
1 parent 0eb8c16
commit f37ef4c
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/common/src/test_utils/rand_array.rs b/src/common/src/test_utils/rand_array.rs
@@ -25,6 +25,7 @@ use rand::prelude::Distribution;
 use rand::rngs::SmallRng;
 use rand::{Rng, SeedableRng};
 
+use crate::array::serial_array::Serial;
 use crate::array::{Array, ArrayBuilder, ArrayRef, JsonbVal, ListValue, StructValue};
 use crate::types::{
     Decimal, IntervalUnit, NaiveDateTimeWrapper, NaiveDateWrapper, NaiveTimeWrapper, NativeType,
@@ -117,6 +118,12 @@ impl RandValue for bool {
     }
 }
 
+impl RandValue for Serial {
+    fn rand_value<R: Rng>(rand: &mut R) -> Self {
+        i64::rand_value(rand).into()
+    }
+}
+
 impl RandValue for JsonbVal {
     fn rand_value<R: rand::Rng>(_rand: &mut R) -> Self {
         JsonbVal::dummy()
@@ -177,6 +184,7 @@ where
 mod tests {
     use super::*;
     use crate::array::interval_array::IntervalArray;
+    use crate::array::serial_array::SerialArray;
     use crate::array::*;
     use crate::for_all_variants;