BAEL-1997 add additional method to the states

patterns/design-patterns/src/main/java/com/baeldung/state/DeliveredState.java

@@ -12,8 +12,14 @@ public class DeliveredState implements PackageState {
         pkg.setState(new OrderedState());
     }
 
+    @Override
+    public void printStatus() {
+        System.out.println("Package delivered to post office, not received yet.");
+    }
+
     @Override
     public String toString() {
         return "DeliveredState{}";
     }
+
 }

patterns/design-patterns/src/main/java/com/baeldung/state/OrderedState.java

@@ -12,6 +12,11 @@ public class OrderedState implements PackageState {
         System.out.println("The package is in it's root state.");
     }
 
+    @Override
+    public void printStatus() {
+        System.out.println("Package ordered, not delivered to the office yet.");
+    }
+
     @Override
     public String toString() {
         return "OrderedState{}";

patterns/design-patterns/src/main/java/com/baeldung/state/Package.java

@@ -19,4 +19,8 @@ public class Package {
     public void nextState() {
         state.next(this);
     }
+
+    public void printStatus() {
+        state.printStatus();
+    }
 }

patterns/design-patterns/src/main/java/com/baeldung/state/PackageState.java

@@ -5,4 +5,6 @@ public interface PackageState {
     void next(Package pkg);
 
     void prev(Package pkg);
+
+    void printStatus();
 }

patterns/design-patterns/src/main/java/com/baeldung/state/ReceivedState.java

@@ -16,6 +16,11 @@ public class ReceivedState implements PackageState {
         pkg.setState(new DeliveredState());
     }
 
+    @Override
+    public void printStatus() {
+        System.out.println("Package was received by client.");
+    }
+
     @Override
     public String toString() {
         return "ReceivedState{}";

patterns/design-patterns/src/main/java/com/baeldung/state/StateDemo.java

@@ -0,0 +1,19 @@
+package com.baeldung.state;
+
+public class StateDemo {
+
+    public static void main(String[] args) {
+
+        Package pkg = new Package();
+        pkg.printStatus();
+
+        pkg.nextState();
+        pkg.printStatus();
+
+        pkg.nextState();
+        pkg.printStatus();
+
+        pkg.nextState();
+        pkg.printStatus();
+    }
+}