blob: a19280288b71a5b3393d01f64adfd94a18bc4c6f (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
|
/**
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.apache.hadoop.metrics2.sink.timeline.cache;
import java.io.IOException;
import java.io.OutputStream;
import java.net.HttpURLConnection;
import java.net.URL;
import org.apache.hadoop.metrics2.sink.timeline.AbstractTimelineMetricsSink;
import org.apache.hadoop.metrics2.sink.timeline.TimelineMetric;
import org.apache.hadoop.metrics2.sink.timeline.TimelineMetrics;
import org.apache.hadoop.metrics2.sink.timeline.UnableToConnectException;
import org.junit.Assert;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.powermock.core.classloader.annotations.PrepareForTest;
import org.powermock.modules.junit4.PowerMockRunner;
import static org.easymock.EasyMock.expect;
import static org.powermock.api.easymock.PowerMock.createNiceMock;
import static org.powermock.api.easymock.PowerMock.expectNew;
import static org.powermock.api.easymock.PowerMock.replayAll;
@RunWith(PowerMockRunner.class)
@PrepareForTest({AbstractTimelineMetricsSink.class, URL.class,
HttpURLConnection.class})
public class HandleConnectExceptionTest {
private static final String COLLECTOR_URL = "collector";
private TestTimelineMetricsSink sink;
@Before
public void init(){
sink = new TestTimelineMetricsSink();
OutputStream os = createNiceMock(OutputStream.class);
HttpURLConnection connection = createNiceMock(HttpURLConnection.class);
URL url = createNiceMock(URL.class);
AbstractTimelineMetricsSink.NUMBER_OF_SKIPPED_COLLECTOR_EXCEPTIONS = 2;
try {
expectNew(URL.class, "collector").andReturn(url).anyTimes();
expect(url.openConnection()).andReturn(connection).anyTimes();
expect(connection.getOutputStream()).andReturn(os).anyTimes();
expect(connection.getResponseCode()).andThrow(new IOException()).anyTimes();
replayAll();
} catch (Exception e) {
//no-op
}
}
@Test
public void handleTest(){
emitMetricsWithExpectedException(new TimelineMetrics());
try {
sink.emitMetrics(new TimelineMetrics());
} catch (Exception e) {
Assert.fail("There should be no exception");
}
emitMetricsWithExpectedException(new TimelineMetrics());
}
private void emitMetricsWithExpectedException(TimelineMetrics timelineMetrics) {
try{
sink.emitMetrics(timelineMetrics);
Assert.fail();
}catch(UnableToConnectException e){
Assert.assertEquals(COLLECTOR_URL, e.getConnectUrl());
}catch(Exception e){
Assert.fail(e.getMessage());
}
}
class TestTimelineMetricsSink extends AbstractTimelineMetricsSink{
@Override
protected String getCollectorUri() {
return COLLECTOR_URL;
}
@Override
protected int getTimeoutSeconds() {
return 10;
}
@Override
public boolean emitMetrics(TimelineMetrics metrics) {
return super.emitMetrics(metrics);
}
}
}
|