BAEL-3987 (eugenp#9139)

vatsalgosar · web-flow · commit d3f060f79086 · 2020-05-28T17:37:23.000+01:00
* BAEL-3987 - Cast an int to an enum value

* refactoring

* code refactored

* BAEL-3987 - Updated code snippets

* bAEL-3987 - Fixed indentation
diff --git a/core-java-modules/core-java-lang-2/src/main/java/com/baeldung/inttoenum/PizzaStatus.java b/core-java-modules/core-java-lang-2/src/main/java/com/baeldung/inttoenum/PizzaStatus.java
@@ -0,0 +1,36 @@
+package com.baeldung.inttoenum;
+
+import java.util.HashMap;
+import java.util.Map;
+
+public enum PizzaStatus {
+    ORDERED(5),
+    READY(2),
+    DELIVERED(0);
+
+    private int timeToDelivery;
+
+    PizzaStatus(int timeToDelivery) {
+        this.timeToDelivery = timeToDelivery;
+    }
+
+    public int getTimeToDelivery() {
+        return timeToDelivery;
+    }
+
+    private static Map<Integer, PizzaStatus> timeToDeliveryToEnumValuesMapping = new HashMap<>();
+
+    static {
+        PizzaStatus[] pizzaStatuses = PizzaStatus.values();
+        for (int pizzaStatusIndex = 0; pizzaStatusIndex < pizzaStatuses.length; pizzaStatusIndex++) {
+            timeToDeliveryToEnumValuesMapping.put(
+                pizzaStatuses[pizzaStatusIndex].getTimeToDelivery(),
+                pizzaStatuses[pizzaStatusIndex]
+            );
+        }
+    }
+
+    public static PizzaStatus castIntToEnum(int timeToDelivery) {
+        return timeToDeliveryToEnumValuesMapping.get(timeToDelivery);
+    }
+}
diff --git a/core-java-modules/core-java-lang-2/src/test/java/com/baeldung/inttoenum/IntToEnumUnitTest.java b/core-java-modules/core-java-lang-2/src/test/java/com/baeldung/inttoenum/IntToEnumUnitTest.java
@@ -0,0 +1,27 @@
+package com.baeldung.inttoenum;
+
+import org.junit.Test;
+
+import static org.junit.Assert.assertEquals;
+
+public class IntToEnumUnitTest {
+
+    @Test
+    public void whenIntToEnumUsingValuesMethod_thenReturnEnumObject() {
+        int timeToDeliveryForOrderedPizzaStatus = 5;
+        PizzaStatus[] pizzaStatuses = PizzaStatus.values();
+        PizzaStatus pizzaOrderedStatus = null;
+        for (int pizzaStatusIndex = 0; pizzaStatusIndex < pizzaStatuses.length; pizzaStatusIndex++) {
+            if (pizzaStatuses[pizzaStatusIndex].getTimeToDelivery() == timeToDeliveryForOrderedPizzaStatus) {
+                pizzaOrderedStatus = pizzaStatuses[pizzaStatusIndex];
+            }
+        }
+        assertEquals(pizzaOrderedStatus, PizzaStatus.ORDERED);
+    }
+
+    @Test
+    public void whenIntToEnumUsingMap_thenReturnEnumObject() {
+        int timeToDeliveryForOrderedPizzaStatus = 5;
+        assertEquals(PizzaStatus.castIntToEnum(timeToDeliveryForOrderedPizzaStatus), PizzaStatus.ORDERED);
+    }
+}
