Fix 4 code smells in FireworksTowerCombatTask (2 medium, 2 low)

UQcsse3200 · Oct 16, 2023 · ef9a838 · ef9a838
1 parent 6703cb4
commit ef9a838
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 4 deletions.
diff --git a/source/core/src/main/com/csse3200/game/components/tasks/FireworksTowerCombatTask.java b/source/core/src/main/com/csse3200/game/components/tasks/FireworksTowerCombatTask.java
@@ -45,7 +45,7 @@ public class FireworksTowerCombatTask extends DefaultTask implements PriorityTas
     public enum STATE {
         IDLE, ATTACK, DEATH
     }
-    public STATE towerState = STATE.IDLE;
+    private STATE towerState = STATE.IDLE;
 
     /**
      * @param priority Task priority when targets are detected (0 when nothing is present)
@@ -77,6 +77,7 @@ public void start() {
      * updates the current state of the tower based on the current state of the game. If enemies are detected, attack
      * state is activated and otherwise idle state remains.
      */
+    @Override
     public void update() {
         if (timeSource.getTime() >= endTime) {
             updateTowerState();
@@ -111,7 +112,7 @@ public void updateTowerState() {
                         Entity newProjectile = ProjectileFactory.createSplitFireWorksFireball(PhysicsLayer.NPC,
                                 new Vector2(100, owner.getEntity().getPosition().y), new Vector2(2f, 2f), 3);
                         newProjectile.setPosition((float) (owner.getEntity().getPosition().x + 0.25),
-                                (float) (owner.getEntity().getPosition().y));
+                                (owner.getEntity().getPosition().y));
                         ServiceLocator.getEntityService().register(newProjectile);
                     } else {
                         owner.getEntity().getEvents().trigger(IDLE);
@@ -139,6 +140,7 @@ public STATE getState() {
     /**
      * stops the current animation and switches back the state of the tower to IDLE.
      */
+    @Override
     public void stop() {
         super.stop();
         owner.getEntity().getEvents().trigger(IDLE);
@@ -159,4 +161,21 @@ public int getPriority() {
     public boolean isTargetVisible() {
         return physics.raycast(towerPosition, maxRangePosition, TARGET, hit);
     }
+
+    /**
+     * Function for getting the tower's state
+     *
+     * @return The state of this tower
+     */
+    public STATE getTowerState() {
+        return this.towerState;
+    }
+
+    /**
+     * Function for setting the tower's state
+     * @param newState The new state of this tower
+     */
+    public void setTowerState(STATE newState) {
+        this.towerState = newState;
+    }
 }
diff --git a/source/core/src/test/com/csse3200/game/components/tasks/FireworksTowerCombatTaskTest.java b/source/core/src/test/com/csse3200/game/components/tasks/FireworksTowerCombatTaskTest.java
@@ -64,7 +64,7 @@ public void testUpdateTowerStateWithTargetInRange() {
         entity.getEvents().addListener(FireworksTowerCombatTask.ATTACK, attack);
         //Jump to IDLE state
         fireworksTowerCombatTask.start();
-        fireworksTowerCombatTask.towerState = FireworksTowerCombatTask.STATE.IDLE;
+        fireworksTowerCombatTask.setTowerState(FireworksTowerCombatTask.STATE.IDLE);
 
         ServiceLocator.getPhysicsService().getPhysics().update();
         entity.update();
@@ -92,7 +92,7 @@ public void testUpdateTowerStateWithTargetNotInRange() {
         entity.getEvents().addListener(FireworksTowerCombatTask.IDLE, idle);
         entity.getEvents().addListener(FireworksTowerCombatTask.ATTACK, attack);
 
-        fireworksTowerCombatTask.towerState = FireworksTowerCombatTask.STATE.IDLE;
+        fireworksTowerCombatTask.setTowerState(FireworksTowerCombatTask.STATE.IDLE);
 
         ServiceLocator.getPhysicsService().getPhysics().update();
         entity.update();