1 files changed, 30 insertions, 34 deletions
diff --git a/src/test/java/com/att/nsa/mr/tools/AuthCommandTest.java b/src/test/java/com/att/nsa/mr/tools/AuthCommandTest.java
index 4b5e2d5..fdb7671 100644
--- a/src/test/java/com/att/nsa/mr/tools/AuthCommandTest.java
+++ b/src/test/java/com/att/nsa/mr/tools/AuthCommandTest.java
@@ -22,32 +22,29 @@ package com.att.nsa.mr.tools;
 
 import static org.junit.Assert.assertTrue;
 
-import java.io.FileNotFoundException;
-import java.io.IOException;
 import java.io.PrintStream;
-import java.util.ArrayList;
-import java.util.concurrent.TimeUnit;
 
 import org.junit.After;
 import org.junit.Before;
 import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.mockito.InjectMocks;
+import org.mockito.Mock;
+import org.mockito.MockitoAnnotations;
+import org.powermock.modules.junit4.PowerMockRunner;
 
 import com.att.nsa.cmdtool.CommandNotReadyException;
-import com.att.nsa.mr.client.HostSelector;
-import com.att.nsa.mr.client.MRPublisher.message;
-import com.att.nsa.mr.test.support.MRBatchingPublisherMock.Entry;
 
+@RunWith(PowerMockRunner.class)
 public class AuthCommandTest {
+	@InjectMocks
 	private AuthCommand command = null;
-	private String[] parts = new String[5];
+	@Mock
+	private PrintStream printStream;
 
 	@Before
 	public void setUp() throws Exception {
-		command = new AuthCommand();
-		
-		for (int i  = 0; i < parts.length; i++) {
-			parts[i] = "String" + (i + 1);
-		} 
+		MockitoAnnotations.initMocks(this);
 
 	}
 
@@ -63,7 +60,7 @@ public class AuthCommandTest {
 		assertTrue(true);
 
 	}
-	
+
 	@Test
 	public void testCheckReady() {
 
@@ -76,42 +73,41 @@ public class AuthCommandTest {
 		assertTrue(true);
 
 	}
-	
+
 	@Test
 	public void testExecute() {
-		
+
 		try {
-			command.execute(parts, new MRCommandContext(), new PrintStream("/filename"));
+			String[] parts = new String[5];
+			command.execute(parts, new MRCommandContext(), printStream);
 		} catch (CommandNotReadyException e) {
 			// TODO Auto-generated catch block
 			e.printStackTrace();
-		} catch (FileNotFoundException e) {
-			// TODO Auto-generated catch block
-			e.printStackTrace();
 		}
 		assertTrue(true);
 
 	}
-	
-	
+
 	@Test
-	public void testDisplayHelp() {
-		
+	public void testExecute1() {
+
 		try {
-			command.displayHelp(new PrintStream("/filename"));
-		} catch (NullPointerException e) {
-			// TODO Auto-generated catch block
-			e.printStackTrace();
-		} catch (FileNotFoundException e) {
+			String[] parts = { "userName", "password" };
+			command.execute(parts, new MRCommandContext(), printStream);
+		} catch (CommandNotReadyException e) {
 			// TODO Auto-generated catch block
 			e.printStackTrace();
 		}
 		assertTrue(true);
 
 	}
-	
-	
-	
-	
-	
+
+	@Test
+	public void testDisplayHelp() {
+
+		command.displayHelp(printStream);
+		assertTrue(true);
+
+	}
+
 }