fix unit test

risingwavelabs · Sep 10, 2024 · a888d33 · a888d33
1 parent 4ad79ce
commit a888d33
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/src/connector/src/parser/protobuf/parser.rs b/src/connector/src/parser/protobuf/parser.rs
@@ -708,7 +708,8 @@ mod test {
     }
 
     fn pb_eq(a: &ProtobufAccess, field_name: &str, value: ScalarImpl) {
-        let dummy_type = DataType::Varchar;
+        let field = a.descriptor().get_field_by_name(field_name).unwrap();
+        let dummy_type = protobuf_type_mapping(&field, &mut vec![]).unwrap();
         let d = a.access_owned(&[field_name], &dummy_type).unwrap().unwrap();
         assert_eq!(d, value, "field: {} value: {:?}", field_name, d);
     }

diff --git a/src/connector/src/parser/unified/protobuf.rs b/src/connector/src/parser/unified/protobuf.rs
@@ -32,6 +32,11 @@ impl ProtobufAccess {
     pub fn new(message: DynamicMessage) -> Self {
         Self { message }
     }
+
+    #[cfg(test)]
+    pub fn descriptor(&self) -> prost_reflect::MessageDescriptor {
+        self.message.descriptor()
+    }
 }
 
 impl Access for ProtobufAccess {