1 | |
|
2 | |
|
3 | |
|
4 | |
|
5 | |
|
6 | |
|
7 | |
|
8 | |
|
9 | |
|
10 | |
|
11 | |
|
12 | |
|
13 | |
|
14 | |
|
15 | |
|
16 | |
|
17 | |
|
18 | |
package org.apache.commons.pipeline.testFramework; |
19 | |
|
20 | |
import java.util.ArrayList; |
21 | |
import java.util.Collection; |
22 | |
import java.util.Collections; |
23 | |
import java.util.EventObject; |
24 | |
import java.util.HashMap; |
25 | |
import java.util.List; |
26 | |
import java.util.Map; |
27 | |
|
28 | |
import org.apache.commons.pipeline.Feeder; |
29 | |
import org.apache.commons.pipeline.Stage; |
30 | |
import org.apache.commons.pipeline.StageContext; |
31 | |
import org.apache.commons.pipeline.StageEventListener; |
32 | |
|
33 | |
|
34 | |
|
35 | |
|
36 | |
|
37 | 47 | public class TestStageContext implements StageContext { |
38 | 47 | public List<StageEventListener> listeners = Collections.synchronizedList( new ArrayList<StageEventListener>() ); |
39 | 47 | public List<EventObject> raisedEvents = new ArrayList<EventObject>(); |
40 | 47 | public Map<String, TestFeeder> branchFeeders = new HashMap<String,TestFeeder>(); |
41 | 47 | public Map<Stage, Feeder> downstreamFeeders = new HashMap<Stage,Feeder>(); |
42 | 47 | public Map<String, Object> env = new HashMap<String, Object>(); |
43 | |
|
44 | |
public void registerListener(StageEventListener listener) { |
45 | 2 | this.listeners.add(listener); |
46 | 2 | } |
47 | |
|
48 | |
public void raise(EventObject ev) { |
49 | 0 | this.raisedEvents.add(ev); |
50 | 0 | notifyListeners( ev ); |
51 | 0 | } |
52 | |
|
53 | |
private void notifyListeners( EventObject event ) |
54 | |
{ |
55 | 0 | for( StageEventListener listener : listeners ) |
56 | |
{ |
57 | 0 | listener.notify( event ); |
58 | |
} |
59 | 0 | } |
60 | |
|
61 | |
|
62 | |
|
63 | |
|
64 | |
|
65 | |
public Feeder getBranchFeeder(String key) { |
66 | 2 | if (branchFeeders.containsKey(key)) { |
67 | 1 | return branchFeeders.get(key); |
68 | |
} else { |
69 | 1 | TestFeeder feeder = new TestFeeder(); |
70 | 1 | branchFeeders.put(key, feeder); |
71 | 1 | return feeder; |
72 | |
} |
73 | |
} |
74 | |
|
75 | |
|
76 | |
|
77 | |
|
78 | |
|
79 | |
public Feeder getDownstreamFeeder(Stage stage) { |
80 | 50 | if (downstreamFeeders.containsKey(stage)) { |
81 | 50 | return downstreamFeeders.get(stage); |
82 | |
} else { |
83 | 0 | TestFeeder feeder = new TestFeeder(); |
84 | 0 | downstreamFeeders.put(stage, feeder); |
85 | 0 | return feeder; |
86 | |
} |
87 | |
} |
88 | |
|
89 | |
|
90 | |
|
91 | |
|
92 | |
|
93 | |
public void registerDownstreamFeeder(Stage stage, Feeder feeder) { |
94 | 57 | this.downstreamFeeders.put(stage, feeder); |
95 | 57 | } |
96 | |
|
97 | |
public Collection<StageEventListener> getRegisteredListeners() { |
98 | 0 | return this.listeners; |
99 | |
} |
100 | |
|
101 | |
|
102 | |
|
103 | |
|
104 | |
|
105 | |
|
106 | |
|
107 | |
|
108 | |
public Object getEnv(String key) { |
109 | 0 | return this.env.get(key); |
110 | |
} |
111 | |
} |