1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18 package org.apache.hadoop.chukwa.extraction.demux;
19
20 import junit.framework.TestCase;
21 import org.apache.hadoop.chukwa.conf.ChukwaConfiguration;
22 import org.apache.hadoop.chukwa.extraction.CHUKWA_CONSTANT;
23 import org.apache.hadoop.chukwa.datatrigger.TriggerEvent;
24 import org.apache.hadoop.fs.Path;
25
26 import java.util.Iterator;
27
28 public class TestPostDemuxTrigger extends TestCase {
29
30 static final Path[] SAMPLE_PATHS = new Path[] { new Path("/") };
31
32 protected void setUp() throws Exception {
33 MockTriggerAction.reset();
34 }
35
36 public void testSuccessTrigger() throws Exception {
37 ChukwaConfiguration conf = new ChukwaConfiguration();
38 conf.set(CHUKWA_CONSTANT.POST_DEMUX_SUCCESS_ACTION,
39 "org.apache.hadoop.chukwa.extraction.demux.MockTriggerAction");
40
41 PostProcessorManager postProcessManager = new PostProcessorManager(conf);
42 assertTrue("processPostMoveTriggers returned false",
43 postProcessManager.processPostMoveTriggers(SAMPLE_PATHS));
44
45 assertEquals("Trigger never invoked", SAMPLE_PATHS.length,
46 MockTriggerAction.getTriggerEvents().size());
47 Iterator events = MockTriggerAction.getTriggerEvents().iterator();
48 assertEquals("Incorrect Trigger event found", TriggerEvent.POST_DEMUX_SUCCESS,
49 events.next());
50 }
51
52 public void testMultiSuccessTrigger() throws Exception {
53 ChukwaConfiguration conf = new ChukwaConfiguration();
54 conf.set(CHUKWA_CONSTANT.POST_DEMUX_SUCCESS_ACTION,
55 "org.apache.hadoop.chukwa.extraction.demux.MockTriggerAction," +
56 "org.apache.hadoop.chukwa.extraction.demux.MockTriggerAction");
57
58 PostProcessorManager postProcessManager = new PostProcessorManager(conf);
59 assertTrue("processPostMoveTriggers returned false",
60 postProcessManager.processPostMoveTriggers(SAMPLE_PATHS));
61
62 assertEquals("Trigger never invoked", 2 * SAMPLE_PATHS.length,
63 MockTriggerAction.getTriggerEvents().size());
64 Iterator events = MockTriggerAction.getTriggerEvents().iterator();
65 assertEquals("Incorrect Trigger event found", TriggerEvent.POST_DEMUX_SUCCESS,
66 events.next());
67 assertEquals("Incorrect Trigger event found", TriggerEvent.POST_DEMUX_SUCCESS,
68 events.next());
69 }
70
71 }