Kaladaran/briar
0
0
Fork 0
mirror of https://code.briarproject.org/briar/briar.git synced 2026-02-19 22:29:53 +01:00
Code Issues Projects Releases Wiki Activity

Add unit test for connection limiter.

Browse Source
This commit is contained in:
akwizgran
2020-05-05 17:16:10 +01:00
parent d905451f48
commit 104a82aea9
1 changed files with 55 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

55
bramble-core/src/test/java/org/briarproject/bramble/plugin/bluetooth/BluetoothConnectionLimiterImplTest.java Normal file
View File

@@ -0,0 +1,55 @@
package org.briarproject.bramble.plugin.bluetooth;
import org.briarproject.bramble.api.plugin.TransportConnectionReader;
import org.briarproject.bramble.api.plugin.TransportConnectionWriter;
import org.briarproject.bramble.api.plugin.duplex.DuplexTransportConnection;
import org.briarproject.bramble.test.BrambleMockTestCase;
import org.jmock.Expectations;
import org.junit.Test;
import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertTrue;
public class BluetoothConnectionLimiterImplTest extends BrambleMockTestCase {
private final DuplexTransportConnection conn =
context.mock(DuplexTransportConnection.class);
private final TransportConnectionReader reader =
context.mock(TransportConnectionReader.class);
private final TransportConnectionWriter writer =
context.mock(TransportConnectionWriter.class);
@Test
public void testLimiterAllowsOneOutgoingConnection() {
BluetoothConnectionLimiter limiter =
new BluetoothConnectionLimiterImpl();
assertTrue(limiter.canOpenContactConnection());
assertTrue(limiter.contactConnectionOpened(conn));
assertFalse(limiter.canOpenContactConnection());
}
@Test
public void testLimiterAllowsSecondIncomingConnection() throws Exception {
BluetoothConnectionLimiter limiter =
new BluetoothConnectionLimiterImpl();
assertTrue(limiter.canOpenContactConnection());
assertTrue(limiter.contactConnectionOpened(conn));
assertFalse(limiter.canOpenContactConnection());
// The limiter allows a second incoming connection
assertTrue(limiter.contactConnectionOpened(conn));
// The limiter closes any further incoming connections
expectCloseConnection();
assertFalse(limiter.contactConnectionOpened(conn));
}
private void expectCloseConnection() throws Exception {
context.checking(new Expectations() {{
oneOf(conn).getReader();
will(returnValue(reader));
oneOf(reader).dispose(false, false);
oneOf(conn).getWriter();
will(returnValue(writer));
oneOf(writer).dispose(false);
}});
}
}