/**
* Logback: the reliable, generic, fast and flexible logging framework.
* Copyright (C) 1999-2015, QOS.ch. All rights reserved.
*
* This program and the accompanying materials are dual-licensed under
* either the terms of the Eclipse Public License v1.0 as published by
* the Eclipse Foundation
*
* or (per the licensee's choosing)
*
* under the terms of the GNU Lesser General Public License version 2.1
* as published by the Free Software Foundation.
*/
package ch.qos.logback.classic.db;
import ch.qos.logback.classic.LoggerContext;
import ch.qos.logback.classic.joran.JoranConfigurator;
import ch.qos.logback.core.db.DriverManagerConnectionSource;
import ch.qos.logback.core.joran.spi.JoranException;
import ch.qos.logback.core.status.StatusChecker;
import ch.qos.logback.core.testUtil.RandomUtil;
import ch.qos.logback.core.util.EnvUtil;
import ch.qos.logback.core.util.StatusPrinter;
import org.junit.*;
import org.slf4j.Logger;
import org.slf4j.MDC;
import java.net.InetAddress;
import java.sql.Connection;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;
import java.util.HashMap;
import java.util.Map;
import static org.junit.Assert.*;
public class DBAppenderIntegrationTest {
static String LOCAL_HOST_NAME;
static String[] CONFORMING_HOST_LIST = new String[] { "Orion" };
static String[] POSTGRES_CONFORMING_HOST_LIST = new String[] { "haro" };
static String[] MYSQL_CONFORMING_HOST_LIST = new String[] { "xharo" };
static String[] ORACLE_CONFORMING_HOST_LIST = new String[] { "xharo" };
int diff = RandomUtil.getPositiveInt();
LoggerContext lc = new LoggerContext();
@BeforeClass
public static void setUpBeforeClass() throws Exception {
InetAddress localhostIA = InetAddress.getLocalHost();
LOCAL_HOST_NAME = localhostIA.getHostName();
}
@AfterClass
public static void tearDownAfterClass() throws Exception {
}
@Before
public void setUp() throws Exception {
lc.setName("lc" + diff);
}
@After
public void tearDown() throws Exception {
// lc will never be used again
lc.stop();
}
DriverManagerConnectionSource getConnectionSource() {
ch.qos.logback.classic.Logger root = (ch.qos.logback.classic.Logger) lc.getLogger(Logger.ROOT_LOGGER_NAME);
DBAppender dbAppender = (DBAppender) root.getAppender("DB");
assertNotNull(dbAppender);
return (DriverManagerConnectionSource) dbAppender.getConnectionSource();
}
public void doTest(String configFile) throws JoranException, SQLException {
JoranConfigurator configurator = new JoranConfigurator();
configurator.setContext(lc);
configurator.doConfigure(configFile);
Logger logger = lc.getLogger(DBAppenderIntegrationTest.class);
// the key userid is used in SiftingAppender test
// suffix with diff to avoid collision
MDC.put("userid" + diff, "user" + diff);
int runLength = 5;
for (int i = 1; i <= runLength; i++) {
logger.debug("This is a debug message. Message number: " + (diff + i));
}
Exception e = new Exception("Just testing", getCause());
logger.error("At last an error.", e);
StatusPrinter.printInCaseOfErrorsOrWarnings(lc);
long lastEventId = getLastEventId();
verify(lastEventId);
// check that there were no errors
StatusChecker checker = new StatusChecker(lc);
checker.assertIsErrorFree();
}
long getLastEventId() throws SQLException {
DriverManagerConnectionSource cs = getConnectionSource();
Connection con = cs.getConnection();
Statement statement = con.createStatement();
statement.setMaxRows(1);
ResultSet rs = statement.executeQuery("select event_id from logging_event order by event_id desc");
rs.next();
long eventId = rs.getLong(1);
rs.close();
statement.close();
return eventId;
}
void verify(long lastEventId) throws SQLException {
verifyDebugMsg(lastEventId);
verifyException(lastEventId);
verifyProperty(lastEventId);
}
void verifyDebugMsg(long lastEventId) throws SQLException {
DriverManagerConnectionSource cs = getConnectionSource();
Connection con = cs.getConnection();
Statement statement = con.createStatement();
ResultSet rs = statement.executeQuery("select formatted_message from logging_event where event_id='" + (lastEventId - 1) + "'");
rs.next();
String msg = rs.getString(1);
assertEquals("This is a debug message. Message number: " + (diff + 5), msg);
}
@SuppressWarnings("unchecked")
void verifyProperty(long lastEventId) throws SQLException {
DriverManagerConnectionSource cs = getConnectionSource();
Connection con = cs.getConnection();
Statement statement = con.createStatement();
ResultSet rs = statement.executeQuery("select mapped_key, mapped_value from logging_event_property where event_id='" + (lastEventId - 1) + "'");
Map<String, String> witness = lc.getCopyOfPropertyMap();
witness.putAll(MDC.getCopyOfContextMap());
Map<String, String> map = new HashMap<String, String>();
while (rs.next()) {
String key = rs.getString(1);
String val = rs.getString(2);
map.put(key, val);
}
assertEquals(witness, map);
}
void verifyException(long lastEventId) throws SQLException {
DriverManagerConnectionSource cs = getConnectionSource();
Connection con = cs.getConnection();
Statement statement = con.createStatement();
ResultSet rs = statement.executeQuery("select trace_line from logging_event_exception where event_id='" + (lastEventId) + "' AND I='0'");
rs.next();
String traceLine = rs.getString(1);
assertEquals("java.lang.Exception: Just testing", traceLine);
}
Throwable getCause() {
return new IllegalStateException("test cause");
}
static boolean isConformingHostAndJDK16OrHigher(String[] conformingHostList) {
if (!EnvUtil.isJDK6OrHigher()) {
return false;
}
for (String conformingHost : conformingHostList) {
if (conformingHost.equalsIgnoreCase(LOCAL_HOST_NAME)) {
return true;
}
}
return false;
}
static boolean isConformingHostAndJDK16OrHigher() {
return isConformingHostAndJDK16OrHigher(CONFORMING_HOST_LIST);
}
@Test
public void sqlserver() throws Exception {
// perform test only on conforming hosts
if (!isConformingHostAndJDK16OrHigher()) {
return;
}
doTest("src/test/input/integration/db/sqlserver-with-driver.xml");
}
@Test
public void oracle10g() throws Exception {
// perform test only on conforming hosts
if (!isConformingHostAndJDK16OrHigher(ORACLE_CONFORMING_HOST_LIST)) {
return;
}
doTest("src/test/input/integration/db/oracle10g-with-driver.xml");
}
@Test
@Ignore
public void oracle11g() throws Exception {
// perform test only on conforming hosts
if (!isConformingHostAndJDK16OrHigher()) {
return;
}
doTest("src/test/input/integration/db/oracle11g-with-driver.xml");
}
@Test
public void mysql() throws Exception {
// perform test only on conforming hosts
if (!isConformingHostAndJDK16OrHigher(MYSQL_CONFORMING_HOST_LIST)) {
return;
}
doTest("src/test/input/integration/db/mysql-with-driver.xml");
}
@Test
public void postgres() throws Exception {
// perform test only on conforming hosts
if (!isConformingHostAndJDK16OrHigher(POSTGRES_CONFORMING_HOST_LIST)) {
return;
}
System.out.println("running postgres() test");
doTest("src/test/input/integration/db/postgresql-with-driver.xml");
}
}