/*
    *  Licensed to the Apache Software Foundation (ASF) under one
    *  or more contributor license agreements.  See the NOTICE file
    *  distributed with this work for additional information
    *  regarding copyright ownership.  The ASF licenses this file
    *  to you under the Apache License, Version 2.0 (the
    *  "License"); you may not use this file except in compliance
    *  with the License.  You may obtain a copy of the License at
    *
   *    http://www.apache.org/licenses/LICENSE-2.0
   *
   *  Unless required by applicable law or agreed to in writing,
   *  software distributed under the License is distributed on an
   *  "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
   *  KIND, either express or implied.  See the License for the
   *  specific language governing permissions and limitations
   *  under the License.
   *
   */
  package org.apache.mina.filter.codec;
  
  import org.apache.mina.filter.codec.demux.MessageDecoderResult;
  import org.apache.mina.filter.codec.demux.MessageDecoderAdapter;
  import org.apache.mina.filter.codec.demux.DemuxingProtocolDecoder;
  import org.apache.mina.filter.codec.ProtocolDecoderOutput;
  import org.apache.mina.core.session.IoSession;
  import org.apache.mina.core.session.DummySession;
  import org.apache.mina.core.buffer.IoBuffer;
  import org.apache.mina.core.service.DefaultTransportMetadata;
  import org.apache.mina.core.file.FileRegion;
  import org.apache.mina.transport.socket.SocketSessionConfig;
  import org.easymock.EasyMock;
  import org.junit.Test;
  
  import java.net.InetSocketAddress;
  import java.nio.charset.Charset;
  
  /**
   * Simple Unit Test showing that the DemuxingProtocolDecoder has
   * inconsistent behavior if used with a non fragmented transport.
   * 
   * @author <a href="http://mina.apache.org">Apache MINA Project</a>
  */
  public class DemuxingProtocolDecoderBugTest {
  
      private static void doTest(IoSession session) throws Exception {
          ProtocolDecoderOutput mock = EasyMock.createMock(ProtocolDecoderOutput.class);
          mock.write(Character.valueOf('A'));
          mock.write(Character.valueOf('B'));
          mock.write(Integer.valueOf(1));
          mock.write(Integer.valueOf(2));
          mock.write(Character.valueOf('C'));
          EasyMock.replay(mock);
  
          IoBuffer buffer = IoBuffer.allocate(1000);
          buffer.putString("AB12C", Charset.defaultCharset().newEncoder());
          buffer.flip();
  
          DemuxingProtocolDecoder decoder = new DemuxingProtocolDecoder();
          decoder.addMessageDecoder(CharacterMessageDecoder.class);
          decoder.addMessageDecoder(IntegerMessageDecoder.class);
  
          decoder.decode(session, buffer, mock);
  
          EasyMock.verify(mock);
      }
  
      public static class CharacterMessageDecoder extends MessageDecoderAdapter {
          public MessageDecoderResult decodable(IoSession session, IoBuffer in) {
              return Character.isDigit((char) in.get()) ? MessageDecoderResult.NOT_OK : MessageDecoderResult.OK;
          }
  
          public MessageDecoderResult decode(IoSession session, IoBuffer in, ProtocolDecoderOutput out) throws Exception {
              out.write(Character.valueOf((char) in.get()));
              return MessageDecoderResult.OK;
          }
      }
  
      public static class IntegerMessageDecoder extends MessageDecoderAdapter {
          public MessageDecoderResult decodable(IoSession session, IoBuffer in) {
              return Character.isDigit((char) in.get()) ? MessageDecoderResult.OK : MessageDecoderResult.NOT_OK;
          }
  
          public MessageDecoderResult decode(IoSession session, IoBuffer in, ProtocolDecoderOutput out) throws Exception {
              out.write(Integer.parseInt("" + (char) in.get()));
              return MessageDecoderResult.OK;
          }
      }
  
      private static class SessionStub extends DummySession {
          public SessionStub(boolean fragmented) {
              setTransportMetadata(new DefaultTransportMetadata("nio", "socket", false, fragmented,
                      InetSocketAddress.class, SocketSessionConfig.class, IoBuffer.class, FileRegion.class));
          }
      }
  
      /**
       * Test a decoding with fragmentation
       * @throws Exception If the test failed
      */
     @Test
     public void testFragmentedTransport() throws Exception {
         doTest(new SessionStub(true));
     }
 
     /**
      * Test a decoding without fragmentation
      * @throws Exception If the test failed
      */
     @Test
     public void testNonFragmentedTransport() throws Exception {
         doTest(new SessionStub(false));
     }
 }