more progress on the strategy, unit testing

1 files changed, 30 insertions, 0 deletions

diff --git a/src/main/java/design/model/Play.java b/src/main/java/design/model/Play.java
new file mode 100644
index 0000000..b61228b
--- /dev/null
+++ b/src/main/java/design/model/Play.java
@@ -0,0 +1,30 @@
+package design.model;
+
+import java.util.ArrayList;
+import java.util.List;
+
+public class Play {
+    private final int distance;
+    private final List<Swing> swings;
+
+    public Play(int distance) {
+        this.swings = new ArrayList<>();
+        this.distance = distance;
+    }
+
+    public int getDistance() {
+        return distance;
+    }
+
+    public void addSwing(Swing swing) {
+        swings.add(swing);
+    }
+
+    public Swing[] getSwings() {
+        return swings.toArray(Swing[]::new);
+    }
+
+    public int getSwingCount() {
+        return swings.size();
+    }
+}