@@ -370,7 +370,13 @@ public void testExtractMetadataStructuredWithTemplateSuccess() {
370370 );
371371
372372 assertThat (response .getSourceJson (), equalTo (Json .parse (result ).asObject ()));
373- assertThat (response .getSourceJson ().get ("firstName" ).asString (), equalTo ("John" ));
373+ assertThat (response .getAnswer ().get ("firstName" ).asString (), equalTo ("John" ));
374+ assertThat (response .getAnswer ().get ("lastName" ).asString (), equalTo ("Doe" ));
375+ assertThat (response .getAnswer ().get ("age" ).asInt (), equalTo (25 ));
376+ assertThat (response .getAnswer ().get ("hobbies" ).asArray ().get (0 ).asString (), equalTo ("reading" ));
377+
378+ assertThat (response .getCreatedAt (), equalTo (new Date (1355338423123L )));
379+ assertThat (response .getCompletionReason (), equalTo ("done" ));
374380 }
375381
376382 @ Test
@@ -414,6 +420,12 @@ public void testExtractMetadataStructuredWithFieldSuccess() {
414420 );
415421
416422 assertThat (response .getSourceJson (), equalTo (Json .parse (result ).asObject ()));
417- assertThat (response .getSourceJson ().get ("firstName" ).asString (), equalTo ("John" ));
423+ assertThat (response .getAnswer ().get ("firstName" ).asString (), equalTo ("John" ));
424+ assertThat (response .getAnswer ().get ("lastName" ).asString (), equalTo ("Doe" ));
425+ assertThat (response .getAnswer ().get ("age" ).asInt (), equalTo (25 ));
426+ assertThat (response .getAnswer ().get ("hobbies" ).asArray ().get (0 ).asString (), equalTo ("reading" ));
427+
428+ assertThat (response .getCreatedAt (), equalTo (new Date (1355338423123L )));
429+ assertThat (response .getCompletionReason (), equalTo ("done" ));
418430 }
419431}
0 commit comments