Convert example 02-sum to use JUnit

1 files changed, 33 insertions, 0 deletions

diff --git a/examples/02-sum/output/java-junit/src/test/java/org/zwobble/example/ShapesTests.java b/examples/02-sum/output/java-junit/src/test/java/org/zwobble/example/ShapesTests.java
new file mode 100644
index 0000000..2554ca5
--- /dev/null
+++ b/examples/02-sum/output/java-junit/src/test/java/org/zwobble/example/ShapesTests.java
@@ -0,0 +1,33 @@
+package org.zwobble.example;
+
+import static org.junit.jupiter.api.Assertions.assertEquals;
+import org.junit.jupiter.api.Test;
+import org.zwobble.example.types.shapes.Shape;
+import org.zwobble.example.types.shapes.Rectangle;
+import org.zwobble.example.types.shapes.Triangle;
+
+public class ShapesTests {
+    @Test
+    public void rectangle() {
+        Shape rectangle = new Rectangle(10, 25);
+        int area = area(rectangle);
+        assertEquals(area, 250);
+    }
+
+    @Test
+    public void triangle() {
+        Shape triangle = new Triangle(10, 25);
+        int area = area(triangle);
+        assertEquals(area, 125);
+    }
+
+    private static int area(Shape shape) {
+        return switch (shape) {
+            case Rectangle(var width, var height) ->
+                width * height;
+
+            case Triangle(var width, var height) ->
+                width * height / 2;
+        };
+    }
+}