fix: getElement() should always return SimpleXMLElement

danmichaelo · Jul 1, 2017 · a4f592e · a4f592e
1 parent 153d256
commit a4f592e
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/src/Danmichaelo/QuiteSimpleXMLElement/QuiteSimpleXMLElement.php b/src/Danmichaelo/QuiteSimpleXMLElement/QuiteSimpleXMLElement.php
@@ -94,11 +94,11 @@ protected function getElement($elem)
         }
 
         if (gettype($elem) == 'object') {
-            if (in_array(get_class($elem),
-                ['Danmichaelo\QuiteSimpleXMLElement\QuiteSimpleXMLElement', 'SimpleXMLElement'])) {
-                return $elem; // assume it's a SimpleXMLElement
-            } else {
-                throw new InvalidArgumentException('Unknown object given to QuiteSimpleXMLElement. Expected SimpleXMLElement or QuiteSimpleXMLElement.');
+            switch (get_class($elem)) {
+                case 'Danmichaelo\QuiteSimpleXMLElement\QuiteSimpleXMLElement':
+                    return $elem->el();
+                case 'SimpleXMLElement':
+                    return $elem;
             }
         }