Added method to return all message ids after a BatchMessage operation.

mailgun · Oct 12, 2013 · 5d7daee · 5d7daee
1 parent 7323f28
commit 5d7daee
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git a/src/Mailgun/Messages/BatchMessage.php b/src/Mailgun/Messages/BatchMessage.php
@@ -84,12 +84,15 @@ public function sendMessage($message = array(), $files = array()){
 			$this->counters['recipients']['bcc'] = 0;
 			unset($this->message["to"]);
 			array_push($this->messageIds, $response->http_response_body->id);
-			return $this->messageIds;
 		}
 	}
 
 	public function finalize(){
 		return $this->sendMessage();
 	}
+
+	public function getMessageIds(){
+		return $this->messageIds;
+	}
 }
 ?>
diff --git a/tests/Mailgun/Tests/Messages/BatchMessageTest.php b/tests/Mailgun/Tests/Messages/BatchMessageTest.php
@@ -105,6 +105,16 @@ public function testDefaultIDInVariables() {
         $propertyValue = $property->getValue($message);
         $this->assertEquals(1, $propertyValue['test-user@samples.mailgun.org']['id']);
     }
+    public function testgetMessageIds() {
+        $message = $this->client->BatchMessage($this->sampleDomain);
+        $message->addToRecipient("test-user@samples.mailgun.org", array("first" => "Test", "last" => "User"));
+        $message->setFromAddress("samples@mailgun.org", array("first" => "Test", "last" => "User"));
+		$message->setSubject("This is the subject of the message!");
+		$message->setTextBody("This is the text body of the message!");
+        $message->finalize();
+
+        $this->assertEquals(array("1234"), $message->getMessageIds());
+    }
 }
 ?>