scoobycoder
diff --git a/‎bin/halterman/command/CommandPatternTest.class
203 Bytes b/‎bin/halterman/command/CommandPatternTest.class
203 Bytes
diff --git a/‎test/halterman/command/CommandPatternTest.java
Lines changed: 7 additions & 13 deletions b/‎test/halterman/command/CommandPatternTest.java
Lines changed: 7 additions & 13 deletions
@@ -1,19 +1,13 @@
 package halterman.command;
 
-import static org.junit.Assert.*;
-import halterman.command.Command;
-import halterman.command.Light;
-import halterman.command.LightOnCommand;
-import halterman.command.Switch;
-
-import org.mockito.InjectMocks;
-import org.mockito.Mock;
-import org.mockito.Mockito;
-import org.mockito.MockitoAnnotations;
 import static org.mockito.Mockito.verify;
 
 import org.junit.Before;
 import org.junit.Test;
+import org.mockito.InjectMocks;
+import org.mockito.Mock;
+import org.mockito.MockitoAnnotations;
+import org.mockito.Spy;
 
 public class CommandPatternTest {
 
@@ -23,20 +17,20 @@ public class CommandPatternTest {
 	@Mock
 	Light light;
 
-	@Mock
-	Command lightsOn;
+	@Spy
+	Command lightsOn = new LightOnCommand(light);
 
 	@Before
     public void setUp() {
         MockitoAnnotations.initMocks(this);
         lightSwitch = new Switch();
+        lightSwitch.setCommand(lightsOn);
     }
 
 	@Test
 	public void shouldTurnOnLight() {
 		lightSwitch.pressButton();
 
-		
 		verify(light).switchOn();
 	}