[phpunit] Set flag for messages from another mailbox

ddeboer · Oct 5, 2017 · 189815a · 189815a
1 parent 8d1a37d
commit 189815a
Showing 1 changed file with 17 additions and 7 deletions.
diff --git a/tests/MailboxTest.php b/tests/MailboxTest.php
@@ -111,27 +111,37 @@ public function testCustomStatusFlag()
 
     public function testBulkSetFlags()
     {
-        $messages = [1 => 'Message A', 2 => 'Message B', 3 => 'Message C'];
+        // prepare second mailbox with 3 messages
+        $anotherMailbox = $this->createMailbox();
+        $this->createTestMessage($anotherMailbox, 'Message 1');
+        $this->createTestMessage($anotherMailbox, 'Message 2');
+        $this->createTestMessage($anotherMailbox, 'Message 3');
 
-        foreach ($messages as $uid => $subject) {
-            $this->createTestMessage($this->mailbox, $subject);
+        // Message UIDs created in setUp method
+        $messages = [1, 2, 3];
 
+        foreach ($messages as $uid) {
             $message = $this->mailbox->getMessage($uid);
             $this->assertFalse($message->isFlagged());
         }
 
-        $this->mailbox->setFlag('\\Flagged', \array_keys($messages));
+        $this->mailbox->setFlag('\\Flagged', $messages);
 
-        foreach ($messages as $uid => $subject) {
+        foreach ($messages as $uid) {
             $message = $this->mailbox->getMessage($uid);
             $this->assertTrue($message->isFlagged());
         }
 
-        $this->mailbox->clearFlag('\\Flagged', \array_keys($messages));
+        $this->mailbox->clearFlag('\\Flagged', $messages);
 
-        foreach ($messages as $uid => $subject) {
+        foreach ($messages as $uid) {
             $message = $this->mailbox->getMessage($uid);
             $this->assertFalse($message->isFlagged());
         }
+
+        // Set flag for messages from another mailbox
+        $anotherMailbox->setFlag('\\Flagged', [1, 2, 3]);
+
+        $this->assertTrue($anotherMailbox->getMessage(2)->isFlagged());
     }
 }