org.apache.activemq.artemis.logs.AssertionLoggerHandler Maven / Gradle / Ivy
/*
* 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.activemq.artemis.logs;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import java.util.logging.Level;
import org.apache.activemq.artemis.shaded.org.jboss.logmanager.ExtHandler;
import org.apache.activemq.artemis.shaded.org.jboss.logmanager.ExtLogRecord;
/**
* This class contains a tool where programs could intercept for LogMessage given an interval of time between {@link #startCapture()}
* and {@link #stopCapture()}
*
* Be careful with this use as this is intended for testing only (such as testcases)
*/
public class AssertionLoggerHandler extends ExtHandler {
private static final Map messages = new ConcurrentHashMap<>();
private static boolean capture = false;
/**
* {@inheritDoc}
*/
@Override
public void flush() {
}
/**
* {@inheritDoc}
*/
@Override
public void close() throws SecurityException {
}
@Override
protected void doPublish(final ExtLogRecord record) {
if (capture) {
messages.put(record.getFormattedMessage(), record);
}
}
/**
* is there any record matching Level?
*
* @param level
* @return
*/
public static boolean hasLevel(Level level) {
for (ExtLogRecord record : messages.values()) {
if (record.getLevel().equals(level)) {
return true;
}
}
return false;
}
public static boolean findText(long mstimeout, String... text) {
long timeMax = System.currentTimeMillis() + mstimeout;
do {
if (findText(text)) {
return true;
}
}
while (timeMax > System.currentTimeMillis());
return false;
}
/**
* Find a line that contains the parameters passed as an argument
*
* @param text
* @return
*/
public static boolean findText(final String... text) {
for (Map.Entry entry : messages.entrySet()) {
String key = entry.getKey();
boolean found = true;
for (String txtCheck : text) {
found = key.contains(txtCheck);
if (!found) {
// If the main log message doesn't contain what we're looking for let's look in the message from the exception (if there is one).
Throwable throwable = entry.getValue().getThrown();
if (throwable != null && throwable.getMessage() != null) {
found = throwable.getMessage().contains(txtCheck);
if (!found) {
break;
}
} else {
break;
}
}
}
if (found) {
return true;
}
}
return false;
}
public static final void clear() {
messages.clear();
}
public static final void startCapture() {
clear();
capture = true;
}
public static final void stopCapture() {
capture = false;
clear();
}
}