Add ability to replace existing tokens easily

terminal42 · Oct 14, 2024 · 5c5d552 · 5c5d552
1 parent 2113e27
commit 5c5d552
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 0 deletions.
diff --git a/src/Token/TokenCollection.php b/src/Token/TokenCollection.php
@@ -29,6 +29,19 @@ public static function fromSerializedArray(array $data): self
         return new self($tokens);
     }
 
+    public function replaceToken(Token $token): self
+    {
+        $existing = $this->getByName($token->getName());
+
+        if (null !== $existing) {
+            $this->remove($existing);
+        }
+
+        $this->addToken($token);
+
+        return $this;
+    }
+
     /**
      * Provides a fluent interface alternative to add() with a type hint.
      */

diff --git a/tests/Token/TokenCollectionTest.php b/tests/Token/TokenCollectionTest.php
@@ -61,6 +61,23 @@ public function testCollectionHandling(): void
         $this->assertNull($tokenCollection->getByName('form_i_do_not_exist'));
     }
 
+    public function testReplace(): void
+    {
+        $tokenCollection = new TokenCollection();
+        $tokenCollection->addToken(Token::fromValue('test', 'foobar'));
+        $tokenCollection->addToken(Token::fromValue('test', 'foobar new'));
+
+        $this->assertSame(2, $tokenCollection->count());
+        $this->assertSame('foobar', $tokenCollection->getByName('test')->getValue(), 'foobar');
+
+        $tokenCollection = new TokenCollection();
+        $tokenCollection->replaceToken(Token::fromValue('test', 'foobar'));
+        $tokenCollection->replaceToken(Token::fromValue('test', 'foobar new'));
+
+        $this->assertSame(1, $tokenCollection->count());
+        $this->assertSame('foobar new', $tokenCollection->getByName('test')->getValue(), 'foobar');
+    }
+
     public function testMerge(): void
     {
         $tokenCollectionA = new TokenCollection();