org.mule.tck.junit4.DomainFunctionalTestCase Maven / Gradle / Ivy

Go to download
/*
 * 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.tck.junit4;

import org.mule.api.MuleContext;
import org.mule.tck.probe.PollingProber;
import org.mule.tck.probe.Probe;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import org.junit.After;
import org.junit.Before;

public abstract class DomainFunctionalTestCase extends AbstractMuleTestCase
{

    private final Map muleContexts = new HashMap();
    private final List disposedContexts = new ArrayList();
    private MuleContext domainContext;

    protected abstract String getDomainConfig();

    public synchronized void disposeMuleContext(final MuleContext muleContext)
    {
        disposedContexts.add(muleContext);
        muleContext.dispose();
        new PollingProber(10000, 100).check(new Probe()
        {
            @Override
            public boolean isSatisfied()
            {
                return muleContext.isDisposed();
            }

            @Override
            public String describeFailure()
            {
                return "mule context timeout during dispose";
            }
        });
    }

    @Before
    public void setUpMuleContexts() throws Exception
    {
        domainContext = new DomainContextBuilder().setDomainConfig(getDomainConfig()).build();
        ApplicationConfig[] applicationConfigs = getConfigResources();
        for (ApplicationConfig applicationConfig : applicationConfigs)
        {
            MuleContext muleContext = createAppMuleContext(applicationConfig.applicationResources);
            muleContexts.put(applicationConfig.applicationName, muleContext);
        }
    }

    @After
    public void disposeMuleContexts()
    {
        for (MuleContext muleContext : muleContexts.values())
        {
            try
            {
                disposeMuleContext(muleContext);
            }
            catch (Exception e)
            {
                //Nothing to do
            }
        }
        muleContexts.clear();
        disposeMuleContext(domainContext);
    }

    protected MuleContext createAppMuleContext(String[] configResource) throws Exception
    {
        return new ApplicationContextBuilder().setDomainContext(domainContext).setApplicationResources(configResource).build();
    }

    public abstract ApplicationConfig[] getConfigResources();

    public MuleContext getMuleContextForApp(String applicationName)
    {
        return muleContexts.get(applicationName);
    }

    public MuleContext getMuleContextForDomain()
    {
        return domainContext;
    }

    public class ApplicationConfig
    {

        String applicationName;
        String[] applicationResources;

        public ApplicationConfig(String applicationName, String[] applicationResources)
        {
            this.applicationName = applicationName;
            this.applicationResources = applicationResources;
        }
    }

}