/**
* Licensed to the Austrian Association for Software Tool Integration (AASTI)
* under one or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information regarding copyright
* ownership. The AASTI 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.openengsb.core.services.internal;
import static org.hamcrest.CoreMatchers.is;
import static org.junit.Assert.assertThat;
import static org.junit.Assert.fail;
import static org.mockito.Matchers.anyInt;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.verifyZeroInteractions;
import static org.mockito.Mockito.when;
import java.util.Arrays;
import java.util.Dictionary;
import java.util.Hashtable;
import java.util.Map;
import org.junit.Before;
import org.junit.Test;
import org.openengsb.core.api.remote.CustomJsonMarshaller;
import org.openengsb.core.api.remote.MethodCall;
import org.openengsb.core.api.remote.MethodResult;
import org.openengsb.core.api.remote.RequestHandler;
import org.openengsb.core.api.remote.UseCustomJasonMarshaller;
import org.openengsb.core.test.AbstractOsgiMockServiceTest;
import org.openengsb.core.util.DefaultOsgiUtilsService;
import org.osgi.framework.Constants;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.google.common.base.Preconditions;
import com.google.common.collect.ImmutableMap;
public class RequestHandlerImplTest extends AbstractOsgiMockServiceTest {
private RequestHandler requestHandler;
public interface TestInterface {
Integer test(Integer arg);
}
public static class CustomMarshaller implements CustomJsonMarshaller<Integer> {
@Override
public Integer transformArg(Object arg) {
Preconditions.checkArgument(arg instanceof Integer);
return (Integer) arg + 1;
}
}
@Before
public void setup() throws Exception {
RequestHandlerImpl requestHandlerImpl = new RequestHandlerImpl();
requestHandlerImpl.setUtilsService(new DefaultOsgiUtilsService(bundleContext));
requestHandler = requestHandlerImpl;
}
private TestInterface mockServiceWithProps(Map<String, Object> propData) {
TestInterface mockService = mock(TestInterface.class);
Dictionary<String, Object> props = new Hashtable<String, Object>(propData);
registerService(mockService, props, TestInterface.class);
return mockService;
}
private TestInterface registerServiceWithProps(TestInterface mockService, Map<String, Object> propData) {
Dictionary<String, Object> props = new Hashtable<String, Object>(propData);
registerService(mockService, props, TestInterface.class);
return mockService;
}
@Test
public void testMethodCallWithNullParameters_shouldBeSerialized() throws Exception {
Object[] args = { 1, null, 2 };
String[] classes = { "java.lang.Integer", "java.lang.Integer", "java.lang.Integer" };
MethodCall methodCall = new MethodCall("test", args, Arrays.asList(classes));
ObjectMapper objectMapper = new ObjectMapper();
String writeValueAsString = objectMapper.writerWithDefaultPrettyPrinter().writeValueAsString(methodCall);
MethodCall readValue = objectMapper.readValue(writeValueAsString, MethodCall.class);
assertThat(readValue, is(methodCall));
}
@Test
public void testCallByServiceId_shouldCallService() throws Exception {
TestInterface mockService = mockService(TestInterface.class, "testid");
when(mockService.test(anyInt())).thenReturn(21);
Map<String, String> metaData = ImmutableMap.of("serviceId", "testid");
MethodCall c = new MethodCall("test", new Object[]{ 42 }, metaData);
MethodResult result = requestHandler.handleCall(c);
verify(mockService).test(42);
assertThat(result.getClassName(), is(Integer.class.getName()));
assertThat((Integer) result.getArg(), is(21));
}
@Test
public void testCallByFilter_shouldCallService() throws Exception {
Map<String, Object> propData = ImmutableMap.of("testprop", (Object) new String[]{ "bla", "bleh", });
TestInterface mockService = mockServiceWithProps(propData);
when(mockService.test(anyInt())).thenReturn(21);
Map<String, String> metaData = ImmutableMap.of("serviceFilter", "(testprop=bla)");
MethodCall c = new MethodCall("test", new Object[]{ 42 }, metaData);
MethodResult result = requestHandler.handleCall(c);
verify(mockService).test(42);
assertThat(result.getClassName(), is(Integer.class.getName()));
assertThat((Integer) result.getArg(), is(21));
}
@Test
public void testCallByFilterAndServiceId_shouldCallService() throws Exception {
Map<String, Object> propData =
ImmutableMap.of("testprop", (Object) new String[]{ "bla", "bleh", }, Constants.SERVICE_PID, "xxx");
TestInterface mockService = mockServiceWithProps(propData);
when(mockService.test(anyInt())).thenReturn(21);
Map<String, String> metaData = ImmutableMap.of("serviceFilter", "(testprop=bla)", "serviceId", "xxx");
MethodCall c = new MethodCall("test", new Object[]{ 42 }, metaData);
MethodResult result = requestHandler.handleCall(c);
verify(mockService).test(42);
assertThat(result.getClassName(), is(Integer.class.getName()));
assertThat((Integer) result.getArg(), is(21));
}
@Test
public void testCallByFilterAndServiceIdWithCustomMarshaller_shouldCallService() throws Exception {
Map<String, Object> propData =
ImmutableMap.of("testprop", (Object) new String[]{ "blub", "bleh", }, Constants.SERVICE_PID, "xxx");
TestInterface realObject = new TestInterface() {
@Override
public Integer test(@UseCustomJasonMarshaller(CustomMarshaller.class) Integer arg) {
return arg;
}
};
registerServiceWithProps(realObject, propData);
Map<String, String> metaData = ImmutableMap.of("serviceFilter", "(testprop=blub)", "serviceId", "xxx");
MethodCall c = new MethodCall("test", new Object[]{ 42 }, metaData);
MethodResult result = requestHandler.handleCall(c);
assertThat(result.getClassName(), is(Integer.class.getName()));
assertThat((Integer) result.getArg(), is(43));
}
@Test
public void testCallButBothAttrsNull_shouldThrowException() throws Exception {
Map<String, Object> propData =
ImmutableMap.of("testprop", (Object) new String[]{ "bla", "bleh", });
TestInterface mockService = mockServiceWithProps(propData);
MethodCall c = new MethodCall("test", new Object[]{ 42 });
try {
requestHandler.handleCall(c);
fail("IllegalArgumentException expected");
} catch (IllegalArgumentException e) {
verifyZeroInteractions(mockService);
}
}
}