J2EE中使用Spring AOP框架和EJB组件(5)

public class ForeignTradeAdviceTest extends TestCase { 
TradeManager tradeManager; 
private Mock baseTradeManagerMock; 
private Mock yahooFeedMock; 
protected void setUp() throws Exception { 
super.setUp(); 
baseTradeManagerMock = new Mock(TradeManager.class, "baseTradeManager"); 
TradeManager baseTradeManager = (TradeManager) baseTradeManagerMock.proxy(); 

yahooFeedMock = new Mock(TradeManager.class, "yahooFeed"); 
TradeManager yahooFeed = (TradeManager) yahooFeedMock.proxy(); 
Map stubs = new HashMap(); 
stubs.put("yahooFeed", yahooFeed); 
stubs.put("baseTradeManager", baseTradeManager); 

ConfigurableApplicationContext ctx = new ClassPathXmlApplicationContext(CTX_NAME); 
ctx.getBeanFactory().addBeanPostProcessor(new StubPostProcessor(stubs)); 
tradeManager = (TradeManager) proxyFactory.getProxy(); 
} 
...  

public void testAdvice() throws Throwable { 
String symbol = "testSymbol"; 
BigDecimal expectedPrice = new BigDecimal("0.222"); 
baseTradeManagerMock.expects(new InvokeOnceMatcher()).method("getPrice") 
.with(new IsEqual(symbol)).will(new ReturnStub(null)); 

yahooFeedMock.expects(new InvokeOnceMatcher()).method("getPrice") 
.with(new IsEqual(symbol)).will(new ReturnStub(expectedPrice)); 

BigDecimal price = tradeManager.getPrice(symbol); 
assertEquals("Invalid price", expectedPrice, price); 
baseTradeManagerMock.verify(); 
yahooFeedMock.verify(); 
} 

...    
public static TestSuite suite() { 
BigDecimal v1 = new BigDecimal("0.22"); 
BigDecimal v2 = new BigDecimal("0.33"); 

RuntimeException e1 = new RuntimeException("e1"); 
RuntimeException e2 = new RuntimeException("e2"); 

TestSuite suite = new TestSuite(ForeignTradeAdviceTest.class.getName()); 
suite.addTest(new ForeignTradeAdviceTest
(true, v1,   null, false, null, null, v1,   null)); 
suite.addTest(new ForeignTradeAdviceTest
(true, null, e1,   false, null, null, null, e1)); 
suite.addTest(new ForeignTradeAdviceTest
(true, null, null, true,  v2,   null, v2,   null)); 
suite.addTest(new ForeignTradeAdviceTest
(true, null, null, true,  null, null, null, null)); 
suite.addTest(new ForeignTradeAdviceTest
(true, null, null, true,  null, e2,   null, e2)); 
return suite; 
} 

public ForeignTradeAdviceTest( 
boolean baseCall, BigDecimal baseValue, Throwable baseException, 
boolean yahooCall, BigDecimal yahooValue, Throwable yahooException, 
BigDecimal expectedValue, Throwable expectedException) { 
super("test"); 
this.baseCall = baseCall; 
this.baseWill = baseException==null ?  
(Stub) new ReturnStub(baseValue) : new ThrowStub(baseException); 
this.yahooCall = yahooCall; 
this.yahooWill = yahooException==null ?  
(Stub) new ReturnStub(yahooValue) : new ThrowStub(yahooException); 
this.expectedValue = expectedValue; 
this.expectedException = expectedException; 
} 

public void test() throws Throwable { 
String symbol = "testSymbol"; 
if(baseCall) { 
baseTradeManagerMock.expects(new InvokeOnceMatcher()) 
.method("getPrice").with(new IsEqual(symbol)).will(baseWill); 
} 

if(yahooCall) { 
yahooFeedMock.expects(new InvokeOnceMatcher()) 
.method("getPrice").with(new IsEqual(symbol)).will(yahooWill); 
} 

try { 
BigDecimal price = tradeManager.getPrice(symbol); 
assertEquals("Invalid price", expectedValue, price); 
} catch(Exception e) { 
if(expectedException==null) { 
throw e; 
} 
} 
baseTradeManagerMock.verify(); 
yahooFeedMock.verify(); 
} 
public String getName() { 
return super.getName()+" "+ 
baseCalled+" "+baseValue+" "+baseException+" "+ 
yahooCalled+" "+yahooValue+" "+yahooException+" "+ 
expectedValue+" "+expectedException; 
} 
...