/*
* #%L
* Talend :: ESB :: Job :: Controller
* %%
* Copyright (C) 2011 - 2012 Talend Inc.
* %%
* 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.
* #L%
*/
package org.talend.esb.job.controller.internal;
import java.util.Dictionary;
import java.util.Hashtable;
import java.util.concurrent.ExecutorService;
import org.easymock.EasyMockSupport;
import org.junit.Before;
import org.junit.Test;
import org.osgi.framework.BundleContext;
import org.osgi.framework.Constants;
import org.osgi.framework.ServiceRegistration;
import static org.easymock.EasyMock.anyObject;
import static org.easymock.EasyMock.eq;
import static org.easymock.EasyMock.expect;
import static org.easymock.EasyMock.isA;
import routines.system.api.ESBEndpointInfo;
import routines.system.api.ESBEndpointRegistry;
import routines.system.api.TalendESBJob;
import routines.system.api.TalendESBRoute;
import routines.system.api.TalendJob;
public class JobLauncherListenerTest extends EasyMockSupport {
public static final String NAME = "name";
public static final String MANAGED_SERVICE_NAME = "org.osgi.service.cm.ManagedService";
public static final String[] EMPTY_STRING_ARR = new String[0];
@SuppressWarnings("serial")
public static final Dictionary<?, ?> PROPERTIES = new Hashtable<String, String>() {
{
put(Constants.SERVICE_PID, NAME);
}
};
private BundleContext context;
private ServiceRegistration sr;
private ExecutorService execService;
private JobLauncherImpl jobLauncher;
@Before
public void setUp() {
context = createMock(BundleContext.class);
execService = createMock(ExecutorService.class);
sr = createNiceMock(ServiceRegistration.class);
jobLauncher = new JobLauncherImpl();
jobLauncher.setBundleContext(context);
jobLauncher.setExecutorService(execService);
}
@Test
public void routeAddedManagedServiceRegistered() throws Exception {
TalendESBRoute route = createMock(TalendESBRoute.class);
expectManagedJobStarting(RouteAdapter.class);
replayAll();
jobLauncher.routeAdded(route, NAME);
verifyAll();
}
private void expectManagedJobStarting(Class<? extends JobTask> taskClass) {
expect(context.registerService(
eq(MANAGED_SERVICE_NAME), isA(taskClass), eq(PROPERTIES))).andReturn(sr);
execService.execute(isA(taskClass));
}
@Test
public void managedServiceForRouteUnregisteredIfRegisteredBefore() throws Exception {
TalendESBRoute route = createNiceMock(TalendESBRoute.class);
sr.unregister();
expectManagedJobStarting(RouteAdapter.class);
replayAll();
jobLauncher.routeAdded(route, NAME);
jobLauncher.routeRemoved(route, NAME);
verifyAll();
}
@Test
public void jobAddedManagedServiceRegistered() throws Exception {
TalendJob job = createNiceMock(TalendJob.class);
expectManagedJobStarting(SimpleJobTask.class);
replayAll();
jobLauncher.jobAdded(job, NAME);
verifyAll();
}
@Test
public void managedServiceForJobUnregisteredIfRegisteredBefore() throws Exception {
TalendJob job = createNiceMock(TalendJob.class);
sr.unregister();
expectManagedJobStarting(SimpleJobTask.class);
replayAll();
jobLauncher.jobAdded(job, NAME);
jobLauncher.jobRemoved(job, NAME);
verifyAll();
}
@Test
public void esbJobAddedEndpointRegistrySpecifiedForJob() {
ESBEndpointInfo endpointInfo = createNiceMock(ESBEndpointInfo.class);
TalendESBJob esbJob = createMock(TalendESBJob.class);
esbJob.setEndpointRegistry((ESBEndpointRegistry) anyObject());
expect(esbJob.getEndpoint()).andReturn(endpointInfo);
replayAll();
jobLauncher.esbJobAdded(esbJob, NAME);
verifyAll();
}
@Test
public void esbConsumerOnlyJobAddedJobStartedImmediatlyAsSimpleJob() {
TalendESBJob esbJob = createMock(TalendESBJob.class);
esbJob.setEndpointRegistry((ESBEndpointRegistry) anyObject());
expect(esbJob.getEndpoint()).andReturn(null);
expectManagedJobStarting(SimpleJobTask.class);
replayAll();
jobLauncher.esbJobAdded(esbJob, NAME);
verifyAll();
}
@Test
public void esbConsumerOnlyJobUnregisters() {
TalendESBJob esbJob = createMock(TalendESBJob.class);
sr.unregister();
esbJob.setEndpointRegistry((ESBEndpointRegistry) anyObject());
expect(esbJob.getEndpoint()).andReturn(null).times(2);
expectManagedJobStarting(SimpleJobTask.class);
replayAll();
jobLauncher.esbJobAdded(esbJob, NAME);
jobLauncher.esbJobRemoved(esbJob, NAME);
verifyAll();
}
}