| /* |
| * 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.karaf.decanter.itests.collector; |
| |
| import org.apache.karaf.itests.KarafTestSupport; |
| import org.apache.karaf.jaas.boot.principal.RolePrincipal; |
| import org.junit.Assert; |
| import org.junit.Test; |
| import org.junit.runner.RunWith; |
| import org.ops4j.pax.exam.Configuration; |
| import org.ops4j.pax.exam.MavenUtils; |
| import org.ops4j.pax.exam.Option; |
| import org.ops4j.pax.exam.junit.PaxExam; |
| import org.ops4j.pax.exam.karaf.options.KarafDistributionOption; |
| import org.ops4j.pax.exam.spi.reactors.ExamReactorStrategy; |
| import org.ops4j.pax.exam.spi.reactors.PerClass; |
| import org.osgi.service.event.Event; |
| import org.osgi.service.event.EventConstants; |
| import org.osgi.service.event.EventHandler; |
| import org.osgi.service.http.HttpService; |
| |
| import javax.inject.Inject; |
| import javax.servlet.ServletException; |
| import javax.servlet.http.HttpServlet; |
| import javax.servlet.http.HttpServletRequest; |
| import javax.servlet.http.HttpServletResponse; |
| import java.io.*; |
| import java.util.ArrayList; |
| import java.util.Hashtable; |
| import java.util.List; |
| import java.util.stream.Stream; |
| |
| @RunWith(PaxExam.class) |
| @ExamReactorStrategy(PerClass.class) |
| public class PrometheusCollectorTest extends KarafTestSupport { |
| |
| @Inject |
| private HttpService httpService; |
| |
| @Configuration |
| public Option[] config() { |
| String karafVersion = MavenUtils.getArtifactVersion("org.apache.karaf", "apache-karaf"); |
| Option[] options = new Option[]{ |
| KarafDistributionOption.editConfigurationFilePut("etc/system.properties", "decanter.version", System.getProperty("decanter.version")), |
| KarafDistributionOption.features("mvn:org.apache.karaf.features/standard/" + karafVersion + "/xml/features", "http") |
| }; |
| return Stream.of(super.config(), options).flatMap(Stream::of).toArray(Option[]::new); |
| } |
| |
| @Test(timeout = 120000) |
| public void test() throws Exception { |
| System.out.println("Deploying Prometheus test servlet ..."); |
| httpService.registerServlet("/prometheus", new HttpServlet() { |
| @Override |
| public void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException { |
| try (BufferedWriter writer = new BufferedWriter(new OutputStreamWriter(response.getOutputStream()))) { |
| writer.write("# HELP Test Test\n"); |
| writer.write("# TYPE Test gauge\n"); |
| writer.write("Test 0.0"); |
| writer.flush(); |
| } |
| } |
| }, null, null); |
| |
| String httpList = executeCommand("http:list"); |
| while (!httpList.contains("Deployed")) { |
| Thread.sleep(500); |
| httpList = executeCommand("http:list"); |
| } |
| |
| System.out.println("Adding test event handler ..."); |
| // create event handler |
| List<Event> received = new ArrayList(); |
| EventHandler eventHandler = new EventHandler() { |
| @Override |
| public void handleEvent(Event event) { |
| received.add(event); |
| } |
| }; |
| Hashtable serviceProperties = new Hashtable(); |
| serviceProperties.put(EventConstants.EVENT_TOPIC, "decanter/collect/*"); |
| bundleContext.registerService(EventHandler.class, eventHandler, serviceProperties); |
| |
| System.out.println("Installing Decanter Collector Prometheus ..."); |
| File file = new File(System.getProperty("karaf.etc"), "org.apache.karaf.decanter.collector.prometheus.cfg"); |
| try (BufferedWriter writer = new BufferedWriter(new FileWriter(file))) { |
| writer.write("prometheus.url=http://localhost:" + getHttpPort() + "/prometheus"); |
| writer.flush(); |
| } |
| String configList = executeCommand("config:list '(service.pid=org.apache.karaf.decanter.collector.prometheus)'"); |
| while (!configList.contains("service.pid")) { |
| Thread.sleep(500); |
| configList = executeCommand("config:list '(service.pid=org.apache.karaf.decanter.collector.prometheus)'"); |
| } |
| System.out.println(executeCommand("feature:repo-add decanter " + System.getProperty("decanter.version"), new RolePrincipal("admin"))); |
| System.out.println(executeCommand("feature:install decanter-collector-prometheus", new RolePrincipal("admin"))); |
| |
| System.out.println("Waiting events ..."); |
| while (received.size() < 1) { |
| Thread.sleep(500); |
| } |
| |
| System.out.println(""); |
| |
| for (int i = 0; i < received.size(); i++) { |
| for (String property : received.get(i).getPropertyNames()) { |
| System.out.println(property + " = " + received.get(i).getProperty(property)); |
| } |
| System.out.println("========"); |
| } |
| |
| System.out.println(""); |
| |
| Assert.assertEquals(1, received.size()); |
| |
| Assert.assertEquals(0.0, received.get(0).getProperty("Test")); |
| Assert.assertEquals("decanter/collect/prometheus", received.get(0).getProperty("event.topics")); |
| Assert.assertEquals("prometheus", received.get(0).getProperty("type")); |
| } |
| |
| } |