예제 #1
0
 @Test
 public void createProtocol() {
   final Protocol protocol = context.mock(Protocol.class);
   final ProtocolProvider provider = context.mock(ProtocolProvider.class);
   context.checking(
       new Expectations() {
         {
           allowing(provider).getProtocol(1337);
           will(returnValue(protocol));
           allowing(provider).getProtocol(with(any(Integer.class)));
           will(returnValue(null));
         }
       });
   prot.addProtocolProvider(provider);
   assertEquals(protocol, prot.getProtocolById(1337));
   assertNull(prot.getProtocolById(1));
 }
예제 #2
0
  @Test
  public void changesToCorrectProtocol() throws IOException {
    final Protocol protocol = context.mock(Protocol.class);
    final ProtocolProvider provider = context.mock(ProtocolProvider.class);
    context.checking(
        new Expectations() {
          {
            allowing(provider).getProtocol(123);
            will(returnValue(protocol));
            oneOf(protocol).parseFirst(with(any(InputStream.class)));
            oneOf(protocol).init(mockedClient);
          }
        });
    prot.addProtocolProvider(provider);

    ByteArrayOutputStream out = new ByteArrayOutputStream();
    CData.writeU16(out, 1);
    CData.writeByte(out, 123);
    prot.messageReceived(mockedClient, new ByteArrayInputStream(out.toByteArray()));

    assertSame(protocol, prot.getCurrentProtocol());
    context.assertIsSatisfied();
  }