/*
* Copyright (c) 2010-2016. Axon Framework
*
* 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.axonframework.commandhandling;
import org.axonframework.common.Registration;
import org.axonframework.messaging.GenericMessage;
import org.axonframework.messaging.MetaData;
import org.axonframework.messaging.annotation.ClasspathParameterResolverFactory;
import org.axonframework.messaging.annotation.ParameterResolverFactory;
import org.axonframework.messaging.unitofwork.CurrentUnitOfWork;
import org.axonframework.messaging.unitofwork.UnitOfWork;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import java.util.ArrayList;
import java.util.HashSet;
import java.util.LinkedList;
import java.util.Map;
import static org.junit.Assert.*;
import static org.mockito.Mockito.*;
/**
* @author Allard Buijze
*/
public class AnnotationCommandHandlerAdapterTest {
private AnnotationCommandHandlerAdapter testSubject;
private CommandBus mockBus;
private MyCommandHandler mockTarget;
private UnitOfWork<CommandMessage<?>> mockUnitOfWork;
private ParameterResolverFactory parameterResolverFactory;
@Before
@SuppressWarnings("unchecked")
public void setUp() {
mockBus = mock(CommandBus.class);
mockTarget = new MyCommandHandler();
parameterResolverFactory = ClasspathParameterResolverFactory.forClass(getClass());
testSubject = new AnnotationCommandHandlerAdapter(mockTarget, parameterResolverFactory);
mockUnitOfWork = mock(UnitOfWork.class);
when(mockUnitOfWork.resources()).thenReturn(mock(Map.class));
when(mockUnitOfWork.getCorrelationData()).thenReturn(MetaData.emptyInstance());
when(mockBus.subscribe(any(), any())).thenReturn(mock(Registration.class));
CurrentUnitOfWork.set(mockUnitOfWork);
}
@After
public void tearDown() {
CurrentUnitOfWork.clear(mockUnitOfWork);
}
@Test
public void testHandlerDispatching_VoidReturnType() throws Exception {
Object actualReturnValue = testSubject.handle(GenericCommandMessage.asCommandMessage(""));
assertEquals(null, actualReturnValue);
assertEquals(1, mockTarget.voidHandlerInvoked);
assertEquals(0, mockTarget.returningHandlerInvoked);
}
@Test
public void testHandlerDispatching_WithReturnType() throws Exception {
Object actualReturnValue = testSubject.handle(GenericCommandMessage.asCommandMessage(1L));
assertEquals(1L, actualReturnValue);
assertEquals(0, mockTarget.voidHandlerInvoked);
assertEquals(1, mockTarget.returningHandlerInvoked);
}
@Test
public void testHandlerDispatching_WithCustomCommandName() throws Exception {
Object actualReturnValue = testSubject.handle(new GenericCommandMessage<>(new GenericMessage<>(1L), "almostLong"));
assertEquals(1L, actualReturnValue);
assertEquals(0, mockTarget.voidHandlerInvoked);
assertEquals(0, mockTarget.returningHandlerInvoked);
assertEquals(1, mockTarget.almostDuplicateReturningHandlerInvoked);
}
@Test
public void testHandlerDispatching_ThrowingException() throws Exception {
try {
testSubject.handle(GenericCommandMessage.asCommandMessage(new HashSet()));
fail("Expected exception");
} catch (Exception ex) {
assertEquals(Exception.class, ex.getClass());
return;
}
fail("Shouldn't make it till here");
}
@Test
public void testSubscribe() {
testSubject.subscribe(mockBus);
verify(mockBus).subscribe(Long.class.getName(), testSubject);
verify(mockBus).subscribe(String.class.getName(), testSubject);
verify(mockBus).subscribe(HashSet.class.getName(), testSubject);
verify(mockBus).subscribe(ArrayList.class.getName(), testSubject);
verify(mockBus).subscribe("almostLong", testSubject);
verifyNoMoreInteractions(mockBus);
}
@Test(expected = NoHandlerForCommandException.class)
public void testHandle_NoHandlerForCommand() throws Exception {
testSubject.handle(GenericCommandMessage.asCommandMessage(new LinkedList<>()));
verify(mockUnitOfWork.resources(), never()).put(ParameterResolverFactory.class.getName(),
parameterResolverFactory);
}
private static class MyCommandHandler {
private int voidHandlerInvoked;
private int returningHandlerInvoked;
private int almostDuplicateReturningHandlerInvoked;
@SuppressWarnings({"UnusedDeclaration"})
@CommandHandler
public void myVoidHandler(String stringCommand, UnitOfWork<CommandMessage<?>> unitOfWork) {
voidHandlerInvoked++;
}
@CommandHandler(commandName = "almostLong")
public Long myAlmostDuplicateReturningHandler(Long longCommand, UnitOfWork<CommandMessage<?>> unitOfWork) {
assertNotNull("The UnitOfWork was not passed to the command handler", unitOfWork);
almostDuplicateReturningHandlerInvoked++;
return longCommand;
}
@CommandHandler
public Long myReturningHandler(Long longCommand, UnitOfWork<CommandMessage<?>> unitOfWork) {
assertNotNull("The UnitOfWork was not passed to the command handler", unitOfWork);
returningHandlerInvoked++;
return longCommand;
}
@SuppressWarnings({"UnusedDeclaration"})
@CommandHandler
public void exceptionThrowingHandler(HashSet o) throws Exception {
throw new Exception("Some exception");
}
@SuppressWarnings({"UnusedDeclaration"})
@CommandHandler
public void exceptionThrowingHandler(ArrayList o) throws Exception {
throw new RuntimeException("Some exception");
}
}
}