1818use Patchlevel \EventSourcing \Store \Criteria \FromPlayheadCriterion ;
1919use Patchlevel \EventSourcing \Store \Criteria \StreamCriterion ;
2020use Patchlevel \EventSourcing \Store \Criteria \ToIndexCriterion ;
21+ use Patchlevel \EventSourcing \Store \Header \IndexHeader ;
2122use Patchlevel \EventSourcing \Store \Header \PlayheadHeader ;
2223use Patchlevel \EventSourcing \Store \Header \StreamNameHeader ;
2324use Patchlevel \EventSourcing \Store \InMemoryStore ;
@@ -166,7 +167,20 @@ public function testLoadFromIndex(): void
166167
167168 $ messages = iterator_to_array ($ stream );
168169
169- self ::assertSame ([$ message3 , $ message4 ], $ messages );
170+ self ::assertCount (2 , $ messages );
171+ self ::assertSame (
172+ $ message3 ->header (PlayheadHeader::class)->playhead ,
173+ $ messages [0 ]->header (PlayheadHeader::class)->playhead ,
174+ );
175+ self ::assertSame (
176+ 3 ,
177+ $ messages [0 ]->header (IndexHeader::class)->index ,
178+ );
179+ self ::assertFalse ($ message4 ->hasHeader (PlayheadHeader::class));
180+ self ::assertSame (
181+ 4 ,
182+ $ messages [1 ]->header (IndexHeader::class)->index ,
183+ );
170184 }
171185
172186 public function testLoadToIndex (): void
@@ -186,7 +200,23 @@ public function testLoadToIndex(): void
186200
187201 $ messages = iterator_to_array ($ stream );
188202
189- self ::assertSame ([$ message1 , $ message2 ], $ messages );
203+ self ::assertCount (2 , $ messages );
204+ self ::assertSame (
205+ $ message1 ->header (AggregateHeader::class)->playhead ,
206+ $ messages [0 ]->header (AggregateHeader::class)->playhead ,
207+ );
208+ self ::assertSame (
209+ 1 ,
210+ $ messages [0 ]->header (IndexHeader::class)->index ,
211+ );
212+ self ::assertSame (
213+ $ message2 ->header (AggregateHeader::class)->playhead ,
214+ $ messages [1 ]->header (AggregateHeader::class)->playhead ,
215+ );
216+ self ::assertSame (
217+ 2 ,
218+ $ messages [1 ]->header (IndexHeader::class)->index ,
219+ );
190220 }
191221
192222 public function testLoadByStreamNameWithLikeAll (): void
0 commit comments