Examples with MockLogAppender org.opensearch.test.MockLogAppender used on opensource projects

Example 11 with MockLogAppender

use of org.opensearch.test.MockLogAppender in project OpenSearch by opensearch-project.

the class PluginsServiceTests method testFindPluginDirs.

public void testFindPluginDirs() throws Exception {
    final Path plugins = createTempDir();
    try (MockLogAppender mockLogAppender = MockLogAppender.createForLoggers(LogManager.getLogger(PluginsService.class))) {
        mockLogAppender.addExpectation(new MockLogAppender.SeenEventExpectation("[.test] warning", "org.opensearch.plugins.PluginsService", Level.WARN, "Non-plugin file located in the plugins folder with the following name: [.DS_Store]"));
        final Path fake = plugins.resolve("fake");
        Path testFile = plugins.resolve(".DS_Store");
        Files.createFile(testFile);
        PluginTestUtil.writePluginProperties(fake, "description", "description", "name", "fake", "version", "1.0.0", "opensearch.version", Version.CURRENT.toString(), "java.version", System.getProperty("java.specification.version"), "classname", "test.DummyPlugin");
        try (InputStream jar = PluginsServiceTests.class.getResourceAsStream("dummy-plugin.jar")) {
            Files.copy(jar, fake.resolve("plugin.jar"));
        }
        assertThat(PluginsService.findPluginDirs(plugins), containsInAnyOrder(fake));
        mockLogAppender.assertAllExpectationsMatched();
    }
}

Example 12 with MockLogAppender

use of org.opensearch.test.MockLogAppender in project OpenSearch by opensearch-project.

the class AbstractSimpleTransportTestCase method testTracerLog.

@TestLogging(value = "org.opensearch.transport.TransportService.tracer:trace", reason = "to ensure we log network events on TRACE level")
public void testTracerLog() throws Exception {
    TransportRequestHandler<TransportRequest> handler = (request, channel, task) -> channel.sendResponse(new StringMessageResponse(""));
    TransportRequestHandler<StringMessageRequest> handlerWithError = (request, channel, task) -> {
        if (request.timeout() > 0) {
            Thread.sleep(request.timeout);
        }
        channel.sendResponse(new RuntimeException(""));
    };
    TransportResponseHandler<StringMessageResponse> noopResponseHandler = new TransportResponseHandler<StringMessageResponse>() {

        @Override
        public StringMessageResponse read(StreamInput in) throws IOException {
            return new StringMessageResponse(in);
        }

        @Override
        public void handleResponse(StringMessageResponse response) {
        }

        @Override
        public void handleException(TransportException exp) {
        }

        @Override
        public String executor() {
            return ThreadPool.Names.SAME;
        }
    };
    serviceA.registerRequestHandler("internal:test", ThreadPool.Names.SAME, StringMessageRequest::new, handler);
    serviceA.registerRequestHandler("internal:testNotSeen", ThreadPool.Names.SAME, StringMessageRequest::new, handler);
    serviceA.registerRequestHandler("internal:testError", ThreadPool.Names.SAME, StringMessageRequest::new, handlerWithError);
    serviceB.registerRequestHandler("internal:test", ThreadPool.Names.SAME, StringMessageRequest::new, handler);
    serviceB.registerRequestHandler("internal:testNotSeen", ThreadPool.Names.SAME, StringMessageRequest::new, handler);
    serviceB.registerRequestHandler("internal:testError", ThreadPool.Names.SAME, StringMessageRequest::new, handlerWithError);
    String includeSettings;
    String excludeSettings;
    if (randomBoolean()) {
        // sometimes leave include empty (default)
        includeSettings = randomBoolean() ? "*" : "";
        excludeSettings = "internal:testNotSeen";
    } else {
        includeSettings = "internal:test,internal:testError";
        excludeSettings = "DOESN'T_MATCH";
    }
    clusterSettingsA.applySettings(Settings.builder().put(TransportSettings.TRACE_LOG_INCLUDE_SETTING.getKey(), includeSettings).put(TransportSettings.TRACE_LOG_EXCLUDE_SETTING.getKey(), excludeSettings).build());
    final Logger logger = LogManager.getLogger("org.opensearch.transport.TransportService.tracer");
    try (MockLogAppender appender = MockLogAppender.createForLoggers(logger)) {
        final String requestSent = ".*\\[internal:test].*sent to.*\\{TS_B}.*";
        final MockLogAppender.LoggingExpectation requestSentExpectation = new MockLogAppender.PatternSeenEventExpectation("sent request", "org.opensearch.transport.TransportService.tracer", Level.TRACE, requestSent);
        final String requestReceived = ".*\\[internal:test].*received request.*";
        final MockLogAppender.LoggingExpectation requestReceivedExpectation = new MockLogAppender.PatternSeenEventExpectation("received request", "org.opensearch.transport.TransportService.tracer", Level.TRACE, requestReceived);
        final String responseSent = ".*\\[internal:test].*sent response.*";
        final MockLogAppender.LoggingExpectation responseSentExpectation = new MockLogAppender.PatternSeenEventExpectation("sent response", "org.opensearch.transport.TransportService.tracer", Level.TRACE, responseSent);
        final String responseReceived = ".*\\[internal:test].*received response from.*\\{TS_B}.*";
        final MockLogAppender.LoggingExpectation responseReceivedExpectation = new MockLogAppender.PatternSeenEventExpectation("received response", "org.opensearch.transport.TransportService.tracer", Level.TRACE, responseReceived);
        appender.addExpectation(requestSentExpectation);
        appender.addExpectation(requestReceivedExpectation);
        appender.addExpectation(responseSentExpectation);
        appender.addExpectation(responseReceivedExpectation);
        StringMessageRequest request = new StringMessageRequest("", 10);
        serviceA.sendRequest(nodeB, "internal:test", request, TransportRequestOptions.EMPTY, noopResponseHandler);
        assertBusy(appender::assertAllExpectationsMatched);
        final String errorResponseSent = ".*\\[internal:testError].*sent error response.*";
        final MockLogAppender.LoggingExpectation errorResponseSentExpectation = new MockLogAppender.PatternSeenEventExpectation("sent error response", "org.opensearch.transport.TransportService.tracer", Level.TRACE, errorResponseSent);
        final String errorResponseReceived = ".*\\[internal:testError].*received response from.*\\{TS_B}.*";
        final MockLogAppender.LoggingExpectation errorResponseReceivedExpectation = new MockLogAppender.PatternSeenEventExpectation("received error response", "org.opensearch.transport.TransportService.tracer", Level.TRACE, errorResponseReceived);
        appender.addExpectation(errorResponseSentExpectation);
        appender.addExpectation(errorResponseReceivedExpectation);
        serviceA.sendRequest(nodeB, "internal:testError", new StringMessageRequest(""), noopResponseHandler);
        assertBusy(appender::assertAllExpectationsMatched);
        final String notSeenSent = "*[internal:testNotSeen]*sent to*";
        final MockLogAppender.LoggingExpectation notSeenSentExpectation = new MockLogAppender.UnseenEventExpectation("not seen request sent", "org.opensearch.transport.TransportService.tracer", Level.TRACE, notSeenSent);
        final String notSeenReceived = ".*\\[internal:testNotSeen].*received request.*";
        final MockLogAppender.LoggingExpectation notSeenReceivedExpectation = new MockLogAppender.PatternSeenEventExpectation("not seen request received", "org.opensearch.transport.TransportService.tracer", Level.TRACE, notSeenReceived);
        appender.addExpectation(notSeenSentExpectation);
        appender.addExpectation(notSeenReceivedExpectation);
        PlainTransportFuture<StringMessageResponse> future = new PlainTransportFuture<>(noopResponseHandler);
        serviceA.sendRequest(nodeB, "internal:testNotSeen", new StringMessageRequest(""), future);
        future.txGet();
        assertBusy(appender::assertAllExpectationsMatched);
    }
}

Example 13 with MockLogAppender

use of org.opensearch.test.MockLogAppender in project OpenSearch by opensearch-project.

the class ClusterApplierServiceTests method testLongClusterStateUpdateLogging.

@TestLogging(value = "org.opensearch.cluster.service:WARN", reason = "to ensure that we log cluster state events on WARN level")
public void testLongClusterStateUpdateLogging() throws Exception {
    try (MockLogAppender mockAppender = MockLogAppender.createForLoggers(LogManager.getLogger(ClusterApplierService.class))) {
        mockAppender.addExpectation(new MockLogAppender.UnseenEventExpectation("test1 shouldn't see because setting is too low", ClusterApplierService.class.getCanonicalName(), Level.WARN, "*cluster state applier task [test1] took [*] which is above the warn threshold of *"));
        mockAppender.addExpectation(new MockLogAppender.SeenEventExpectation("test2", ClusterApplierService.class.getCanonicalName(), Level.WARN, "*cluster state applier task [test2] took [32s] which is above the warn threshold of [*]: " + "[running task [test2]] took [*"));
        mockAppender.addExpectation(new MockLogAppender.SeenEventExpectation("test4", ClusterApplierService.class.getCanonicalName(), Level.WARN, "*cluster state applier task [test3] took [34s] which is above the warn threshold of [*]: " + "[running task [test3]] took [*"));
        final CountDownLatch latch = new CountDownLatch(4);
        final CountDownLatch processedFirstTask = new CountDownLatch(1);
        clusterApplierService.currentTimeOverride = threadPool.relativeTimeInMillis();
        clusterApplierService.runOnApplierThread("test1", currentState -> clusterApplierService.currentTimeOverride += TimeValue.timeValueSeconds(1).millis(), new ClusterApplyListener() {

            @Override
            public void onSuccess(String source) {
                latch.countDown();
                processedFirstTask.countDown();
            }

            @Override
            public void onFailure(String source, Exception e) {
                fail();
            }
        });
        processedFirstTask.await();
        clusterApplierService.runOnApplierThread("test2", currentState -> {
            clusterApplierService.currentTimeOverride += TimeValue.timeValueSeconds(32).millis();
            throw new IllegalArgumentException("Testing handling of exceptions in the cluster state task");
        }, new ClusterApplyListener() {

            @Override
            public void onSuccess(String source) {
                fail();
            }

            @Override
            public void onFailure(String source, Exception e) {
                latch.countDown();
            }
        });
        clusterApplierService.runOnApplierThread("test3", currentState -> clusterApplierService.currentTimeOverride += TimeValue.timeValueSeconds(34).millis(), new ClusterApplyListener() {

            @Override
            public void onSuccess(String source) {
                latch.countDown();
            }

            @Override
            public void onFailure(String source, Exception e) {
                fail();
            }
        });
        // Additional update task to make sure all previous logging made it to the loggerName
        // We don't check logging for this on since there is no guarantee that it will occur before our check
        clusterApplierService.runOnApplierThread("test4", currentState -> {
        }, new ClusterApplyListener() {

            @Override
            public void onSuccess(String source) {
                latch.countDown();
            }

            @Override
            public void onFailure(String source, Exception e) {
                fail();
            }
        });
        latch.await();
        mockAppender.assertAllExpectationsMatched();
    }
}

Example 14 with MockLogAppender

use of org.opensearch.test.MockLogAppender in project OpenSearch by opensearch-project.

the class ClusterApplierServiceTests method testClusterStateUpdateLogging.

@TestLogging(value = "org.opensearch.cluster.service:TRACE", reason = "to ensure that we log cluster state events on TRACE level")
public void testClusterStateUpdateLogging() throws Exception {
    try (MockLogAppender mockAppender = MockLogAppender.createForLoggers(LogManager.getLogger(ClusterApplierService.class))) {
        mockAppender.addExpectation(new MockLogAppender.SeenEventExpectation("test1", ClusterApplierService.class.getCanonicalName(), Level.DEBUG, "*processing [test1]: took [1s] no change in cluster state"));
        mockAppender.addExpectation(new MockLogAppender.SeenEventExpectation("test2", ClusterApplierService.class.getCanonicalName(), Level.TRACE, "*failed to execute cluster state applier in [2s]*"));
        mockAppender.addExpectation(new MockLogAppender.SeenEventExpectation("test3", ClusterApplierService.class.getCanonicalName(), Level.DEBUG, "*processing [test3]: took [0s] no change in cluster state*"));
        clusterApplierService.currentTimeOverride = threadPool.relativeTimeInMillis();
        clusterApplierService.runOnApplierThread("test1", currentState -> clusterApplierService.currentTimeOverride += TimeValue.timeValueSeconds(1).millis(), new ClusterApplyListener() {

            @Override
            public void onSuccess(String source) {
            }

            @Override
            public void onFailure(String source, Exception e) {
                fail();
            }
        });
        clusterApplierService.runOnApplierThread("test2", currentState -> {
            clusterApplierService.currentTimeOverride += TimeValue.timeValueSeconds(2).millis();
            throw new IllegalArgumentException("Testing handling of exceptions in the cluster state task");
        }, new ClusterApplyListener() {

            @Override
            public void onSuccess(String source) {
                fail();
            }

            @Override
            public void onFailure(String source, Exception e) {
            }
        });
        // Additional update task to make sure all previous logging made it to the loggerName
        clusterApplierService.runOnApplierThread("test3", currentState -> {
        }, new ClusterApplyListener() {

            @Override
            public void onSuccess(String source) {
            }

            @Override
            public void onFailure(String source, Exception e) {
                fail();
            }
        });
        assertBusy(mockAppender::assertAllExpectationsMatched);
    }
}

Example 15 with MockLogAppender

use of org.opensearch.test.MockLogAppender in project OpenSearch by opensearch-project.

the class MasterServiceTests method testClusterStateUpdateLogging.

@TestLogging(value = "org.opensearch.cluster.service:TRACE", reason = "to ensure that we log cluster state events on TRACE level")
public void testClusterStateUpdateLogging() throws Exception {
    try (MockLogAppender mockAppender = MockLogAppender.createForLoggers(LogManager.getLogger(MasterService.class))) {
        mockAppender.addExpectation(new MockLogAppender.SeenEventExpectation("test1 start", MasterService.class.getCanonicalName(), Level.DEBUG, "executing cluster state update for [test1]"));
        mockAppender.addExpectation(new MockLogAppender.SeenEventExpectation("test1 computation", MasterService.class.getCanonicalName(), Level.DEBUG, "took [1s] to compute cluster state update for [test1]"));
        mockAppender.addExpectation(new MockLogAppender.SeenEventExpectation("test1 notification", MasterService.class.getCanonicalName(), Level.DEBUG, "took [0s] to notify listeners on unchanged cluster state for [test1]"));
        mockAppender.addExpectation(new MockLogAppender.SeenEventExpectation("test2 start", MasterService.class.getCanonicalName(), Level.DEBUG, "executing cluster state update for [test2]"));
        mockAppender.addExpectation(new MockLogAppender.SeenEventExpectation("test2 failure", MasterService.class.getCanonicalName(), Level.TRACE, "failed to execute cluster state update (on version: [*], uuid: [*]) for [test2]*"));
        mockAppender.addExpectation(new MockLogAppender.SeenEventExpectation("test2 computation", MasterService.class.getCanonicalName(), Level.DEBUG, "took [2s] to compute cluster state update for [test2]"));
        mockAppender.addExpectation(new MockLogAppender.SeenEventExpectation("test2 notification", MasterService.class.getCanonicalName(), Level.DEBUG, "took [0s] to notify listeners on unchanged cluster state for [test2]"));
        mockAppender.addExpectation(new MockLogAppender.SeenEventExpectation("test3 start", MasterService.class.getCanonicalName(), Level.DEBUG, "executing cluster state update for [test3]"));
        mockAppender.addExpectation(new MockLogAppender.SeenEventExpectation("test3 computation", MasterService.class.getCanonicalName(), Level.DEBUG, "took [3s] to compute cluster state update for [test3]"));
        mockAppender.addExpectation(new MockLogAppender.SeenEventExpectation("test3 notification", MasterService.class.getCanonicalName(), Level.DEBUG, "took [4s] to notify listeners on successful publication of cluster state (version: *, uuid: *) for [test3]"));
        mockAppender.addExpectation(new MockLogAppender.SeenEventExpectation("test4", MasterService.class.getCanonicalName(), Level.DEBUG, "executing cluster state update for [test4]"));
        try (MasterService clusterManagerService = createClusterManagerService(true)) {
            clusterManagerService.submitStateUpdateTask("test1", new ClusterStateUpdateTask() {

                @Override
                public ClusterState execute(ClusterState currentState) {
                    relativeTimeInMillis += TimeValue.timeValueSeconds(1).millis();
                    return currentState;
                }

                @Override
                public void clusterStateProcessed(String source, ClusterState oldState, ClusterState newState) {
                }

                @Override
                public void onFailure(String source, Exception e) {
                    fail();
                }
            });
            clusterManagerService.submitStateUpdateTask("test2", new ClusterStateUpdateTask() {

                @Override
                public ClusterState execute(ClusterState currentState) {
                    relativeTimeInMillis += TimeValue.timeValueSeconds(2).millis();
                    throw new IllegalArgumentException("Testing handling of exceptions in the cluster state task");
                }

                @Override
                public void clusterStateProcessed(String source, ClusterState oldState, ClusterState newState) {
                    fail();
                }

                @Override
                public void onFailure(String source, Exception e) {
                }
            });
            clusterManagerService.submitStateUpdateTask("test3", new ClusterStateUpdateTask() {

                @Override
                public ClusterState execute(ClusterState currentState) {
                    relativeTimeInMillis += TimeValue.timeValueSeconds(3).millis();
                    return ClusterState.builder(currentState).incrementVersion().build();
                }

                @Override
                public void clusterStateProcessed(String source, ClusterState oldState, ClusterState newState) {
                    relativeTimeInMillis += TimeValue.timeValueSeconds(4).millis();
                }

                @Override
                public void onFailure(String source, Exception e) {
                    fail();
                }
            });
            clusterManagerService.submitStateUpdateTask("test4", new ClusterStateUpdateTask() {

                @Override
                public ClusterState execute(ClusterState currentState) {
                    return currentState;
                }

                @Override
                public void clusterStateProcessed(String source, ClusterState oldState, ClusterState newState) {
                }

                @Override
                public void onFailure(String source, Exception e) {
                    fail();
                }
            });
            assertBusy(mockAppender::assertAllExpectationsMatched);
        }
    }
}