WIP

it-bens · Sep 29, 2021 · 5df5637 · 5df5637
1 parent ac92819
commit 5df5637
Showing 1 changed file with 12 additions and 4 deletions.
diff --git a/tests/TransformationEnvelopeTest.php b/tests/TransformationEnvelopeTest.php
@@ -74,12 +74,20 @@ public function testWrapNotOverrideStamp(): void
     public function testWrapOverrideStamp(): void
     {
         $input = new Object1('I\'m Mr. Meeseeks, look at me!');
-        $stamp1 = new InputClassStamp(Object1::class, 0);
-        $stamp2 = new InputClassStamp(Object3::class, 1);
-        $envelope = TransformationEnvelope::wrap($input, [$stamp1, $stamp2]);
+        $inputClassStamp1 = new InputClassStamp(Object1::class, 0);
+        $inputClassStamp2 = new InputClassStamp(Object3::class, 1);
+        $additionalDataStamp1 = new AdditionalDataStamp(['test' => 1], 0);
+        $additionalDataStamp2 = new AdditionalDataStamp(['test' => 2], 1);
+        $envelope = TransformationEnvelope::wrap($input, [
+            $inputClassStamp1,
+            $inputClassStamp2,
+            $additionalDataStamp1,
+            $additionalDataStamp2
+        ]);
 
-        $this->assertCount(1, $envelope->getStamps());
+        $this->assertCount(2, $envelope->getStamps());
         $this->assertEquals(Object3::class, $envelope->getStamp(InputClassStamp::class)->getInputClassName());
+        $this->assertEquals(2, $envelope->getStamp(AdditionalDataStamp::class)->getAdditionalData()['test']);
     }
 
     public function testWrapWithDifferentStamps(): void