+
+ @Test
+ public void testInitMetric() throws FileNotFoundException, IOException
+ {
+ Configuration configuration = Mockito.mock(Configuration.class);
+ Properties properties = new Properties();
+ properties.put("metric.enabled", "true");
+ Mockito.when(configuration.getProperties()).thenReturn(properties);
+ PowerMockito.mockStatic(MRClientFactory.class);
+ PowerMockito.when(MRClientFactory.createConsumer(Mockito.anyString())).thenThrow(new IOException());
+ DmaapConsumerImpl consumer = Mockito.spy(new DmaapConsumerImpl(hosts, topic, group, id, key, secret));
+ Whitebox.setInternalState(consumer, "configuration", configuration);
+ MetricService metricService = Mockito.mock(MetricService.class);
+ MetricRegistry metricRegistry = Mockito.mock(MetricRegistry.class);
+ MetricBuilderFactory metricBuilderFactory = Mockito.spy(new MetricBuilderFactoryImpl());
+ DmaapRequestCounterBuilder builder = Mockito.mock(DmaapRequestCounterBuilder.class);
+ DmaapRequestCounterMetric metric = Mockito.mock(DmaapRequestCounterMetric.class);
+ Mockito.when(builder.withName(Mockito.anyString())).thenReturn(builder);
+ Mockito.when(builder.withType(Mockito.any())).thenReturn(builder);
+ Mockito.when(builder.withPublishedMessage(Mockito.anyLong())).thenReturn(builder);
+ Mockito.when(builder.withRecievedMessage(Mockito.anyLong())).thenReturn(builder);
+ Mockito.when(builder.build()).thenReturn(metric);
+ Mockito.when(metricBuilderFactory.dmaapRequestCounterBuilder()).thenReturn(builder);
+ Mockito.when(metricRegistry.register(Mockito.any())).thenReturn(true);
+ PublishingPolicy policy = Mockito.mock(PublishingPolicy.class);
+ PolicyBuilderFactory policyFactory = Mockito.mock(PolicyBuilderFactory.class);
+ Mockito.when(metricRegistry.policyBuilderFactory()).thenReturn(policyFactory);
+ ScheduledPolicyBuilder policyBuilder = Mockito.mock(ScheduledPolicyBuilder.class);
+ Mockito.when(policyBuilder.withPublishers(Mockito.any())).thenReturn(policyBuilder);
+ Mockito.when(policyBuilder.withMetrics(Mockito.any())).thenReturn(policyBuilder);
+ Mockito.when(policyBuilder.build()).thenReturn(policy);
+ Mockito.when(policyFactory.scheduledPolicyBuilder()).thenReturn(policyBuilder);
+ Mockito.when(metricRegistry.metricBuilderFactory()).thenReturn(metricBuilderFactory);
+ Mockito.when(metricService.createRegistry("APPC")).thenReturn(metricRegistry);
+ Mockito.doReturn(metricService).when(consumer).getMetricService();
+ consumer.fetch(1, 1);
+ Mockito.verify(policy).init();
+ }
+