test: fix expectation

Author: Devdutt Shenoi

src/event/format/json.rs

@@ -665,7 +665,7 @@ mod tests {
             .to_data(None, None, None, SchemaVersion::V0)
             .unwrap();
         let (schema, _) =
-            Event::infer_schema(&data[0], &store_schema, None, false, SchemaVersion::V0).unwrap();
+            Event::infer_schema(&data[1], &store_schema, None, false, SchemaVersion::V0).unwrap();
         let rb =
             Event::into_recordbatch(Utc::now(), &data, &schema, None, SchemaVersion::V0).unwrap();
 
@@ -718,7 +718,7 @@ mod tests {
             .to_data(None, None, None, SchemaVersion::V0)
             .unwrap();
         let (schema, _) =
-            Event::infer_schema(&data[0], &store_schema, None, false, SchemaVersion::V0).unwrap();
+            Event::infer_schema(&data[1], &store_schema, None, false, SchemaVersion::V0).unwrap();
         let rb =
             Event::into_recordbatch(Utc::now(), &data, &schema, None, SchemaVersion::V0).unwrap();
 
@@ -792,23 +792,12 @@ mod tests {
 
     #[test]
     fn arr_schema_mismatch() {
-        let json = json!([
-            {
-                "a": null,
-                "b": "hello",
-                "c": 1.24
-            },
-            {
-                "a": 1,
-                "b": "hello",
-                "c": 1
-            },
-            {
-                "a": 1,
-                "b": "hello",
-                "c": null
-            },
-        ]);
+        let json = json!(
+        {
+            "a": 1,
+            "b": "hello",
+            "c": 1
+        });
 
         let store_schema = fields_to_map(
             [
@@ -819,13 +808,14 @@ mod tests {
             .into_iter(),
         );
 
-        let data = Event::new(json, 0 /* doesn't matter */, LogSource::Json)
-            .to_data(None, None, None, SchemaVersion::V0)
-            .unwrap();
-
-        assert!(
-            Event::infer_schema(&data[0], &store_schema, None, false, SchemaVersion::V0).is_err()
-        );
+        assert!(Event::infer_schema(
+            json.as_object().unwrap(),
+            &store_schema,
+            None,
+            false,
+            SchemaVersion::V0
+        )
+        .is_err());
     }
 
     #[test]
@@ -857,7 +847,7 @@ mod tests {
             .to_data(None, None, None, SchemaVersion::V0)
             .unwrap();
         let (schema, _) =
-            Event::infer_schema(&data[0], &store_schema, None, false, SchemaVersion::V0).unwrap();
+            Event::infer_schema(&data[3], &store_schema, None, false, SchemaVersion::V0).unwrap();
         let rb =
             Event::into_recordbatch(Utc::now(), &data, &schema, None, SchemaVersion::V0).unwrap();
 
@@ -935,7 +925,7 @@ mod tests {
             .to_data(None, None, None, SchemaVersion::V1)
             .unwrap();
         let (schema, _) =
-            Event::infer_schema(&data[0], &store_schema, None, false, SchemaVersion::V1).unwrap();
+            Event::infer_schema(&data[3], &store_schema, None, false, SchemaVersion::V1).unwrap();
         let rb =
             Event::into_recordbatch(Utc::now(), &data, &schema, None, SchemaVersion::V1).unwrap();