Move tests to Prophecy

mtymek · Mar 21, 2017 · 4e0e449 · 4e0e449
1 parent 0bc89a9
commit 4e0e449
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/test/MtMailTest/ComposerPlugin/DefaultHeadersTest.php b/test/MtMailTest/ComposerPlugin/DefaultHeadersTest.php
@@ -74,22 +74,22 @@ public function testPluginCanInjectHeaderObjects()
  'subject' => $subject,
  ]);
 
- $headers = $this->getMock('Zend\Mail\Headers', ['addHeader']);
- $headers->expects($this->at(0))->method('addHeader')->with($subject);
+ $headers = $this->prophesize(Headers::class);
+ $headers->addHeader($subject);
  $message = new Message();
- $message->setHeaders($headers);
+ $message->setHeaders($headers->reveal());
  $event = new ComposerEvent();
  $event->setMessage($message);
  $this->plugin->injectDefaultHeaders($event);
  }
 
  public function testPluginCanInjectTemplateSpecificHeaderObjects()
  {
- $headers = $this->getMock('Zend\Mail\Headers', ['addHeader']);
- $headers->expects($this->at(0))->method('addHeader')->with((new Subject())->setSubject('Default subject'));
+ $headers = $this->prophesize(Headers::class);
+ $headers->addHeader((new Subject())->setSubject('Default subject'));
  $template = new HeaderObjectProviderTemplate();
  $message = new Message();
- $message->setHeaders($headers);
+ $message->setHeaders($headers->reveal());
  $event = new ComposerEvent();
  $event->setMessage($message);
  $event->setTemplate($template);