22namespace TraderInteractiveTest \Mongo ;
33
44use PHPUnit \Framework \TestCase ;
5+ use TraderInteractive \Mongo \QueueAwareTrait ;
6+ use TraderInteractive \Mongo \QueueInterface ;
57
68class QueueAwareTraitTest extends TestCase
79{
@@ -15,11 +17,10 @@ class QueueAwareTraitTest extends TestCase
1517 */
1618 public function setQueue ()
1719 {
18- $ mockQueue = $ this ->getObjectForTrait ('\TraderInteractive\Mongo\QueueAwareTrait ' );
19- $ this ->assertAttributeEquals (null , 'mongoQueue ' , $ mockQueue );
20- $ mongoQueue = $ this ->getMockBuilder ('\\TraderInteractive \\Mongo \\QueueInterface ' )->getMock ();
20+ $ mockQueue = $ this ->getObjectForTrait (QueueAwareTrait::class);
21+ $ mongoQueue = $ this ->getMockBuilder (QueueInterface::class)->getMock ();
2122 $ mockQueue ->setQueue ($ mongoQueue );
22- $ this ->assertAttributeEquals ($ mongoQueue , ' mongoQueue ' , $ mockQueue );
23+ $ this ->assertSame ($ mongoQueue , $ mockQueue-> getQueue () );
2324 }
2425
2526 /**
@@ -32,8 +33,8 @@ public function setQueue()
3233 */
3334 public function getQueue ()
3435 {
35- $ mockQueue = $ this ->getObjectForTrait (' \TraderInteractive\Mongo\ QueueAwareTrait' );
36- $ mongoQueue = $ this ->getMockBuilder ('\\ TraderInteractive \\ Mongo \\ QueueInterface ' )->getMock ();
36+ $ mockQueue = $ this ->getObjectForTrait (QueueAwareTrait::class );
37+ $ mongoQueue = $ this ->getMockBuilder (QueueInterface::class )->getMock ();
3738 $ mockQueue ->setQueue ($ mongoQueue );
3839 $ this ->assertEquals ($ mongoQueue , $ mockQueue ->getQueue ());
3940 }
0 commit comments