View Javadoc

1   package org.apache.helix;
2   
3   /*
4    * Licensed to the Apache Software Foundation (ASF) under one
5    * or more contributor license agreements.  See the NOTICE file
6    * distributed with this work for additional information
7    * regarding copyright ownership.  The ASF licenses this file
8    * to you under the Apache License, Version 2.0 (the
9    * "License"); you may not use this file except in compliance
10   * with the License.  You may obtain a copy of the License at
11   *
12   *   http://www.apache.org/licenses/LICENSE-2.0
13   *
14   * Unless required by applicable law or agreed to in writing,
15   * software distributed under the License is distributed on an
16   * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
17   * KIND, either express or implied.  See the License for the
18   * specific language governing permissions and limitations
19   * under the License.
20   */
21  
22  import org.apache.helix.Mocks.MockHealthReportProvider;
23  import org.apache.helix.Mocks.MockManager;
24  import org.apache.helix.healthcheck.ParticipantHealthReportCollectorImpl;
25  import org.testng.annotations.BeforeMethod;
26  import org.testng.annotations.Test;
27  
28  public class TestParticipantHealthReportCollectorImpl {
29  
30  	protected ParticipantHealthReportCollectorImpl _providerImpl;
31  	protected HelixManager _manager;
32  	protected MockHealthReportProvider _mockProvider;
33  	
34  	 @BeforeMethod (groups = {"unitTest"})
35  	public void setup()
36  	{
37  		 _providerImpl = new ParticipantHealthReportCollectorImpl(new MockManager(), "instance_123");
38  		 _mockProvider = new MockHealthReportProvider();
39  	}
40  	
41  	 @Test (groups = {"unitTest"})
42  	  public void testStart() throws Exception
43  	  {
44  		 _providerImpl.start();
45  		 _providerImpl.start();
46  	  }
47  	 
48  	 @Test (groups = {"unitTest"})
49  	  public void testStop() throws Exception
50  	  {
51  		 _providerImpl.stop();
52  		 _providerImpl.stop();
53  	  }
54  	 
55  	 @Test (groups = {"unitTest"})
56  	 public void testAddProvider() throws Exception 
57  	 {
58  		 _providerImpl.removeHealthReportProvider(_mockProvider);
59  		 _providerImpl.addHealthReportProvider(_mockProvider);
60  		 _providerImpl.addHealthReportProvider(_mockProvider);
61  	 }
62  	 
63  	 @Test (groups = {"unitTest"})
64  	 public void testRemoveProvider() throws Exception
65  	 {
66  		 _providerImpl.addHealthReportProvider(_mockProvider);
67  		 _providerImpl.removeHealthReportProvider(_mockProvider);
68  		 _providerImpl.removeHealthReportProvider(_mockProvider);
69  	 }
70  }