instanceofnew

core-java-modules/core-java-16/src/main/java/com/baeldung/instanceofalternative/enumallt/Anatotitan.java

@@ -1,15 +0,0 @@
-package com.baeldung.instanceofalternative.enumallt;
-
-public class Anatotitan extends Dinosaur {
-    // polymorphism
-    @Override
-    public String move() {
-        return "running";
-    }
-
-    // non-polymorphism
-    public String run() {
-        return "running";
-    }
-
-}

core-java-modules/core-java-16/src/main/java/com/baeldung/instanceofalternative/enumallt/Dinosaur.java

@@ -1,15 +0,0 @@
-package com.baeldung.instanceofalternative.enumallt;
-
-public class Dinosaur {
-
-    DinosaurEnum dinosaur;
-
-    public DinosaurEnum getDinosaur() {
-        return dinosaur;
-    }
-
-    public void setDinosaur(DinosaurEnum dinosaur) {
-        this.dinosaur = dinosaur;
-    }
-
-}

core-java-modules/core-java-16/src/main/java/com/baeldung/instanceofalternative/enumallt/Euraptor.java

@@ -1,15 +0,0 @@
-package com.baeldung.instanceofalternative.enumallt;
-
-public class Euraptor extends Dinosaur {
-    // polymorphism
-    @Override
-    public String move() {
-        return "flying";
-    }
-
-    // non-polymorphism
-    public String flies() {
-        return "flying";
-    }
-
-}

core-java-modules/core-java-16/src/test/java/com/baeldung/instanceoftest/EnumUnitTest.java

@@ -10,20 +10,17 @@ public class EnumUnitTest {
     @Test
     public void givenADinosaurSpecie_whenGroupBySubclass_thenGetMovementOfEuraptor() {
 
-        Dinosaur dinosaur = new Dinosaur();
-        dinosaur.setDinosaur(DinosaurEnum.Euraptor);
-
-        assertEquals("flying", dinosaur.getDinosaur()
-            .move());
+        assertEquals("running", moveDinosaur(DinosaurEnum.Euraptor));
     }
 
     @Test
     public void givenADinosaurSpecie_whenGroupBySubclass_thenGetMovementOfAnatotitan() {
-        Dinosaur dinosaur = new Dinosaur();
-        dinosaur.setDinosaur(DinosaurEnum.Anatotitan);
+        assertEquals("running", moveDinosaur(DinosaurEnum.Anatotitan));
+    }
+
+    public static String moveDinosaur(DinosaurEnum dinosaurenum) {
+        return dinosaurenum.move();
 
-        assertEquals("running", dinosaur.getDinosaur()
-            .move());
     }
 
 }