@@ -62,7 +62,7 @@ public async Task CanWriteChunksWithCustomDefinition()
6262 Assert . NotEqual ( Guid . Empty , record . Key ) ;
6363 Assert . Equal ( documentId , record . DocumentId ) ;
6464 Assert . NotEmpty ( record . SerializedContent ) ;
65- Assert . Equal ( "custom schema content" , ( ( TextContent ) record . Content ! ) . Text ) ;
65+ Assert . Equal ( ( ( TextContent ) chunks [ 0 ] . Content ) . Text , ( ( TextContent ) record . Content ! ) . Text ) ;
6666 }
6767
6868 [ Fact ]
@@ -94,7 +94,7 @@ public async Task CanWriteChunks()
9494 Assert . NotEqual ( Guid . Empty , record . Key ) ;
9595 Assert . Equal ( documentId , record . DocumentId ) ;
9696 Assert . NotEmpty ( record . SerializedContent ) ;
97- Assert . Equal ( "some content" , ( ( TextContent ) record . Content ! ) . Text ) ;
97+ Assert . Equal ( ( ( TextContent ) chunks [ 0 ] . Content ) . Text , ( ( TextContent ) record . Content ! ) . Text ) ;
9898 Assert . True ( testEmbeddingGenerator . WasCalled ) ;
9999 }
100100
@@ -125,7 +125,7 @@ public async Task CanWriteChunksWithMetadata()
125125 Assert . NotNull ( record ) ;
126126 Assert . Equal ( documentId , record . DocumentId ) ;
127127 Assert . NotEmpty ( record . SerializedContent ) ;
128- Assert . Equal ( "some content" , ( ( TextContent ) record . Content ! ) . Text ) ;
128+ Assert . Equal ( ( ( TextContent ) chunks [ 0 ] . Content ) . Text , ( ( TextContent ) record . Content ! ) . Text ) ;
129129 Assert . Equal ( "important" , record . Classification ) ;
130130 }
131131
0 commit comments