|
1
3 package org.jmock.core;
4
5 import java.util.ArrayList  ;
6 import java.util.Iterator ;
7 import java.util.ListIterator ;
8 import org.jmock.core.stub.TestFailureStub;
9
10
11 public class FIFOInvocationDispatcher
12 implements InvocationDispatcher
13 {
14 public static final String NO_EXPECTATIONS_MESSAGE = "No expectations set";
15
16 private ArrayList invokables = new ArrayList ();
17 private Stub defaultStub = new TestFailureStub("no match found");
18
19 public Object dispatch( Invocation invocation ) throws Throwable {
20 ListIterator i = invokables.listIterator();
21 while (i.hasNext()) {
22 Invokable invokable = (Invokable)i.next();
23 if (invokable.matches(invocation)) {
24 return invokable.invoke(invocation);
25 }
26 }
27
28 return defaultStub.invoke(invocation);
29 }
30
31 public void setDefaultStub( Stub defaultStub ) {
32 this.defaultStub = defaultStub;
33 }
34
35 public void add( Invokable invokable ) {
36 invokables.add(invokable);
37 }
38
39 public void verify() {
40 Iterator i = invokables.iterator();
41 while (i.hasNext()) {
42 ((Verifiable)i.next()).verify();
43 }
44 }
45
46 public void clear() {
47 invokables.clear();
48 }
49
50 public StringBuffer describeTo( StringBuffer buffer ) {
51 if (anyInvokableHasDescription()) {
52 writeInvokablesTo(buffer);
53 } else {
54 buffer.append(NO_EXPECTATIONS_MESSAGE);
55 }
56
57 return buffer;
58 }
59
60 private void writeInvokablesTo( StringBuffer buffer ) {
61 Iterator iterator = invokables.iterator();
62 while (iterator.hasNext()) {
63 Invokable invokable = (Invokable)iterator.next();
64 if (invokable.hasDescription()) {
65 invokable.describeTo(buffer).append("\n");
66 }
67 }
68 }
69
70 private boolean anyInvokableHasDescription() {
71 Iterator iterator = invokables.iterator();
72 while (iterator.hasNext()) {
73 if (((Invokable)iterator.next()).hasDescription()) return true;
74 }
75 return false;
76 }
77 }
78 |
Popular Tags |
Java API By Example, From Geeks To Geeks.