TransformationGraphLookupStrategyTestCase.java example

Explorer
MULE-master
- mule-mule-4.x
/*
 * Copyright (c) MuleSoft, Inc.  All rights reserved.  http://www.mulesoft.com
 * The software in this package is published under the terms of the CPAL v1.0
 * license, a copy of which has been included with this distribution in the
 * LICENSE.txt file.
 */
package org.mule.runtime.core.transformer.graph;

import static junit.framework.Assert.assertEquals;
import static junit.framework.Assert.fail;
import static org.mockito.Mockito.mock;

import org.mule.runtime.core.api.transformer.Converter;
import org.mule.runtime.api.metadata.DataType;
import org.mule.tck.junit4.AbstractMuleTestCase;
import org.mule.tck.size.SmallTest;
import org.mule.runtime.core.transformer.CompositeConverter;
import org.mule.runtime.core.transformer.builder.MockConverterBuilder;

import java.util.List;

import org.junit.Assert;
import org.junit.Test;

@SmallTest
public class TransformationGraphLookupStrategyTestCase extends AbstractMuleTestCase {

  private static final DataType XML_DATA_TYPE = mock(DataType.class, "XML_DATA_TYPE");
  private static final DataType JSON_DATA_TYPE = mock(DataType.class, "JSON_DATA_TYPE");
  private static final DataType INPUT_STREAM_DATA_TYPE = mock(DataType.class, "INPUT_STREAM_DATA_TYPE");
  private static final DataType STRING_DATA_TYPE = mock(DataType.class, "STRING_DATA_TYPE");

  private TransformationGraph graph = new TransformationGraph();
  private TransformationGraphLookupStrategy lookupStrategyTransformation = new TransformationGraphLookupStrategy(graph);

  @Test
  public void lookupTransformersNoSourceInGraph() throws Exception {
    Converter inputStreamToXml = new MockConverterBuilder().from(INPUT_STREAM_DATA_TYPE).to(XML_DATA_TYPE).build();
    graph.addConverter(inputStreamToXml);

    List<Converter> converters = lookupStrategyTransformation.lookupConverters(JSON_DATA_TYPE, INPUT_STREAM_DATA_TYPE);

    assertEquals(0, converters.size());
  }

  @Test
  public void lookupTransformersNoTargetInGraph() throws Exception {
    Converter inputStreamToXml = new MockConverterBuilder().from(INPUT_STREAM_DATA_TYPE).to(XML_DATA_TYPE).build();
    graph.addConverter(inputStreamToXml);

    List<Converter> converters = lookupStrategyTransformation.lookupConverters(INPUT_STREAM_DATA_TYPE, JSON_DATA_TYPE);

    assertEquals(0, converters.size());
  }

  @Test
  public void findsDirectTransformation() throws Exception {
    Converter inputStreamToXml = new MockConverterBuilder().from(INPUT_STREAM_DATA_TYPE).to(XML_DATA_TYPE).build();
    graph.addConverter(inputStreamToXml);

    List<Converter> converters = lookupStrategyTransformation.lookupConverters(INPUT_STREAM_DATA_TYPE, XML_DATA_TYPE);

    assertEquals(1, converters.size());
    assertEquals(inputStreamToXml, converters.get(0));
  }

  @Test
  public void findsMultipleDirectTransformations() throws Exception {
    Converter inputStreamToXml = new MockConverterBuilder().from(INPUT_STREAM_DATA_TYPE).to(XML_DATA_TYPE).build();
    graph.addConverter(inputStreamToXml);
    Converter betterInputStreamToXml = new MockConverterBuilder().from(INPUT_STREAM_DATA_TYPE).to(XML_DATA_TYPE).build();
    graph.addConverter(betterInputStreamToXml);

    List<Converter> converters = lookupStrategyTransformation.lookupConverters(INPUT_STREAM_DATA_TYPE, XML_DATA_TYPE);

    assertEquals(2, converters.size());
    Assert.assertTrue(converters.contains(inputStreamToXml));
    Assert.assertTrue(converters.contains(betterInputStreamToXml));
  }

  @Test
  public void findsCompositeConverter() throws Exception {
    Converter inputStreamToString =
        new MockConverterBuilder().named("inputStreamToString").from(INPUT_STREAM_DATA_TYPE).to(STRING_DATA_TYPE).build();
    graph.addConverter(inputStreamToString);
    Converter stringToJson = new MockConverterBuilder().named("stringToJson").from(STRING_DATA_TYPE).to(JSON_DATA_TYPE).build();
    graph.addConverter(stringToJson);

    List<Converter> converters = lookupStrategyTransformation.lookupConverters(INPUT_STREAM_DATA_TYPE, JSON_DATA_TYPE);

    assertEquals(1, converters.size());
    assertContainsCompositeTransformer(converters, inputStreamToString, stringToJson);
  }

  @Test
  public void findsMultipleCompositeConvertersWithMultipleEdgesFromSource() throws Exception {
    Converter inputStreamToString =
        new MockConverterBuilder().named("inputStreamToString").from(INPUT_STREAM_DATA_TYPE).to(STRING_DATA_TYPE).build();
    graph.addConverter(inputStreamToString);
    Converter inputStreamToJson =
        new MockConverterBuilder().named("inputStreamToJson").from(INPUT_STREAM_DATA_TYPE).to(JSON_DATA_TYPE).build();
    graph.addConverter(inputStreamToJson);
    Converter jsonToXml = new MockConverterBuilder().named("jsonToXml").from(JSON_DATA_TYPE).to(XML_DATA_TYPE).build();
    graph.addConverter(jsonToXml);
    Converter jsonToString = new MockConverterBuilder().named("jsonToString").from(JSON_DATA_TYPE).to(STRING_DATA_TYPE).build();
    graph.addConverter(jsonToString);
    Converter stringToXml = new MockConverterBuilder().named("stringToXml").from(STRING_DATA_TYPE).to(XML_DATA_TYPE).build();
    graph.addConverter(stringToXml);
    Converter stringToJson = new MockConverterBuilder().named("stringToJson").from(STRING_DATA_TYPE).to(JSON_DATA_TYPE).build();
    graph.addConverter(stringToJson);

    List<Converter> converters = lookupStrategyTransformation.lookupConverters(INPUT_STREAM_DATA_TYPE, XML_DATA_TYPE);

    assertEquals(4, converters.size());
    assertContainsCompositeTransformer(converters, inputStreamToString, stringToXml);
    assertContainsCompositeTransformer(converters, inputStreamToJson, jsonToXml);
    assertContainsCompositeTransformer(converters, inputStreamToString, stringToJson, jsonToXml);
  }

  private void assertContainsCompositeTransformer(List<Converter> converters, Converter... composedConverters) {
    for (Converter converter : converters) {
      if (converter instanceof CompositeConverter) {
        CompositeConverter compositeConverter = (CompositeConverter) converter;
        if (compositeConverter.getConverters().size() != composedConverters.length) {
          continue;
        }

        boolean matches = true;
        for (int i = 0; i < composedConverters.length - 1; i++) {
          if (composedConverters[i] != compositeConverter.getConverters().get(i)) {
            matches = false;
            break;
          }
        }

        if (matches) {
          return;
        }
      }
    }

    fail("Converter list does not contain a composite converter with: " + composedConverters);
  }
}