/*
 * Copyright 2008 the original author or authors.
 * 
 * Licensed 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.openehealth.ipf.platform.camel.core.adapter;

import org.apache.camel.AggregationStrategy;
import org.apache.camel.Exchange;
import org.apache.camel.impl.DefaultCamelContext;
import org.apache.camel.support.DefaultExchange;
import org.junit.jupiter.api.Test;
import org.openehealth.ipf.platform.camel.core.support.transform.min.TestAggregator;

import static org.apache.camel.builder.Builder.body;
import static org.junit.jupiter.api.Assertions.assertEquals;

/**
 * @author Martin Krasser
 */
public class AggregatorAdapterTest {

    private AggregationStrategy strategy;

    @Test
    public void testAggregateDefault() {
        strategy = new AggregatorAdapter(new TestAggregator());
        var a = exchangeWithInBody("a");
        var b = exchangeWithInBody("b");
        strategy.aggregate(a, b);
        assertEquals("a:b", a.getIn().getBody());
    }
    
    @Test
    public void testAggregateCustom() {
        strategy = new AggregatorAdapter(new TestAggregator()).aggregationInput(body());
        var a = exchangeWithInBody("a");
        var b = exchangeWithOutBody("b");
        strategy.aggregate(a, b);
        assertEquals("a:b", a.getIn().getBody());
    }
    
    private static Exchange exchangeWithInBody(Object body) {
        var exchange = exchange();
        exchange.getIn().setBody(body);
        return exchange;
    }
    
    private static Exchange exchangeWithOutBody(Object body) {
        var exchange = exchange();
        exchange.getMessage().setBody(body);
        return exchange;
    }
    
    private static Exchange exchange() {
        return new DefaultExchange(new DefaultCamelContext());
    }
}
