Support getting list of all values for given field name

Author: clue

src/Protocol/Message.php

@@ -34,6 +34,32 @@ public function getFieldValue($key)
         return null;
     }
 
+    /**
+     * Returns a list of all values for the field with the given $key
+     *
+     * @param string $key
+     * @return array
+     */
+    public function getFieldValues($key)
+    {
+        $values = array();
+        $key = strtolower($key);
+
+        foreach ($this->fields as $part => $value) {
+            if (strtolower($part) === $key) {
+                if (is_array($value)) {
+                    foreach ($value as $v) {
+                        $values []= $v;
+                    }
+                } else {
+                    $values []= $value;
+                }
+            }
+        }
+
+        return $values;
+    }
+
     public function toJson()
     {
         return json_encode($this->getFields());

tests/Protocol/ActionTest.php

@@ -28,6 +28,12 @@ public function testOneFieldValue()
         $action = new Action(array('Action' => 'name'));
 
         $this->assertEquals("Action: name\r\n\r\n", $action->getMessageSerialized());
+
+        $this->assertEquals('name', $action->getFieldValue('Action'));
+        $this->assertEquals(array('name'), $action->getFieldValues('Action'));
+
+        $this->assertEquals(null, $action->getFieldValue('unknown'));
+        $this->assertEquals(array(), $action->getFieldValues('unknown'));
     }
 
     public function testMultipleFieldsSingleValue()
@@ -44,6 +50,7 @@ public function testOneFieldNoValue()
         $this->assertEquals("\r\n", $action->getMessageSerialized());
 
         $this->assertNull($action->getFieldValue('Key'));
+        $this->assertEquals(array(), $action->getFieldValues('Key'));
     }
 
     public function testOneFieldNoValues()
@@ -53,13 +60,17 @@ public function testOneFieldNoValues()
         $this->assertEquals("\r\n", $action->getMessageSerialized());
 
         $this->assertNull($action->getFieldValue('Key'));
+        $this->assertEquals(array(), $action->getFieldValues('Key'));
     }
 
     public function testOneFieldMultipleValues()
     {
         $action = new Action(array('Key' => array('Value1', 'Value2')));
 
         $this->assertEquals("Key: Value1\r\nKey: Value2\r\n\r\n", $action->getMessageSerialized());
+
+        $this->assertEquals('Value1', $action->getFieldValue('Key'));
+        $this->assertEquals(array('Value1', 'Value2'), $action->getFieldValues('Key'));
     }
 
     public function testOneFieldMultipleValuesIgnoreNulls()
@@ -69,6 +80,7 @@ public function testOneFieldMultipleValuesIgnoreNulls()
         $this->assertEquals("Key: value\r\n\r\n", $action->getMessageSerialized());
 
         $this->assertEquals('value', $action->getFieldValue('Key'));
+        $this->assertEquals(array('value'), $action->getFieldValues('Key'));
     }
 
     public function testOneFieldMultipleKeyValues()
@@ -78,5 +90,6 @@ public function testOneFieldMultipleKeyValues()
         $this->assertEquals("Variables: first=on\r\nVariables: second=off\r\n\r\n", $action->getMessageSerialized());
 
         $this->assertEquals('first=on', $action->getFieldValue('Variables'));
+        $this->assertEquals(array('first=on', 'second=off'), $action->getFieldValues('Variables'));
     }
 }