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));
}
