add simple config test

vectordotdev · Dec 7, 2023 · f8cc865 · f8cc865
1 parent 7eac7f0
commit f8cc865
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/lib/codecs/src/encoding/format/protobuf.rs b/lib/codecs/src/encoding/format/protobuf.rs
@@ -234,12 +234,27 @@ mod tests {
         };
     }
 
+    fn test_data_dir() -> PathBuf {
+        PathBuf::from(std::env::var_os("CARGO_MANIFEST_DIR").unwrap()).join("tests/data/protobuf")
+    }
+
     fn test_message_descriptor(message_type: &str) -> MessageDescriptor {
         let path = PathBuf::from(std::env::var_os("CARGO_MANIFEST_DIR").unwrap())
             .join("tests/data/protobuf/protos/test.desc");
         get_message_descriptor(&path, &format!("test.{message_type}")).unwrap()
     }
 
+    #[test]
+    fn test_config_input_type() {
+        let config = ProtobufSerializerConfig {
+            protobuf: ProtobufSerializerOptions {
+                desc_file: test_data_dir().join("test_protobuf.desc"),
+                message_type: "test_protobuf.Person".into(),
+            },
+        };
+        assert_eq!(config.input_type(), DataType::all());
+    }
+
     #[test]
     fn test_encode_integers() {
         let message = encode_message(