[jhlee] 좌표계산기 실습 및 피드백

src/main/java/coordinate/FigureFactory.java

@@ -1,9 +1,33 @@
 package coordinate;
 
+import java.util.HashMap;
 import java.util.List;
+import java.util.function.Function;
+
+public class FigureFactory implements FigureCreator{
+    static final HashMap<Integer, Function<List<Point>, Figure>> figures = new HashMap<>();
+
+    static {
+        figures.put(Line.LINE_POINT_SIZE, (Line::new));
+        figures.put(Triangle.TRIANGLE_POINT_SIZE, Triangle::new);
+        figures.put(Rectangle.RECTANGLE_POINT_SIZE, Rectangle::new);
+    }
+
+    @Override
+    public Figure create(List<Point> points) {
+        Function<List<Point>, Figure> figure = figures.get(points.size());
+
+        if (figure == null) {
+            throw new IllegalArgumentException("유효하지 않은 도형입니다.");
+        }
+
+        return figure.apply(points);
+
+    }
 
-public class FigureFactory {
     static Figure getInstance(List<Point> points) {
+
+
         if (points.size() == Line.LINE_POINT_SIZE) {
             return new Line(points);
         }

src/test/java/coordinate/FigureFactoryTest.java

@@ -6,8 +6,10 @@
 import java.util.List;
 
 import static org.assertj.core.api.Assertions.assertThat;
+import static org.assertj.core.api.Assertions.assertThatThrownBy;
 
 public class FigureFactoryTest {
+
     @Test
     public void line() {
         List<Point> points = Arrays.asList(
@@ -43,4 +45,56 @@ public void rectangle() {
         assertThat(figure).isInstanceOfAny(Rectangle.class);
         assertThat(figure.getName()).isEqualTo("사각형");
     }
+
+    @Test
+    public void lineCreate() {
+        List<Point> points = Arrays.asList(
+                Point.of(1, 2),
+                Point.of(2, 3));
+
+        Figure figure = new FigureFactory().create(points);
+        assertThat(figure).isInstanceOfAny(Line.class);
+        assertThat(figure.getName()).isEqualTo("선");
+    }
+
+    @Test
+    public void triangleCreate() {
+        List<Point> points = Arrays.asList(
+                Point.of(1, 1),
+                Point.of(4, 1),
+                Point.of(1, 4));
+
+        Figure figure = new FigureFactory().create(points);
+        assertThat(figure).isInstanceOfAny(Triangle.class);
+        assertThat(figure.getName()).isEqualTo("삼각형");
+    }
+
+    @Test
+    public void rectangleCreate() {
+        List<Point> points = Arrays.asList(
+                Point.of(1, 1),
+                Point.of(4, 1),
+                Point.of(1, 4),
+                Point.of(4, 4));
+
+        Figure figure = new FigureFactory().create(points);
+        assertThat(figure).isInstanceOfAny(Rectangle.class);
+        assertThat(figure.getName()).isEqualTo("사각형");
+    }
+
+    @Test
+    public void unkown() {
+        List<Point> points = Arrays.asList(
+                Point.of(1, 1),
+                Point.of(4, 1),
+                Point.of(1, 4),
+                Point.of(4, 4),
+                Point.of(5, 5));
+
+        assertThatThrownBy(() -> {
+            new FigureFactory().create(points);
+        })
+                .hasMessage("유효하지 않은 도형입니다.")
+                .isInstanceOf(IllegalArgumentException.class);
+    }
 }