Merge pull request #146 from vanitasvitae/fixProxyTest

Fix shouldPreserveAddressOrderOnInsertions test
This commit is contained in:
Florian Schmaus 2017-07-05 08:33:12 +02:00 committed by GitHub
commit 6775cf862d
1 changed files with 16 additions and 14 deletions

View File

@ -30,6 +30,8 @@ import java.net.Socket;
import java.net.SocketException; import java.net.SocketException;
import java.net.UnknownHostException; import java.net.UnknownHostException;
import java.util.ArrayList; import java.util.ArrayList;
import java.util.Iterator;
import java.util.LinkedHashSet;
import java.util.List; import java.util.List;
import org.jivesoftware.smack.util.StringUtils; import org.jivesoftware.smack.util.StringUtils;
@ -39,7 +41,7 @@ import org.junit.Test;
/** /**
* Test for Socks5Proxy class. * Test for Socks5Proxy class.
* *
* @author Henning Staib * @author Henning Staib
*/ */
public class Socks5ProxyTest { public class Socks5ProxyTest {
@ -73,7 +75,7 @@ public class Socks5ProxyTest {
/** /**
* The SOCKS5 proxy should use a free port above the one configured. * The SOCKS5 proxy should use a free port above the one configured.
* *
* @throws Exception should not happen * @throws Exception should not happen
*/ */
@Test @Test
@ -102,13 +104,11 @@ public class Socks5ProxyTest {
@Test @Test
public void shouldPreserveAddressOrderOnInsertions() { public void shouldPreserveAddressOrderOnInsertions() {
Socks5Proxy proxy = Socks5Proxy.getSocks5Proxy(); Socks5Proxy proxy = Socks5Proxy.getSocks5Proxy();
List<String> addresses = new ArrayList<>(proxy.getLocalAddresses());
LinkedHashSet<String> addresses = new LinkedHashSet<>(proxy.getLocalAddresses());
for (int i = 1 ; i <= 3; i++) { for (int i = 1 ; i <= 3; i++) {
String addr = Integer.toString(i); addresses.add(Integer.toString(i));
if (!addresses.contains(addr)) {
addresses.add(addr);
}
} }
for (String address : addresses) { for (String address : addresses) {
@ -116,8 +116,10 @@ public class Socks5ProxyTest {
} }
List<String> localAddresses = proxy.getLocalAddresses(); List<String> localAddresses = proxy.getLocalAddresses();
Iterator<String> iterator = addresses.iterator();
for (int i = 0; i < addresses.size(); i++) { for (int i = 0; i < addresses.size(); i++) {
assertEquals(addresses.get(i), localAddresses.get(i)); assertEquals(iterator.next(), localAddresses.get(i));
} }
} }
@ -165,7 +167,7 @@ public class Socks5ProxyTest {
/** /**
* If the SOCKS5 proxy accepts a connection that is not a SOCKS5 connection it should close the * If the SOCKS5 proxy accepts a connection that is not a SOCKS5 connection it should close the
* corresponding socket. * corresponding socket.
* *
* @throws Exception should not happen * @throws Exception should not happen
*/ */
@Test @Test
@ -196,7 +198,7 @@ public class Socks5ProxyTest {
/** /**
* The SOCKS5 proxy should reply with an error message if no supported authentication methods * The SOCKS5 proxy should reply with an error message if no supported authentication methods
* are given in the SOCKS5 request. * are given in the SOCKS5 request.
* *
* @throws Exception should not happen * @throws Exception should not happen
*/ */
@Test @Test
@ -227,7 +229,7 @@ public class Socks5ProxyTest {
/** /**
* The SOCKS5 proxy should respond with an error message if the client is not allowed to connect * The SOCKS5 proxy should respond with an error message if the client is not allowed to connect
* with the proxy. * with the proxy.
* *
* @throws Exception should not happen * @throws Exception should not happen
*/ */
@Test @Test
@ -249,7 +251,7 @@ public class Socks5ProxyTest {
// send valid SOCKS5 message // send valid SOCKS5 message
out.write(new byte[] { (byte) 0x05, (byte) 0x00, (byte) 0x00, (byte) 0x03, (byte) 0x01, out.write(new byte[] { (byte) 0x05, (byte) 0x00, (byte) 0x00, (byte) 0x03, (byte) 0x01,
(byte) 0xAA, (byte) 0x00, (byte) 0x00 }); (byte) 0xAA, (byte) 0x00, (byte) 0x00 });
// verify error message // verify error message
assertEquals((byte) 0x05, (byte) in.read()); assertEquals((byte) 0x05, (byte) in.read());
@ -269,7 +271,7 @@ public class Socks5ProxyTest {
/** /**
* A Client should successfully establish a connection to the SOCKS5 proxy. * A Client should successfully establish a connection to the SOCKS5 proxy.
* *
* @throws Exception should not happen * @throws Exception should not happen
*/ */
@Test @Test
@ -297,7 +299,7 @@ public class Socks5ProxyTest {
// send valid SOCKS5 message // send valid SOCKS5 message
out.write(new byte[] { (byte) 0x05, (byte) 0x00, (byte) 0x00, (byte) 0x03, (byte) 0x01, out.write(new byte[] { (byte) 0x05, (byte) 0x00, (byte) 0x00, (byte) 0x03, (byte) 0x01,
(byte) 0xAA, (byte) 0x00, (byte) 0x00 }); (byte) 0xAA, (byte) 0x00, (byte) 0x00 });
// verify response // verify response
assertEquals((byte) 0x05, (byte) in.read()); assertEquals((byte) 0x05, (byte) in.read());