public function testGetFlags() { $flag = $this->getMockBuilder('\\StateMachine\\Flag')->disableOriginalConstructor()->getMock(); $flag->expects($this->any())->method('getName')->willReturn('flagName'); $subject = $this->getMock('\\stdClass'); $payload = new Payload('id', $subject); $payload->setFlag($flag); $this->assertEquals($flag, $payload->getFlag('flagName')); }