use of io.cdap.cdap.api.metrics.RuntimeMetrics in project cdap by caskdata.
the class TestFrameworkTestRun method testAppWithServices.
@Category(SlowTests.class)
@Test
public void testAppWithServices() throws Exception {
ApplicationManager applicationManager = deployApplication(AppWithServices.class);
LOG.info("Deployed.");
ServiceManager serviceManager = applicationManager.getServiceManager(AppWithServices.SERVICE_NAME).start();
serviceManager.waitForRun(ProgramRunStatus.RUNNING, 10, TimeUnit.SECONDS);
LOG.info("Service Started");
URL serviceURL = serviceManager.getServiceURL(15, TimeUnit.SECONDS);
Assert.assertNotNull(serviceURL);
// Call the ping endpoint
URL url = new URL(serviceURL, "ping2");
HttpRequest request = HttpRequest.get(url).build();
HttpResponse response = executeHttp(request);
Assert.assertEquals(200, response.getResponseCode());
// Call the failure endpoint
url = new URL(serviceURL, "failure");
request = HttpRequest.get(url).build();
response = executeHttp(request);
Assert.assertEquals(500, response.getResponseCode());
Assert.assertTrue(response.getResponseBodyAsString().contains("Exception"));
// Call the verify ClassLoader endpoint
url = new URL(serviceURL, "verifyClassLoader");
request = HttpRequest.get(url).build();
response = executeHttp(request);
Assert.assertEquals(200, response.getResponseCode());
RuntimeMetrics serviceMetrics = serviceManager.getMetrics();
serviceMetrics.waitForinput(3, 5, TimeUnit.SECONDS);
Assert.assertEquals(3, serviceMetrics.getInput());
Assert.assertEquals(2, serviceMetrics.getProcessed());
Assert.assertEquals(1, serviceMetrics.getException());
// in the AppWithServices the handlerName is same as the serviceName - "ServerService" handler
RuntimeMetrics handlerMetrics = getMetricsManager().getServiceHandlerMetrics(NamespaceId.DEFAULT.getNamespace(), AppWithServices.APP_NAME, AppWithServices.SERVICE_NAME, AppWithServices.SERVICE_NAME);
handlerMetrics.waitForinput(3, 5, TimeUnit.SECONDS);
Assert.assertEquals(3, handlerMetrics.getInput());
Assert.assertEquals(2, handlerMetrics.getProcessed());
Assert.assertEquals(1, handlerMetrics.getException());
// we can verify metrics, by adding getServiceMetrics in MetricsManager and then disabling the system scope test in
// TestMetricsCollectionService
LOG.info("DatasetUpdateService Started");
Map<String, String> args = ImmutableMap.of(AppWithServices.WRITE_VALUE_RUN_KEY, AppWithServices.DATASET_TEST_VALUE, AppWithServices.WRITE_VALUE_STOP_KEY, AppWithServices.DATASET_TEST_VALUE_STOP);
ServiceManager datasetWorkerServiceManager = applicationManager.getServiceManager(AppWithServices.DATASET_WORKER_SERVICE_NAME).start(args);
WorkerManager datasetWorker = applicationManager.getWorkerManager(AppWithServices.DATASET_UPDATE_WORKER).start(args);
datasetWorker.waitForRun(ProgramRunStatus.RUNNING, 10, TimeUnit.SECONDS);
datasetWorkerServiceManager.waitForRun(ProgramRunStatus.RUNNING, 10, TimeUnit.SECONDS);
ServiceManager noopManager = applicationManager.getServiceManager("NoOpService").start();
noopManager.waitForRun(ProgramRunStatus.RUNNING, 10, TimeUnit.SECONDS);
// We don't know when the datasetWorker run() method executed, hence need to retry on the service call
// until it can get a value from a dataset, which was written out by the datasetWorker.
AtomicInteger called = new AtomicInteger(0);
AtomicInteger failed = new AtomicInteger(0);
Tasks.waitFor(AppWithServices.DATASET_TEST_VALUE, new Callable<String>() {
@Override
public String call() throws Exception {
URL url = noopManager.getServiceURL();
String path = "ping/" + AppWithServices.DATASET_TEST_KEY;
try {
called.incrementAndGet();
return new Gson().fromJson(callServiceGet(url, path), String.class);
} catch (IOException e) {
failed.incrementAndGet();
LOG.debug("Exception when reading from service {}/{}", url, path, e);
}
return null;
}
}, 30, TimeUnit.SECONDS);
// Validates the metrics emitted by the service call.
handlerMetrics = getMetricsManager().getServiceHandlerMetrics(NamespaceId.DEFAULT.getNamespace(), AppWithServices.APP_NAME, "NoOpService", "NoOpHandler");
handlerMetrics.waitForinput(called.get(), 5, TimeUnit.SECONDS);
handlerMetrics.waitForProcessed(1, 5, TimeUnit.SECONDS);
handlerMetrics.waitForException(failed.get(), 5, TimeUnit.SECONDS);
// Test that a service can discover another service
String path = String.format("discover/%s/%s", AppWithServices.APP_NAME, AppWithServices.DATASET_WORKER_SERVICE_NAME);
url = new URL(serviceURL, path);
request = HttpRequest.get(url).build();
response = executeHttp(request);
Assert.assertEquals(200, response.getResponseCode());
datasetWorker.stop();
datasetWorker.waitForStopped(10, TimeUnit.SECONDS);
datasetWorkerServiceManager.stop();
datasetWorkerServiceManager.waitForStopped(10, TimeUnit.SECONDS);
LOG.info("DatasetUpdateService Stopped");
serviceManager.stop();
serviceManager.waitForStopped(10, TimeUnit.SECONDS);
LOG.info("ServerService Stopped");
// Since all worker are stopped, we can just hit the service to read the dataset. No retry needed.
String result = callServiceGet(noopManager.getServiceURL(), "ping/" + AppWithServices.DATASET_TEST_KEY_STOP);
String decodedResult = new Gson().fromJson(result, String.class);
Assert.assertEquals(AppWithServices.DATASET_TEST_VALUE_STOP, decodedResult);
result = callServiceGet(noopManager.getServiceURL(), "ping/" + AppWithServices.DATASET_TEST_KEY_STOP_2);
decodedResult = new Gson().fromJson(result, String.class);
Assert.assertEquals(AppWithServices.DATASET_TEST_VALUE_STOP_2, decodedResult);
}
Aggregations