/*

* Copyright (c) 2005-2009 Nokia Corporation and/or its subsidiary(-ies). 

* All rights reserved.

* This component and the accompanying materials are made available

* under the terms of "Eclipse Public License v1.0"

* which accompanies this distribution, and is available

* at the URL "http://www.eclipse.org/legal/epl-v10.html".

*

* Initial Contributors:

* Nokia Corporation - initial contribution.

*

* Contributors:

*

* Description: 

*

*/

import java.io.IOException;

import java.io.InputStream;

import java.io.OutputStream;

import java.util.regex.Matcher;

import java.util.regex.Pattern;

import junit.framework.TestCase;

import org.apache.commons.net.telnet.TelnetClient;

public class TelnetTest extends TestCase {

	/* (non-Javadoc)

	 * @see junit.framework.TestCase#setUp()

	 */

	TelnetClient telnetClient = new TelnetClient();

	InputStream is;

	OutputStream os;

	byte promptBuffer[] = new byte[11]; //"localhost# "

	String prompt = "localhost# ";

	//FTP&Telnet-Connection-001

	//To Establish a connection from remote host to Symbian OS device(H2/H4)

	protected void setUp() throws Exception {

		super.setUp();

		String address = System.getProperty("address");

		telnetClient.connect(/*"172.0.0.3"*/address);

		is = telnetClient.getInputStream();

		os = telnetClient.getOutputStream();

		is.read(promptBuffer);

	}

	/* (non-Javadoc)

	 * @see junit.framework.TestCase#tearDown()

	 */

	//FTP&Telnet-Connection-002

	//To Close a connection from remote host to Symbian OS device(H2/H4)

	protected void tearDown() throws Exception {

		super.tearDown();

		os.write("exit\r\n".getBytes());

		os.flush();

		telnetClient.disconnect();

	}

	private String readUntil(InputStream in, String pattern) throws IOException {

		StringBuffer sb = new StringBuffer();

		while(!(sb.toString().endsWith(prompt)))

		{

			sb.append((char)in.read());

		}

		return sb.toString();

	}

	public void testAYT() throws IOException, IllegalArgumentException, InterruptedException

	{

		assertTrue(telnetClient.sendAYT(5000));

	}

	public void testCd() throws IOException

	{

		os.write("cd c:\r\n".getBytes());

		os.flush();

		is.read(promptBuffer);	//"localhost# "

		assertEquals(prompt, new String(promptBuffer));

	}

	public void testPwd() throws IOException

	{

		os.write("pwd\r\n".getBytes());

		os.flush();

		String s = readUntil(is,prompt);

		s = s.substring(1, s.indexOf(prompt));

		assertEquals(":\\Private\\2000cd0f\\\r\n", s);

	}

	public void testCdPwd() throws IOException

	{

		os.write("cd c:\r\n".getBytes());

		os.flush();

		is.read(promptBuffer);	//"localhost# "

		assertEquals(new String(promptBuffer),prompt);

		os.write("pwd\r\n".getBytes());

		os.flush();

		String s = readUntil(is,prompt);

		s = s.substring(0, s.indexOf(prompt));

		assertEquals("c:\r\n", s);

	}

	//DEF117414

	//zsh cannot run external commands after running 82 commands

	//TODO: change maxIter to 100 when fixed

	public void testPs() throws IOException

	{

		int maxIter = 2;

		for(int i=0; i<maxIter; i++)

		{

		os.write("ps\r\n".getBytes());

		os.flush();

		String s = readUntil(is,prompt);

		s = s.substring(0, s.indexOf(prompt));

		System.out.println(s);

		System.out.println("iteration "+(i+1)+" of "+maxIter);

		Pattern p = Pattern.compile(".*ps.EXE\\[2000ee91\\]\\d\\d\\d\\d\r\n$", Pattern.DOTALL | Pattern.CASE_INSENSITIVE); //Pattern.DOTALL => '.' includes end of line

		Matcher m = p.matcher(s);

		assertTrue(m.matches());

		}

	}

	//DEF116570

	//after few successful ps and grep, when you do ps and grep on non exsiting string the emulator hangs

	public void testPsGrepNonExistingString() throws IOException

	{

		int maxIter = 2;

		for(int i=0; i<maxIter; i++)

		{

		os.write("ps | grep nonexisting\r\n".getBytes());

		os.flush();

		String s = readUntil(is,prompt);

		s = s.substring(0, s.indexOf(prompt));

		assertEquals(0,s.length());

		}

	}

	//FTP&Telnet-Remote Execution-001

	//To Launch a process on a remote Symbian OS device(H2/H4) from PC

	public void testFTP_Telnet_Remote_Execution_001() throws IOException

	{

		os.write("testexecute c:\\nonexisting.script\r\n".getBytes());

		os.flush();

		String s = readUntil(is,prompt);

		assertEquals(11, s.length());

	}

	//FTP&Telnet-Remote Execution-002

	//To pool all the process on the symbian OS device(H2/H4) from PC

	public void testFTP_Telnet_Remote_Execution_002() throws IOException

	{

		os.write("ps\r\n".getBytes());

		os.flush();

		String s = readUntil(is,prompt);

		s = s.substring(0, s.indexOf(prompt));

	//	System.out.println(s);

		Pattern p = Pattern.compile(".*ps.EXE\\[2000ee91\\]\\d\\d\\d\\d\r\n$", Pattern.DOTALL | Pattern.CASE_INSENSITIVE); //Pattern.DOTALL => '.' includes end of line

		Matcher m = p.matcher(s);

		assertTrue(m.matches());

	}

	//FTP&Telnet-Remote Execution-003

	// Kill any process on Symbian OS device(H2/H4) from PC

	public void testFTP_Telnet_Remote_Execution_003() throws IOException

	{

		os.write("testexecute\r\n".getBytes());

		os.flush();

		//this shell will lock, open another one and kill TEF

		TelnetClient tc = new TelnetClient();

		tc.connect(telnetClient.getRemoteAddress());

		InputStream is2 = tc.getInputStream();

		OutputStream os2 = tc.getOutputStream();

		readUntil(is2, prompt);

		os2.write("ps | grep TestExecute\r\n".getBytes());

		os2.flush();

		String s = readUntil(is2, prompt);

		Pattern p = Pattern.compile(".*\\D+(\\d+)\\s[\\s:\\d]+TestExecute Script Engine.*", Pattern.DOTALL | Pattern.CASE_INSENSITIVE); //Pattern.DOTALL => '.' includes end of line

		Matcher m = p.matcher(s);

		assertTrue(m.matches());

		String s1 = m.group(1);

		int pid = Integer.parseInt(s1);

		os2.write(("kill "+pid+"\r\n").getBytes());

		os2.flush();

		readUntil(is2, prompt);

		os2.write("bye\r\n".getBytes());

		os2.flush();

		//we should be able now to close the other shell

		readUntil(is, prompt);

	}

	//FTP&Telnet-Remote Execution-004

	//To verify the output stream of any Symbian OS process(H2/H4) to PC

	public void testFTP_Telnet_Remote_Execution_004() throws IOException

	{

		os.write("helloworld $PIPE\r\n".getBytes());

		os.flush();

		String s = readUntil(is, prompt);

		assertTrue(s.length()>0); //should print "ABCDEFG...".length() > 0

	}

	//FTP&Telnet-Remote Execution-005

	//To verify the Standard Error display on Symbian OS device(H2/H4) to PC

	public void testFTP_Telnet_Remote_Execution_005() throws IOException

	{

		os.write("aaa\r\n".getBytes());

		os.flush();

		String s = readUntil(is, prompt);

		assertEquals("zsh: command not found: aaa",s.substring(0, s.lastIndexOf("\r\n"))); 

	}

	//FTP&Telnet-Software Install-001

	//Install a SIS file on the Symbian OS device (H2/H4) from PC

	public void testFTP_Telnet_Install_001() throws IOException

	{

		os.write("install c:\\10210d02.sis\r\n".getBytes());

		os.flush();

		String s = readUntil(is, prompt);

		assertEquals("Return Code: 0",s.substring(0, s.lastIndexOf("\r\n"))); 

	}

	//FTP&Telnet-Software Install-002

	//Uninstall the SIS file on the symbian OS device (H2/H4) from PC

	public void testFTP_Telnet_Install_002() throws IOException

	{

		os.write("uninstall 10210d02\r\n".getBytes());

		os.flush();

		String s = readUntil(is, prompt);

		assertEquals("Return Code: 0",s.substring(0, s.lastIndexOf("\r\n"))); 

	}

}