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.language.datasonnet;
  
  import java.io.InputStream;
  import java.nio.charset.Charset;
  import java.text.SimpleDateFormat;
  import java.util.Arrays;
  import java.util.TimeZone;
  
  import com.datasonnet.document.Document;
  import org.apache.camel.Exchange;
  import org.apache.camel.component.mock.MockEndpoint;
  import org.apache.camel.support.ExchangeHelper;
  import org.apache.camel.test.spring.junit5.CamelSpringTestSupport;
  import org.apache.commons.io.IOUtils;
  import org.junit.jupiter.api.Test;
  import org.skyscreamer.jsonassert.JSONAssert;
  import org.springframework.context.support.AbstractApplicationContext;
  import org.springframework.context.support.ClassPathXmlApplicationContext;
  
  import static org.junit.jupiter.api.Assertions.assertEquals;
  
  public class CamelDatasonnetTest extends CamelSpringTestSupport {
      private MockEndpoint mock;
  
      @Override
      protected AbstractApplicationContext createApplicationContext() {
          return new ClassPathXmlApplicationContext("org/apache/camel/language.datasonnet/camel-context.xml");
      }
  
      @Test
      public void testTransform() throws Exception {
          runCamelTest(loadResourceAsString("simpleMapping_payload.json"),
                  loadResourceAsString("simpleMapping_result.json"),
                  "direct:basicTransform");
      }
  
      @Test
      public void testTransformXML() throws Exception {
          runCamelTest(loadResourceAsString("payload.xml"),
                  loadResourceAsString("readXMLExtTest.json"),
                  "direct:transformXML");
      }
  
      @Test
      public void testTransformCSV() throws Exception {
          runCamelTest(loadResourceAsString("payload.csv"),
                  "{\"account\":\"123\"}",
                  "direct:transformCSV");
      }
  
      @Test
      public void testDatasonnetScript() throws Exception {
          runCamelTest(loadResourceAsString("simpleMapping_payload.json"),
                  loadResourceAsString("simpleMapping_result.json"),
                  "direct:datasonnetScript");
      }
  
      @Test
      public void testNamedImports() throws Exception {
          runCamelTest("{}",
                  loadResourceAsString("namedImports_result.json"),
                  "direct:namedImports");
      }
  
      @Test
      public void testExpressionLanguage() throws Exception {
          runCamelTest("World",
                  "{ \"test\":\"Hello, World\"}",
                  "direct:expressionLanguage");
      }
  
      @Test
      public void testVariable() throws Exception {
          runCamelTest("World",
                  "{ \"test\":\"Hello, World\"}",
                  "direct:variable");
      }
  
      @Test
      public void testNullInput() throws Exception {
          runCamelTest("",
                  "{ \"test\":\"Hello, World\"}",
                  "direct:nullInput");
          runCamelTest(null,
                  "{ \"test\":\"Hello, World\"}",
                  "direct:nullInput");
      }
  
      @Test
      public void testRegistryLibraries() throws Exception {
          runCamelTest("",
                  "{ \"test\":\"Hello, World\"}",
                  "direct:registryLibraries");
      }
  
      @Test
      public void testReadJava() throws Exception {
          Gizmo theGizmo = new Gizmo();
          theGizmo.setName("gizmo");
          theGizmo.setQuantity(123);
          theGizmo.setInStock(true);
          theGizmo.setColors(Arrays.asList("red", "white", "blue"));
  
          Manufacturer manufacturer = new Manufacturer();
          manufacturer.setManufacturerName("ACME Corp.");
          manufacturer.setManufacturerCode("ACME123");
          theGizmo.setManufacturer(manufacturer);
  
          SimpleDateFormat df = new SimpleDateFormat("yyyy-MM-dd");
          df.setTimeZone(TimeZone.getTimeZone("UTC"));
          theGizmo.setDate(df.parse("2020-01-06"));
  
          runCamelTest(theGizmo,
                  loadResourceAsString("javaTest.json"),
                  "direct:readJava");
      }
  
      @Test
      public void testReadJavaDatasonnetHeader() throws Exception {
          Gizmo theGizmo = new Gizmo();
          theGizmo.setName("gizmo");
          theGizmo.setQuantity(123);
          theGizmo.setInStock(true);
          theGizmo.setColors(Arrays.asList("red", "white", "blue"));
  
          Manufacturer manufacturer = new Manufacturer();
          manufacturer.setManufacturerName("ACME Corp.");
          manufacturer.setManufacturerCode("ACME123");
          theGizmo.setManufacturer(manufacturer);
  
          SimpleDateFormat df = new SimpleDateFormat("yyyy-MM-dd");
          df.setTimeZone(TimeZone.getTimeZone("UTC"));
          theGizmo.setDate(df.parse("2020-01-06"));
  
          runCamelTest(theGizmo,
                  loadResourceAsString("javaTest.json"),
                  "direct:readJavaDatasonnetHeader");
      }
  
      @Test
      public void testWriteJava() throws Exception {
          Gizmo theGizmo = new Gizmo();
          theGizmo.setName("gizmo");
          theGizmo.setQuantity(123);
          theGizmo.setInStock(true);
          theGizmo.setColors(Arrays.asList("red", "white", "blue"));
  
          Manufacturer manufacturer = new Manufacturer();
          manufacturer.setManufacturerName("ACME Corp.");
          manufacturer.setManufacturerCode("ACME123");
          theGizmo.setManufacturer(manufacturer);
  
          SimpleDateFormat df = new SimpleDateFormat("yyyy-MM-dd");
          df.setTimeZone(TimeZone.getTimeZone("UTC"));
          theGizmo.setDate(df.parse("2020-01-06"));
  
          String payload = loadResourceAsString("javaTest.json");
  
          template.sendBody("direct:writeJava", payload);
          mock = getMockEndpoint("mock:direct:end");
          Exchange exchange = mock.assertExchangeReceived(mock.getReceivedCounter() - 1);
          Object response = exchange.getIn().getBody();
  
          assertEquals(theGizmo, response);
      }
  
      private void runCamelTest(Object payload, String expectedJson, String uri) throws Exception {
          template.sendBody(uri, payload);
          mock = getMockEndpoint("mock:direct:end");
          Exchange exchange = mock.assertExchangeReceived(mock.getReceivedCounter() - 1);
          Object body = exchange.getMessage().getBody();
          String response;
          if (body instanceof Document) {
              response = ExchangeHelper.convertToMandatoryType(exchange, String.class, ((Document<?>) body).getContent());
          } else {
              response = exchange.getMessage().getBody(String.class);
  
          }
          JSONAssert.assertEquals(expectedJson, response, true);
      }
  
      private String loadResourceAsString(String name) throws Exception {
          InputStream is = getClass().getClassLoader().getResourceAsStream(name);
          return IOUtils.toString(is, Charset.defaultCharset());
      }
  }
  