| /* |
| * 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.camel.component.telegram.springboot; |
| |
| |
| import java.util.concurrent.TimeUnit; |
| |
| import org.apache.camel.CamelContext; |
| import org.apache.camel.EndpointInject; |
| import org.apache.camel.builder.RouteBuilder; |
| import org.apache.camel.component.mock.MockEndpoint; |
| import org.apache.camel.health.HealthCheck; |
| import org.apache.camel.health.HealthCheckRegistry; |
| import org.apache.camel.health.HealthCheckRepository; |
| import org.apache.camel.spring.boot.CamelAutoConfiguration; |
| |
| import org.junit.jupiter.api.Assertions; |
| import org.junit.jupiter.api.Test; |
| |
| |
| |
| import org.springframework.boot.test.context.SpringBootTest; |
| import org.springframework.context.annotation.Bean; |
| import org.springframework.context.annotation.Configuration; |
| import org.springframework.test.annotation.DirtiesContext; |
| import org.springframework.test.annotation.DirtiesContext.ClassMode; |
| import org.apache.camel.test.spring.junit5.CamelSpringBootTest; |
| import org.awaitility.Awaitility; |
| |
| |
| @DirtiesContext(classMode = ClassMode.AFTER_EACH_TEST_METHOD) |
| @CamelSpringBootTest |
| @SpringBootTest( |
| classes = { |
| CamelAutoConfiguration.class, |
| TelegramConsumerHealthCheckErrorTest.class, |
| TelegramConsumerHealthCheckErrorTest.TestConfiguration.class |
| } |
| ) |
| public class TelegramConsumerHealthCheckErrorTest extends TelegramTestSupport { |
| |
| |
| static TelegramMockRoutes mockRoutes; |
| |
| @EndpointInject("mock:telegram") |
| private MockEndpoint endpoint; |
| |
| @Override |
| protected void configureCamelContext(CamelContext context) { |
| // enabling consumers health check is a bit cumbersome via low-level Java code |
| super.configureCamelContext(context); |
| HealthCheckRegistry hcr = context.getExtension(HealthCheckRegistry.class); |
| HealthCheckRepository repo |
| = hcr.getRepository("consumers").orElse((HealthCheckRepository) hcr.resolveById("consumers")); |
| repo.setEnabled(true); |
| hcr.register(repo); |
| } |
| |
| @Test |
| public void testReceptionOfTwoMessages() { |
| HealthCheckRegistry hcr = context.getExtension(HealthCheckRegistry.class); |
| HealthCheckRepository repo = hcr.getRepository("consumers").get(); |
| |
| // wait until HC is DOWN |
| Awaitility.waitAtMost(5, TimeUnit.SECONDS).until( |
| () -> repo.stream().anyMatch(h -> h.call().getState().equals(HealthCheck.State.DOWN))); |
| |
| // if we grab the health check by id, we can also check it afterwards |
| HealthCheck hc = hcr.getCheck("consumer:telegram").get(); |
| |
| // wait until we have the error |
| Awaitility.waitAtMost(5, TimeUnit.SECONDS).until( |
| () -> { |
| HealthCheck.Result rc = hc.call(); |
| Long count = (Long) rc.getDetails().get(HealthCheck.FAILURE_ERROR_COUNT); |
| return count != null && count > 0; |
| }); |
| |
| HealthCheck.Result rc = hc.call(); |
| |
| // and get the detailed error message (and exception) |
| Assertions.assertEquals(HealthCheck.State.DOWN, rc.getState()); |
| String msg = rc.getMessage().get(); |
| long count = (long) rc.getDetails().get(HealthCheck.FAILURE_ERROR_COUNT); |
| Assertions.assertEquals("Consumer failed polling " + count + " times route: telegram (telegram://bots)", msg); |
| // test that the uri is masked |
| Assertions.assertEquals("telegram://bots?authorizationToken=xxxxxx", |
| rc.getDetails().get(HealthCheck.ENDPOINT_URI)); |
| |
| Throwable e = rc.getError().get(); |
| Assertions.assertTrue(e.getMessage().contains("401 Unauthorized")); |
| Assertions.assertEquals(401, rc.getDetails().get(HealthCheck.HTTP_RESPONSE_CODE)); |
| } |
| |
| |
| // ************************************* |
| // Config |
| // ************************************* |
| |
| @Configuration |
| public class TestConfiguration { |
| @Bean |
| public RouteBuilder routeBuilder() { |
| return new RouteBuilder() { |
| @Override |
| public void configure() { |
| from("telegram:bots?authorizationToken=mock-token").routeId("telegram") |
| .convertBodyTo(String.class) |
| .to("mock:telegram"); |
| } |
| }; |
| } |
| |
| } |
| |
| @Override |
| @Bean |
| protected TelegramMockRoutes createMockRoutes() { |
| mockRoutes = |
| new TelegramMockRoutes(port) |
| .addErrorEndpoint( |
| "getUpdates", |
| "GET", |
| 401); |
| return mockRoutes; |
| } |
| } |