Add constructor injection in franchise

src/pizzeria/PizzaFranchise.java

+package pizzeria;
+
+public class PizzaFranchise {
+	private PizzaStore store;
+	
+	public PizzaFranchise(PizzaStore store) {
+		this.store = store;
+	}
+	
+	public Pizza orderPizza(String type) {
+		return store.orderPizza(type);
+	}
+}

src/pizzeria/PizzaTestDrive.java

@@ -3,31 +3,17 @@ package pizzeria;
 public class PizzaTestDrive {
  
 	public static void main(String[] args) {
-		PizzaStore nyStore = new NYPizzaStore();
-		PizzaStore chicagoStore = new ChicagoPizzaStore();
- 
-		Pizza pizza = nyStore.orderPizza("cheese");
-		System.out.println("Ethan ordered a " + pizza.getName() + "\n");
- 
-		pizza = chicagoStore.orderPizza("cheese");
-		System.out.println("Joel ordered a " + pizza.getName() + "\n");
-
-		pizza = nyStore.orderPizza("clam");
-		System.out.println("Ethan ordered a " + pizza.getName() + "\n");
- 
-		pizza = chicagoStore.orderPizza("clam");
-		System.out.println("Joel ordered a " + pizza.getName() + "\n");
-
-		pizza = nyStore.orderPizza("pepperoni");
-		System.out.println("Ethan ordered a " + pizza.getName() + "\n");
- 
-		pizza = chicagoStore.orderPizza("pepperoni");
-		System.out.println("Joel ordered a " + pizza.getName() + "\n");
-
-		pizza = nyStore.orderPizza("veggie");
-		System.out.println("Ethan ordered a " + pizza.getName() + "\n");
- 
-		pizza = chicagoStore.orderPizza("veggie");
-		System.out.println("Joel ordered a " + pizza.getName() + "\n");
+		PizzaStore[] storeList = {new TestPizzaStore(), new NYPizzaStore(), new ChicagoPizzaStore()};
+		String[] customerList = {"Ethan", "Joel", "Martha"}; 
+		String[] pizzaType = {"cheese", "veggie"};
+		
+		for (int i=0; i<storeList.length; i++) {
+			PizzaFranchise franchise = new PizzaFranchise(storeList[i]);
+				
+			for (int j=0; j<pizzaType.length; j++) {
+				Pizza pizza = franchise.orderPizza(pizzaType[j]);
+				System.out.println(customerList[i] + " ordered a " + pizza.getName() + "\n");
+			}
+		}
 	}
 }

src/pizzeria/TestCheesePizza.java

+package pizzeria;
+
+public class TestCheesePizza extends Pizza {
+
+	public TestCheesePizza() { 
+		name = "Test Cheese Pizza";
+		dough = "Test Dough";
+		sauce = "Test Sauce";
+ 
+		toppings.add("Testing cheese toppings");
+	}
+}

src/pizzeria/TestPizzaStore.java

+package pizzeria;
+
+public class TestPizzaStore extends PizzaStore {
+
+	Pizza createPizza(String item) {
+		if (item.equals("cheese")) {
+			return new TestCheesePizza();
+		} else if (item.equals("veggie")) {
+			return new TestVeggiePizza();
+		} else return null;
+	}
+	
+	
+}

src/pizzeria/TestVeggiePizza.java

+package pizzeria;
+
+public class TestVeggiePizza extends Pizza {
+
+	public TestVeggiePizza() {
+		name = "Test Veggie Pizza";
+		dough = "Test Dough";
+		sauce = "Test Sauce";
+ 
+		toppings.add("Testing veggie toppings");
+	}
+}