1   /*
2    *  Copyright 2006 Simon Raess
3    *
4    *  Licensed under the Apache License, Version 2.0 (the "License");
5    *  you may not use this file except in compliance with the License.
6    *  You may obtain a copy of the License at
7    *
8    *      http://www.apache.org/licenses/LICENSE-2.0
9    * 
10   *  Unless required by applicable law or agreed to in writing, software
11   *  distributed under the License is distributed on an "AS IS" BASIS,
12   *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13   *  See the License for the specific language governing permissions and
14   *  limitations under the License.
15   */
16  package net.sf.beep4j.internal;
17  
18  import org.jmock.core.Invocation;
19  import org.jmock.core.Stub;
20  
21  public class ParameterCaptureStub<T> implements Stub {
22  	private final int index;
23  	private final Class<? extends T> type;
24  	private final Stub target;
25  	private T parameter;
26  	public ParameterCaptureStub(int index, Class<? extends T> type, Stub target) {
27  		this.index = index;
28  		this.type = type;
29  		this.target = target;
30  	}
31  	public T getParameter() {
32  		return parameter;
33  	}
34  	public StringBuffer describeTo(StringBuffer buf) {
35  		buf.append("stub[capture parameter " + index + "]");
36  		return buf;
37  	}
38  	public Object invoke(Invocation invocation) throws Throwable {
39  		parameter = type.cast(invocation.parameterValues.get(index));
40  		if (target != null) {
41  			return target.invoke(invocation);
42  		} else {
43  			return null;
44  		}
45  	}
46  }