content: /*
   * 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.camel.component.mina;
  
  import java.nio.charset.StandardCharsets;
  
  import org.apache.camel.BindToRegistry;
  import org.apache.camel.ResolveEndpointFailedException;
  import org.apache.camel.builder.RouteBuilder;
  import org.apache.camel.component.mock.MockEndpoint;
  import org.apache.mina.core.buffer.IoBuffer;
  import org.apache.mina.core.session.IoSession;
  import org.apache.mina.filter.codec.CumulativeProtocolDecoder;
  import org.apache.mina.filter.codec.ProtocolCodecFactory;
  import org.apache.mina.filter.codec.ProtocolDecoder;
  import org.apache.mina.filter.codec.ProtocolDecoderOutput;
  import org.apache.mina.filter.codec.ProtocolEncoder;
  import org.apache.mina.filter.codec.ProtocolEncoderOutput;
  import org.junit.jupiter.api.Test;
  
  import static org.junit.jupiter.api.Assertions.assertEquals;
  import static org.junit.jupiter.api.Assertions.assertThrows;
  
  /**
   * Unit test with custom codec using the VM protocol.
   */
  public class MinaVMCustomCodecTest extends BaseMinaTest {
  
      @BindToRegistry("myCodec")
      private final MyCodec codec1 = new MyCodec();
  
      @Test
      public void testMyCodec() throws Exception {
          MockEndpoint mock = getMockEndpoint("mock:result");
          mock.expectedMessageCount(1);
          mock.expectedBodiesReceived("Bye World");
  
          Object out = template.requestBody(String.format("mina:vm://localhost:%1$s?sync=true&codec=#myCodec", getPort()),
                  "Hello World");
          assertEquals("Bye World", out);
  
          mock.assertIsSatisfied();
      }
  
      @Test
      public void testTCPEncodeUTF8InputIsString() throws Exception {
          final String myUri = String.format("mina:vm://localhost:%1$s?encoding=UTF-8&sync=false", getNextPort());
          context.addRoutes(new RouteBuilder() {
              public void configure() {
                  from(myUri).to("mock:result");
              }
          });
  
          MockEndpoint endpoint = getMockEndpoint("mock:result");
  
          // include a UTF-8 char in the text \u0E08 is a Thai elephant
          String body = "Hello Thai Elephant \u0E08";
  
          endpoint.expectedMessageCount(1);
          endpoint.expectedBodiesReceived(body);
  
          template.sendBody(myUri, body);
          MockEndpoint.assertIsSatisfied(context);
      }
  
      @Test
      public void testBadConfiguration() {
          final String format = String.format("mina:vm://localhost:%1$s?sync=true&codec=#XXX", getPort());
  
          assertThrows(ResolveEndpointFailedException.class, () -> template.sendBody(format, "Hello World"),
                  "Should have thrown a ResolveEndpointFailedException");
      }
  
      @Override
      protected RouteBuilder createRouteBuilder() {
          return new RouteBuilder() {
  
              public void configure() {
                  from(String.format("mina:vm://localhost:%1$s?sync=true&codec=#myCodec", getPort()))
                          .transform(constant("Bye World")).to("mock:result");
              }
          };
      }
  
      private static class MyCodec implements ProtocolCodecFactory {
  
          @Override
          public ProtocolEncoder getEncoder(IoSession is) {
              return new ProtocolEncoder() {
  
                  public void encode(IoSession ioSession, Object message, ProtocolEncoderOutput out) throws Exception {
                      IoBuffer bb = IoBuffer.allocate(32).setAutoExpand(true);
                      String s = (String) message;
                      bb.put(s.getBytes(StandardCharsets.US_ASCII));
                      bb.flip();
                      out.write(bb);
                  }
  
                  public void dispose(IoSession ioSession) {
                      // do nothing
                  }
              };
  
          }
  
          @Override
          public ProtocolDecoder getDecoder(IoSession is) {
              return new CumulativeProtocolDecoder() {
  
                  protected boolean doDecode(IoSession session, IoBuffer in, ProtocolDecoderOutput out) throws Exception {
                      if (in.remaining() > 0) {
                          byte[] buf = new byte[in.remaining()];
                          in.get(buf);
                          out.write(new String(buf, StandardCharsets.US_ASCII));
                          return true;
                      } else {
                          return false;
                      }
                  }
              };
          }
      }
  }
  