PolygonFactoryTest.java

package src.test.java.com.designpatterns.factorypattern;

import org.junit.Assert;
import org.junit.Test;
import src.main.java.com.designpatterns.factorypattern.Polygon;
import src.main.java.com.designpatterns.factorypattern.PolygonFactory;

public class PolygonFactoryTest {
    @Test
    public void testPolygonFactory() {
        String failReason = "";
        PolygonFactory polFactory = new PolygonFactory();

        //Test for triangle
        Polygon triangle = polFactory.getPolygon(3);
        if (!triangle.getType().equals("Triangle")) {
            failReason += "Polygon Factory failed for Triangle.";
        }
        if (triangle.area(4) != 6.928208) {
            failReason += "Triangle area is incorrect!";
        }


        //Test for square
        Polygon square = polFactory.getPolygon(4);
        if (!square.getType().equals("Square")) {
            failReason += "Polygon Factory failed for Square.";
        }
        if (square.area(5) != 25) {
            failReason += "Square area is incorrect!";
        }


        //Test for pentagon
        Polygon pentagon = polFactory.getPolygon(5);
        if (!pentagon.getType().equals("Pentagon")) {
            failReason += "Polygon Factory failed for Pentagon.";
        }
        if (pentagon.area(9) != 311.615424) {
            failReason += "Pentagon area is incorrect!";
        }

        Assert.assertEquals(failReason, failReason, "");
    }
}