public function testGetDefinition() { $definition = 'test'; $segment = $this->getMock('Oro\\Bundle\\SegmentBundle\\Entity\\Segment'); $segment->expects($this->once())->method('getDefinition')->will($this->returnValue($definition)); $this->assertNull($this->entity->getDefinition()); $this->entity->setSegment($segment); $this->assertEquals($definition, $this->entity->getDefinition($definition)); }