View Javadoc

1   /*
2    * Licensed to the Apache Software Foundation (ASF) under one
3    * or more contributor license agreements.  See the NOTICE file
4    * distributed with this work for additional information
5    * regarding copyright ownership.  The ASF licenses this file
6    * to you under the Apache License, Version 2.0 (the
7    * "License"); you may not use this file except in compliance
8    * with the License.  You may obtain a copy of the License at
9    *
10   *     http://www.apache.org/licenses/LICENSE-2.0
11   *
12   * Unless required by applicable law or agreed to in writing, software
13   * distributed under the License is distributed on an "AS IS" BASIS,
14   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15   * See the License for the specific language governing permissions and
16   * limitations under the License.
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  }