From eb9b5fdd1b0e510d5acf7fbc0154032af7184966 Mon Sep 17 00:00:00 2001
From: ruslanpa <ruslan.pavlutskiy@gmail.com>
Date: Mon, 9 Feb 2015 21:52:39 +0200
Subject: [PATCH] [refactor] Update WeaponType enum in factory-method pattern.

---
 .../main/java/com/iluwatar/WeaponType.java    | 25 +++++++------------
 1 file changed, 9 insertions(+), 16 deletions(-)

diff --git a/factory-method/src/main/java/com/iluwatar/WeaponType.java b/factory-method/src/main/java/com/iluwatar/WeaponType.java
index 4097fd3c7..8d5ab54cc 100644
--- a/factory-method/src/main/java/com/iluwatar/WeaponType.java
+++ b/factory-method/src/main/java/com/iluwatar/WeaponType.java
@@ -2,23 +2,16 @@ package com.iluwatar;
 
 public enum WeaponType {
 
-	SHORT_SWORD, SPEAR, AXE;
+	SHORT_SWORD("short sword"), SPEAR("spear"), AXE("axe"), UNDEFINED("");
 
-	@Override
+    private String title;
+
+    WeaponType(String title) {
+        this.title = title;
+    }
+
+    @Override
 	public String toString() {
-		String s = "";
-		switch (this) {
-		case SHORT_SWORD:
-			s = "short sword";
-			break;
-		case SPEAR:
-			s = "spear";
-			break;
-		case AXE:
-			s = "axe";
-			break;
-		}
-		return s;
+		return title;
 	}
-
 }