TDD: Write test code for JschServer

Author: Dokyeongyun

src/main/java/root/common/server/implement/JschServer.java

@@ -9,10 +9,12 @@
 import com.jcraft.jsch.JSchException;
 import com.jcraft.jsch.Session;
 
+import lombok.ToString;
 import lombok.extern.slf4j.Slf4j;
 import root.core.domain.JschConnectionInfo;
 
 @Slf4j
+@ToString
 public class JschServer {
 	private JSch jsch;
 	private Session session;
@@ -40,59 +42,65 @@ public void init() {
 			log.error(e.getMessage());
 		}
 	}
-	
+
 	public Session getSession() {
-		if(session == null) {
+		if (session == null) {
 			return null;
 		}
 		return session;
 	}
-	
-	public Session connect(Session session) {
+
+	public Session connect(Session session) throws JSchException {
+		if(session == null) {
+			throw new NullPointerException("Session is null");
+		}
+		
 		try {
 			session.connect();
 		} catch (JSchException e) {
 			log.error(e.getMessage());
+			throw e;
 		}
+		
 		return session;
 	}
-	
+
 	public void disConnect(Session session) {
 		session.disconnect();
 	}
-	
+
 	public Channel openExecChannel(Session session, String command) {
 		Channel channel = null;
 		try {
 			channel = session.openChannel("exec");
-			//채널접속
-			ChannelExec channelExec = (ChannelExec) channel; //명령 전송 채널사용
+			// 채널접속
+			ChannelExec channelExec = (ChannelExec) channel; // 명령 전송 채널사용
 			channelExec.setPty(true);
-			channelExec.setCommand(command); 
+			channelExec.setCommand(command);
 		} catch (JSchException e) {
 			log.error(e.getMessage());
 		}
 		return channel;
 	}
-	
+
 	public InputStream connectChannel(Channel channel) {
 		InputStream in = null;
 		try {
 			// CallBack
 			in = channel.getInputStream();
-			((ChannelExec) channel).setErrStream(System.err);        
-			
+			((ChannelExec) channel).setErrStream(System.err);
+
 			channel.connect();
 		} catch (Exception e) {
 			log.error(e.getMessage());
 		}
 		return in;
 	}
-	
+
 	public void disConnectChannel(Channel channel) {
-	    channel.disconnect();
+		channel.disconnect();
 	}
-	
+
 	public String getServerName() {
 		return this.jschConnectionInfo.getServerName();
 	}
@@ -104,12 +112,12 @@ public static boolean validateConn(Session session) {
 		}
 
 		try {
-			session.connect(15);
+			session.connect(3000);
 		} catch (JSchException e) {
 			log.error(e.getMessage());
 			return false;
 		}
-		
+
 		return session.isConnected();
 	}
 }

src/test/java/root/common/server/implement/JschServerTest.java

@@ -0,0 +1,81 @@
+package root.common.server.implement;
+
+import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assertions.assertNotNull;
+import static org.junit.jupiter.api.Assertions.assertNull;
+import static org.junit.jupiter.api.Assertions.assertThrows;
+import static org.junit.jupiter.api.Assertions.assertTrue;
+import static org.junit.jupiter.api.Assertions.fail;
+
+import org.junit.jupiter.api.BeforeAll;
+import org.junit.jupiter.api.Test;
+
+import com.jcraft.jsch.JSchException;
+import com.jcraft.jsch.Session;
+
+import root.core.domain.JschConnectionInfo;
+
+public class JschServerTest {
+
+	public static JschServer jsch;
+
+	@BeforeAll
+	public static void setup() {
+		String serverName = "DKY SERVER";
+		String host = "192.168.154.1";
+		String userName = "dky";
+		String port = "22";
+		String password = "ehruddbs1!";
+		jsch = new JschServer(new JschConnectionInfo(serverName, host, port, userName, password));
+	}
+
+	@Test
+	public void testInit_ValidConnInfo() {
+		jsch.init();
+		Session session = jsch.getSession();
+		assertNotNull(session);
+	}
+
+	@Test
+	public void testInit_Fail_InvalidConnInfo() {
+		JschServer jsch = new JschServer(new JschConnectionInfo());
+		jsch.init();
+		Session session = jsch.getSession();
+		assertNull(session);
+	}
+
+	@Test
+	public void testConnect_ConnectionSuccess() {
+		jsch.init();
+		Session session = jsch.getSession();
+		try {
+			session = jsch.connect(session);
+		} catch (JSchException e) {
+			fail();
+		}
+		assertTrue(session.isConnected());
+	}
+
+	@Test
+	public void testConnect_ConnectionFail_SessionIsNull() {
+		JschServer jsch = new JschServer(new JschConnectionInfo());
+		jsch.init();
+		Session session = jsch.getSession();
+		NullPointerException thrown = assertThrows(NullPointerException.class, () -> jsch.connect(session));
+		assertEquals("Session is null", thrown.getMessage());
+	}
+
+	@Test
+	public void testConnect_ConnectionFail_JSchException() {
+		jsch.init();
+		Session session = jsch.getSession();
+		try {
+			jsch.connect(session);
+		} catch (JSchException e) {
+			fail();
+		}
+
+		JSchException thrown = assertThrows(JSchException.class, () -> jsch.connect(session));
+		assertEquals("session is already connected", thrown.getMessage());
+	}
+}