change add to set

sokil · Mar 24, 2016 · 70f9129 · 70f9129 · Psiiirus · Mar 24, 2016
1 parent 3308f93
commit 70f9129
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 4 deletions.
diff --git a/src/Sokil/Vast/Ad/Wrapper.php b/src/Sokil/Vast/Ad/Wrapper.php
@@ -11,11 +11,16 @@ class Wrapper extends \Sokil\Vast\Ad
      * @param $uri
      * @return $this
      */
-    public function addVASTAdTagURI($uri)
+    public function setVASTAdTagURI($uri)
     {
+        // get VASTAdTagURI dom node
+        $VASTAdTagURIDomElement = $this->domElement->getElementsByTagName('VASTAdTagURI')->item(0);
+
         // create VASTAdTagURI-Node
-        $VASTAdTagURIDomElement = $this->domElement->ownerDocument->createElement('VASTAdTagURI');
-        $this->domElement->firstChild->appendChild($VASTAdTagURIDomElement);
+        if(!$VASTAdTagURIDomElement) {
+            $VASTAdTagURIDomElement = $this->domElement->ownerDocument->createElement('VASTAdTagURI');
+            $this->domElement->firstChild->appendChild($VASTAdTagURIDomElement);
+        }
 
         // create VASTAdTagURI-cdata
         $cdata = $this->domElement->ownerDocument->createCDATASection($uri);

diff --git a/tests/Sokil/Vast/DocumentTest.php b/tests/Sokil/Vast/DocumentTest.php
@@ -47,7 +47,7 @@ public function testCreateWrapperAdSection()
         $ad1 = $document->createWrapperAdSection()
             ->setId('ad1')
             ->setAdSystem('Ad Server Name')
-            ->addVASTAdTagURI('http://entertainmentserver.com/vart.xml');
+            ->setVASTAdTagURI('http://entertainmentserver.com/vart.xml');
 
         $actualXml = str_replace(array("\r", "\n"), '', $document->toString());