Sonata\Tests\ProductBundle\Entity\BaseProductTest::testHasOneMainCategory PHP Method

    public function testHasOneMainCategory()
    {
        $product = new Product();
        $pc = $this->getMock('Sonata\\Component\\Product\\ProductCategoryInterface');
        $pc->expects($this->any())->method('getMain')->will($this->returnValue(true));
        $pc2 = $this->getMock('Sonata\\Component\\Product\\ProductCategoryInterface');
        $pc2->expects($this->any())->method('getMain')->will($this->returnValue(true));
        $pc3 = $this->getMock('Sonata\\Component\\Product\\ProductCategoryInterface');
        $pc3->expects($this->any())->method('getMain')->will($this->returnValue(true));
        $this->assertFalse($product->hasOneMainCategory());
        $product->addProductCategory($pc);
        $this->assertTrue($product->hasOneMainCategory());
        $product->addProductCategory($pc2);
        $this->assertFalse($product->hasOneMainCategory());
        $product->addProductCategory($pc3);
        $this->assertFalse($product->hasOneMainCategory());
    }