1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
|
/*******************************************************************************
* Copyright (c) 2011 SAP AG
* All rights reserved. This program and the accompanying materials
* are made available under the terms of the Eclipse Public License v1.0
* which accompanies this distribution, and is available at
* http://www.eclipse.org/legal/epl-v10.html
*
* Contributors:
* Lazar Kirchev, SAP AG - initial contribution
******************************************************************************/
package org.eclipse.equinox.console.telnet;
import org.apache.felix.service.command.CommandProcessor;
import org.apache.felix.service.command.CommandSession;
import org.junit.Assert;
import org.junit.Test;
import org.easymock.EasyMock;
import org.eclipse.equinox.console.common.ConsoleInputStream;
import java.io.InputStream;
import java.io.OutputStream;
import java.io.PrintStream;
import java.net.ServerSocket;
import java.net.Socket;
public class TelnetConnectionTests {
private static final String HOST = "localhost";
private static final int TEST_CONTENT = 100;
private static final int IAC = 255;
@Test
public void testTelneConnection() throws Exception {
ServerSocket servSocket = null;
Socket socketClient = null;
Socket socketServer = null;
TelnetConnection connection = null;
OutputStream outClient = null;
OutputStream outServer = null;
try {
servSocket = new ServerSocket(0);
socketClient = new Socket(HOST, servSocket.getLocalPort());
socketServer = servSocket.accept();
CommandSession session = EasyMock.createMock(CommandSession.class);
session.put((String)EasyMock.anyObject(), EasyMock.anyObject());
EasyMock.expectLastCall().times(3);
EasyMock.expect(session.execute((String)EasyMock.anyObject())).andReturn(null);
session.close();
EasyMock.expectLastCall();
EasyMock.replay(session);
CommandProcessor processor = EasyMock.createMock(CommandProcessor.class);
EasyMock.expect(processor.createSession((ConsoleInputStream) EasyMock.anyObject(), (PrintStream) EasyMock.anyObject(), (PrintStream) EasyMock.anyObject())).andReturn(session);
EasyMock.replay(processor);
connection = new TelnetConnection(socketServer, processor, null);
connection.start();
outClient = socketClient.getOutputStream();
outClient.write(TEST_CONTENT);
outClient.write('\n');
outClient.flush();
InputStream input = socketServer.getInputStream();
int in = input.read();
Assert.assertTrue("Server received [" + in + "] instead of " + TEST_CONTENT + " from the telnet client.", in == TEST_CONTENT);
input = socketClient.getInputStream();
in = input.read();
// here IAC is expected, since when the output stream in TelnetConsoleSession is created, several telnet
// commands are written to it, each of them starting with IAC
Assert.assertTrue("Client receive telnet responses from the server unexpected value [" + in + "] instead of " + IAC + ".", in == IAC);
connection.telnetNegotiationFinished();
Thread.sleep(5000);
EasyMock.verify(session, processor);
} finally {
if (socketClient != null) {
socketClient.close();
}
if (outClient != null) {
outClient.close();
}
if (outServer != null) {
outServer.close();
}
if (socketServer != null) {
socketServer.close();
}
if (servSocket != null) {
servSocket.close();
}
}
}
}
|