Examples with MockTSocket com.twitter.common.thrift.testing.MockTSocket used on opensource projects

Example 1 with MockTSocket

use of com.twitter.common.thrift.testing.MockTSocket in project commons by twitter.

the class ThriftConnectionFactoryTest method createConnectionFactory.

private ThriftConnectionFactory createConnectionFactory(int maxConnections) {
    return new ThriftConnectionFactory("foo", 1234, maxConnections) {

        @Override
        TTransport createTransport(int timeoutMillis) throws TTransportException {
            TTransport transport = new MockTSocket();
            transport.open();
            return transport;
        }
    };
}

Example 2 with MockTSocket

use of com.twitter.common.thrift.testing.MockTSocket in project commons by twitter.

the class DynamicPoolTest method testPoolRebuilds.

@Test
@SuppressWarnings("unchecked")
public void testPoolRebuilds() throws Exception {
    ConnectionFactory<Connection<TTransport, InetSocketAddress>> connectionFactory = control.createMock(ConnectionFactory.class);
    TTransport transport = new MockTSocket();
    Connection<TTransport, InetSocketAddress> connection = new TTransportConnection(transport, InetSocketAddress.createUnresolved("jake", 1137));
    expect(connectionFactory.create(EasyMock.isA(Amount.class))).andReturn(connection);
    ConnectionPool<Connection<TTransport, InetSocketAddress>> fooPool = new ConnectionPool<Connection<TTransport, InetSocketAddress>>(connectionFactory);
    expect(poolFactory.apply(InetSocketAddress.createUnresolved("foo", 42))).andReturn(fooPool);
    control.replay();
    Pair<Set<ObjectPool<Connection<TTransport, InetSocketAddress>>>, Map<InetSocketAddress, ObjectPool<Connection<TTransport, InetSocketAddress>>>> rebuild1 = poolRebuilds.take();
    assertTrue("Should not have any dead pools on initial rebuild", rebuild1.getFirst().isEmpty());
    assertNoLivePools(rebuild1);
    EndpointStatus fooStatus = join("foo");
    Pair<Set<ObjectPool<Connection<TTransport, InetSocketAddress>>>, Map<InetSocketAddress, ObjectPool<Connection<TTransport, InetSocketAddress>>>> rebuild2 = poolRebuilds.take();
    assertTrue("The NULL pool should never be tracked as dead", rebuild2.getFirst().isEmpty());
    assertEquals(transport, connectionPool.get().get());
    fooStatus.leave();
    Pair<Set<ObjectPool<Connection<TTransport, InetSocketAddress>>>, Map<InetSocketAddress, ObjectPool<Connection<TTransport, InetSocketAddress>>>> rebuild3 = poolRebuilds.take();
    assertSame("Expected foo pool to be discarded", fooPool, Iterables.getOnlyElement(rebuild3.getFirst()));
    assertNoLivePools(rebuild1);
    control.verify();
}