Author: borges Date: 2011-06-17 11:52:20 -0400 (Fri, 17 Jun 2011) New Revision: 10848 Added: trunk/hornetq-core/src/test/ trunk/hornetq-core/src/test/java/ trunk/hornetq-core/src/test/java/org/ trunk/hornetq-core/src/test/java/org/hornetq/ trunk/hornetq-core/src/test/java/org/hornetq/core/ trunk/hornetq-core/src/test/java/org/hornetq/core/version/ trunk/hornetq-core/src/test/java/org/hornetq/core/version/impl/ trunk/hornetq-core/src/test/java/org/hornetq/core/version/impl/VersionImplTest.java trunk/hornetq-core/src/test/java/org/hornetq/tests/ trunk/hornetq-core/src/test/java/org/hornetq/tests/util/ trunk/hornetq-core/src/test/java/org/hornetq/tests/util/RandomUtil.java trunk/hornetq-core/src/test/java/org/hornetq/tests/util/UnitTestCase.java Removed: trunk/tests/unit-tests/src/test/java/org/hornetq/tests/unit/core/version/impl/VersionImplTest.java trunk/tests/unit-tests/src/test/java/org/hornetq/tests/util/RandomUtil.java trunk/tests/unit-tests/src/test/java/org/hornetq/tests/util/UnitTestCase.java Modified: trunk/hornetq-core/pom.xml trunk/tests/unit-tests/pom.xml Log: Move base TestCases to hornetq-core/src/test, and one actual test case. Modified: trunk/hornetq-core/pom.xml =================================================================== --- trunk/hornetq-core/pom.xml 2011-06-17 15:50:19 UTC (rev 10847) +++ trunk/hornetq-core/pom.xml 2011-06-17 15:52:20 UTC (rev 10848) @@ -27,6 +27,11 @@ <groupId>org.jboss.netty</groupId> <artifactId>netty</artifactId> </dependency> + <dependency> + <groupId>junit</groupId> + <artifactId>junit</artifactId> + <scope>test</scope> + </dependency> </dependencies> <profiles> Copied: trunk/hornetq-core/src/test/java/org/hornetq/core/version/impl/VersionImplTest.java (from rev 10847, trunk/tests/unit-tests/src/test/java/org/hornetq/tests/unit/core/version/impl/VersionImplTest.java) =================================================================== --- trunk/hornetq-core/src/test/java/org/hornetq/core/version/impl/VersionImplTest.java (rev 0) +++ trunk/hornetq-core/src/test/java/org/hornetq/core/version/impl/VersionImplTest.java 2011-06-17 15:52:20 UTC (rev 10848) @@ -0,0 +1,109 @@ +/* + * Copyright 2009 Red Hat, Inc. + * Red Hat 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.hornetq.core.version.impl; + +import java.io.ByteArrayInputStream; +import java.io.ByteArrayOutputStream; +import java.io.ObjectInputStream; +import java.io.ObjectOutputStream; + +import junit.framework.Assert; + +import org.hornetq.core.version.impl.VersionImpl; +import org.hornetq.tests.util.UnitTestCase; + +/** + * @author <a href="mailto:jmesnil@redhat.com">Jeff Mesnil</a> + * + * @version <tt>$Revision$</tt> + * + */ +public class VersionImplTest extends UnitTestCase +{ + // Constants ----------------------------------------------------- + + // Attributes ---------------------------------------------------- + + // Static -------------------------------------------------------- + + // Constructors -------------------------------------------------- + + // Public -------------------------------------------------------- + + public void testVersionImpl() throws Exception + { + + String versionName = "HORNETQ"; + int majorVersion = 2; + int minorVersion = 0; + int microVersion = 1; + int incrementingVersion = 10; + String versionSuffix = "suffix"; + String nettyVersion = "netty"; + int[] compatibleVersionList = {7,8,9,10}; + VersionImpl version = new VersionImpl(versionName, + majorVersion, + minorVersion, + microVersion, + incrementingVersion, + versionSuffix, + nettyVersion, + compatibleVersionList); + + Assert.assertEquals(versionName, version.getVersionName()); + Assert.assertEquals(majorVersion, version.getMajorVersion()); + Assert.assertEquals(minorVersion, version.getMinorVersion()); + Assert.assertEquals(microVersion, version.getMicroVersion()); + Assert.assertEquals(incrementingVersion, version.getIncrementingVersion()); + Assert.assertEquals(versionSuffix, version.getVersionSuffix()); + } + + public void testEquals() throws Exception + { + String nettyVersion = "netty"; + VersionImpl version = new VersionImpl("HORNETQ", 2, 0, 1, 10, "suffix", nettyVersion, new int[]{7,8,9,10}); + VersionImpl sameVersion = new VersionImpl("HORNETQ", 2, 0, 1, 10, "suffix", nettyVersion, new int[]{7,8,9,10}); + VersionImpl differentVersion = new VersionImpl("HORNETQ", 2, 0, 1, 11, "suffix", nettyVersion, new int[]{7,8,9,10,11}); + + Assert.assertFalse(version.equals(new Object())); + + Assert.assertTrue(version.equals(version)); + Assert.assertTrue(version.equals(sameVersion)); + Assert.assertFalse(version.equals(differentVersion)); + } + + public void testSerialize() throws Exception + { + String nettyVersion = "netty"; + VersionImpl version = new VersionImpl("uyiuy", 3, 7, 6, 12, "uhuhuh", nettyVersion, new int[]{9,10,11,12}); + ByteArrayOutputStream baos = new ByteArrayOutputStream(); + ObjectOutputStream oos = new ObjectOutputStream(baos); + oos.writeObject(version); + oos.flush(); + + ByteArrayInputStream bais = new ByteArrayInputStream(baos.toByteArray()); + ObjectInputStream ois = new ObjectInputStream(bais); + VersionImpl version2 = (VersionImpl)ois.readObject(); + + Assert.assertTrue(version.equals(version2)); + } + + // Package protected --------------------------------------------- + + // Protected ----------------------------------------------------- + + // Private ------------------------------------------------------- + + // Inner classes ------------------------------------------------- +} Copied: trunk/hornetq-core/src/test/java/org/hornetq/tests/util/RandomUtil.java (from rev 10847, trunk/tests/unit-tests/src/test/java/org/hornetq/tests/util/RandomUtil.java) =================================================================== --- trunk/hornetq-core/src/test/java/org/hornetq/tests/util/RandomUtil.java (rev 0) +++ trunk/hornetq-core/src/test/java/org/hornetq/tests/util/RandomUtil.java 2011-06-17 15:52:20 UTC (rev 10848) @@ -0,0 +1,176 @@ +/* + * Copyright 2009 Red Hat, Inc. + * Red Hat 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.hornetq.tests.util; + +import java.util.Random; +import java.util.UUID; + +import javax.transaction.xa.Xid; + +import org.hornetq.api.core.HornetQBuffer; +import org.hornetq.api.core.HornetQBuffers; +import org.hornetq.api.core.SimpleString; +import org.hornetq.core.transaction.impl.XidImpl; + +/** + * @author <a href="mailto:jmesnil@redhat.com">Jeff Mesnil</a> + * + * @version <tt>$Revision$</tt> + * + */ +public final class RandomUtil +{ + // Constants ----------------------------------------------------- + + private static final Random random = new Random(System.currentTimeMillis()); + + // Attributes ---------------------------------------------------- + + // Static -------------------------------------------------------- + + public static String randomString() + { + return UUID.randomUUID().toString(); + } + + public static SimpleString randomSimpleString() + { + return new SimpleString(RandomUtil.randomString()); + } + + public static char randomChar() + { + return RandomUtil.randomString().charAt(0); + } + + public static long randomLong() + { + return RandomUtil.random.nextLong(); + } + + public static long randomPositiveLong() + { + return Math.abs(RandomUtil.randomLong()); + } + + public static int randomInt() + { + return RandomUtil.random.nextInt(); + } + + public static int randomPositiveInt() + { + return Math.abs(RandomUtil.randomInt()); + } + + + public static HornetQBuffer randomBuffer(final int size, final long... data) + { + HornetQBuffer buffer = HornetQBuffers.fixedBuffer(size + 8 * data.length); + + for (long d : data) + { + buffer.writeLong(d); + } + + for (int i = 0 ; i < size; i++) + { + buffer.writeByte(randomByte()); + } + + return buffer; + } + + + + + public static int randomInterval(final int min, final int max) + { + return min + randomMax(max - min); + } + + public static int randomMax(final int max) + { + int value = randomPositiveInt() % max; + + if (value == 0) + { + value = max; + } + + return value; + } + + public static int randomPort() + { + return RandomUtil.random.nextInt(65536); + } + + public static short randomShort() + { + return (short)RandomUtil.random.nextInt(Short.MAX_VALUE); + } + + public static byte randomByte() + { + return Integer.valueOf(RandomUtil.random.nextInt()).byteValue(); + } + + public static boolean randomBoolean() + { + return RandomUtil.random.nextBoolean(); + } + + public static byte[] randomBytes() + { + return RandomUtil.randomString().getBytes(); + } + + public static byte[] randomBytes(final int length) + { + byte[] bytes = new byte[length]; + for (int i = 0; i < bytes.length; i++) + { + bytes[i] = RandomUtil.randomByte(); + } + return bytes; + } + + public static double randomDouble() + { + return RandomUtil.random.nextDouble(); + } + + public static float randomFloat() + { + return RandomUtil.random.nextFloat(); + } + + public static Xid randomXid() + { + return new XidImpl(RandomUtil.randomBytes(), RandomUtil.randomInt(), RandomUtil.randomBytes()); + } + + // Constructors -------------------------------------------------- + + // Public -------------------------------------------------------- + + // Package protected --------------------------------------------- + + // Protected ----------------------------------------------------- + + // Private ------------------------------------------------------- + + // Inner classes ------------------------------------------------- +} Copied: trunk/hornetq-core/src/test/java/org/hornetq/tests/util/UnitTestCase.java (from rev 10847, trunk/tests/unit-tests/src/test/java/org/hornetq/tests/util/UnitTestCase.java) =================================================================== --- trunk/hornetq-core/src/test/java/org/hornetq/tests/util/UnitTestCase.java (rev 0) +++ trunk/hornetq-core/src/test/java/org/hornetq/tests/util/UnitTestCase.java 2011-06-17 15:52:20 UTC (rev 10848) @@ -0,0 +1,1229 @@ +/* + * Copyright 2009 Red Hat, Inc. + * Red Hat 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.hornetq.tests.util; + +import java.beans.BeanInfo; +import java.beans.Introspector; +import java.beans.PropertyDescriptor; +import java.io.BufferedInputStream; +import java.io.BufferedOutputStream; +import java.io.File; +import java.io.FileInputStream; +import java.io.FileOutputStream; +import java.io.IOException; +import java.io.InputStream; +import java.io.OutputStream; +import java.io.PrintWriter; +import java.io.StringWriter; +import java.lang.ref.Reference; +import java.lang.ref.WeakReference; +import java.net.ServerSocket; +import java.nio.ByteBuffer; +import java.util.ArrayList; +import java.util.HashMap; +import java.util.HashSet; +import java.util.Iterator; +import java.util.List; +import java.util.Map; +import java.util.Set; +import java.util.UUID; + +import javax.naming.Context; +import javax.transaction.xa.XAException; +import javax.transaction.xa.Xid; + +import junit.framework.Assert; +import junit.framework.TestCase; +import junit.framework.TestSuite; + +import org.hornetq.api.core.HornetQBuffer; +import org.hornetq.api.core.HornetQException; +import org.hornetq.api.core.SimpleString; +import org.hornetq.api.core.TransportConfiguration; +import org.hornetq.core.asyncio.impl.AsynchronousFileImpl; +import org.hornetq.core.client.impl.ServerLocatorImpl; +import org.hornetq.core.config.Configuration; +import org.hornetq.core.config.impl.ConfigurationImpl; +import org.hornetq.core.journal.impl.AIOSequentialFileFactory; +import org.hornetq.core.logging.Logger; +import org.hornetq.core.persistence.impl.journal.OperationContextImpl; +import org.hornetq.core.postoffice.Binding; +import org.hornetq.core.postoffice.Bindings; +import org.hornetq.core.postoffice.PostOffice; +import org.hornetq.core.postoffice.QueueBinding; +import org.hornetq.core.postoffice.impl.LocalQueueBinding; +import org.hornetq.core.remoting.impl.invm.InVMAcceptorFactory; +import org.hornetq.core.remoting.impl.invm.InVMConnectorFactory; +import org.hornetq.core.remoting.impl.invm.InVMRegistry; +import org.hornetq.core.remoting.impl.netty.NettyAcceptorFactory; +import org.hornetq.core.remoting.impl.netty.NettyConnectorFactory; +import org.hornetq.core.server.HornetQServer; +import org.hornetq.core.server.JournalType; +import org.hornetq.core.server.MessageReference; +import org.hornetq.core.server.Queue; +import org.hornetq.core.server.ServerMessage; +import org.hornetq.core.server.impl.ServerMessageImpl; +import org.hornetq.core.transaction.impl.XidImpl; +import org.hornetq.utils.UUIDGenerator; + +/** + * + * Helper base class for our unit tests + * + * @author <a href="mailto:tim.fox@jboss.com">Tim Fox</a> + * @author <a href="mailto:csuconic@redhat.com">Clebert</a> + * + */ +public abstract class UnitTestCase extends TestCase +{ + // Constants ----------------------------------------------------- + + private static final Logger log = Logger.getLogger(UnitTestCase.class); + + public static final String INVM_ACCEPTOR_FACTORY = InVMAcceptorFactory.class.getCanonicalName(); + + public static final String INVM_CONNECTOR_FACTORY = InVMConnectorFactory.class.getCanonicalName(); + + public static final String NETTY_ACCEPTOR_FACTORY = NettyAcceptorFactory.class.getCanonicalName(); + + public static final String NETTY_CONNECTOR_FACTORY = NettyConnectorFactory.class.getCanonicalName(); + + // Attributes ---------------------------------------------------- + + private static final String testDir = System.getProperty("java.io.tmpdir", "/tmp") + "/hornetq-unit-test"; + + // There is a verification about thread leakages. We only fail a single thread when this happens + private static Set<Thread> alreadyFailedThread = new HashSet<Thread>(); + + // Static -------------------------------------------------------- + + protected Configuration createDefaultConfig() + { + return createDefaultConfig(false); + } + + protected Configuration createDefaultConfig(final boolean netty) + { + if (netty) + { + return createDefaultConfig(new HashMap<String, Object>(), INVM_ACCEPTOR_FACTORY, NETTY_ACCEPTOR_FACTORY); + } + else + { + return createDefaultConfig(new HashMap<String, Object>(), INVM_ACCEPTOR_FACTORY); + } + } + + protected static Configuration createClusteredDefaultConfig(final int index, + final Map<String, Object> params, + final String... acceptors) + { + Configuration config = createDefaultConfig(index, params, acceptors); + + config.setClustered(true); + + return config; + } + + protected static Configuration createDefaultConfig(final int index, + final Map<String, Object> params, + final String... acceptors) + { + Configuration configuration = createBasicConfig(index); + + configuration.getAcceptorConfigurations().clear(); + + for (String acceptor : acceptors) + { + TransportConfiguration transportConfig = new TransportConfiguration(acceptor, params); + configuration.getAcceptorConfigurations().add(transportConfig); + } + + return configuration; + } + + protected static ConfigurationImpl createBasicConfig() + { + return createBasicConfig(0); + } + + /** + * @param serverID + * @return + */ + protected static ConfigurationImpl createBasicConfig(final int serverID) + { + ConfigurationImpl configuration = new ConfigurationImpl(); + configuration.setSecurityEnabled(false); + configuration.setBindingsDirectory(getBindingsDir(serverID, false)); + configuration.setJournalMinFiles(2); + configuration.setJournalDirectory(getJournalDir(serverID, false)); + configuration.setJournalFileSize(100 * 1024); + configuration.setJournalType(getDefaultJournalType()); + configuration.setPagingDirectory(getPageDir(serverID, false)); + configuration.setLargeMessagesDirectory(getLargeMessagesDir(serverID, false)); + configuration.setJournalCompactMinFiles(0); + configuration.setJournalCompactPercentage(0); + return configuration; + } + + protected static Configuration createDefaultConfig(final Map<String, Object> params, final String... acceptors) + { + Configuration configuration = new ConfigurationImpl(); + configuration.setSecurityEnabled(false); + configuration.setJMXManagementEnabled(false); + configuration.setBindingsDirectory(getBindingsDir()); + configuration.setJournalMinFiles(2); + configuration.setJournalDirectory(getJournalDir()); + configuration.setJournalFileSize(100 * 1024); + configuration.setPagingDirectory(getPageDir()); + configuration.setLargeMessagesDirectory(getLargeMessagesDir()); + configuration.setJournalCompactMinFiles(0); + configuration.setJournalCompactPercentage(0); + + configuration.setFileDeploymentEnabled(false); + + configuration.setJournalType(getDefaultJournalType()); + + configuration.getAcceptorConfigurations().clear(); + + for (String acceptor : acceptors) + { + TransportConfiguration transportConfig = new TransportConfiguration(acceptor, params); + configuration.getAcceptorConfigurations().add(transportConfig); + } + + return configuration; + } + + protected static String getUDPDiscoveryAddress() + { + return System.getProperty("TEST-UDP-ADDRESS", "230.1.2.3"); + } + + protected static String getUDPDiscoveryAddress(final int variant) + { + String value = getUDPDiscoveryAddress(); + + int posPoint = value.lastIndexOf('.'); + + int last = Integer.valueOf(value.substring(posPoint + 1)); + + return value.substring(0, posPoint + 1) + (last + variant); + } + + public static int getUDPDiscoveryPort() + { + return Integer.parseInt(System.getProperty("TEST-UDP-PORT", "6750")); + } + + public static int getUDPDiscoveryPort(final int variant) + { + return getUDPDiscoveryPort() + variant; + } + + protected static JournalType getDefaultJournalType() + { + if (AsynchronousFileImpl.isLoaded()) + { + return JournalType.ASYNCIO; + } + else + { + return JournalType.NIO; + } + } + + /** + * @param name + */ + public UnitTestCase(final String name) + { + super(name); + } + + public UnitTestCase() + { + super(); + } + + public static void forceGC() + { + WeakReference<Object> dumbReference = new WeakReference<Object>(new Object()); + // A loop that will wait GC, using the minimal time as possible + while (dumbReference.get() != null) + { + System.gc(); + try + { + Thread.sleep(500); + } + catch (InterruptedException e) + { + } + } + } + + public static void forceGC(final Reference<?> ref, final long timeout) + { + long waitUntil = System.currentTimeMillis() + timeout; + // A loop that will wait GC, using the minimal time as possible + while (ref.get() != null && System.currentTimeMillis() < waitUntil) + { + ArrayList<String> list = new ArrayList<String>(); + for (int i = 0; i < 1000; i++) + { + list.add("Some string with garbage with concatenation " + i); + } + list.clear(); + list = null; + System.gc(); + try + { + Thread.sleep(500); + } + catch (InterruptedException e) + { + } + } + } + + // verify if these weak references are released after a few GCs + public static void checkWeakReferences(final WeakReference<?>... references) + { + + int i = 0; + boolean hasValue = false; + + do + { + hasValue = false; + + if (i > 0) + { + UnitTestCase.forceGC(); + } + + for (WeakReference<?> ref : references) + { + if (ref.get() != null) + { + hasValue = true; + } + } + } + while (i++ <= 30 && hasValue); + + for (WeakReference<?> ref : references) + { + Assert.assertNull(ref.get()); + } + } + + public static String threadDump(final String msg) + { + StringWriter str = new StringWriter(); + PrintWriter out = new PrintWriter(str); + + Map<Thread, StackTraceElement[]> stackTrace = Thread.getAllStackTraces(); + + out.println("*******************************************************************************"); + out.println("Complete Thread dump " + msg); + + for (Map.Entry<Thread, StackTraceElement[]> el : stackTrace.entrySet()) + { + out.println("==============================================================================="); + out.println("Thread " + el.getKey() + + " name = " + + el.getKey().getName() + + " id = " + + el.getKey().getId() + + " group = " + + el.getKey().getThreadGroup()); + out.println(); + for (StackTraceElement traceEl : el.getValue()) + { + out.println(traceEl); + } + } + + out.println("==============================================================================="); + out.println("End Thread dump " + msg); + out.println("*******************************************************************************"); + + return str.toString(); + } + + protected static TestSuite createAIOTestSuite(final Class<?> clazz) + { + TestSuite suite = new TestSuite(clazz.getName() + " testsuite"); + + if (AIOSequentialFileFactory.isSupported()) + { + suite.addTestSuite(clazz); + } + else + { + // System.out goes towards JUnit report + System.out.println("Test " + clazz.getName() + " ignored as AIO is not available"); + } + + return suite; + } + + public static String dumpBytes(final byte[] bytes) + { + StringBuffer buff = new StringBuffer(); + + buff.append(System.identityHashCode(bytes) + ", size: " + bytes.length + " ["); + + for (int i = 0; i < bytes.length; i++) + { + buff.append(bytes[i]); + + if (i != bytes.length - 1) + { + buff.append(", "); + } + } + + buff.append("]"); + + return buff.toString(); + } + + public static String dumbBytesHex(final byte[] buffer, final int bytesPerLine) + { + + StringBuffer buff = new StringBuffer(); + + buff.append("["); + + for (int i = 0; i < buffer.length; i++) + { + buff.append(String.format("%1$2X", buffer[i])); + if (i + 1 < buffer.length) + { + buff.append(", "); + } + if ((i + 1) % bytesPerLine == 0) + { + buff.append("\n "); + } + } + buff.append("]"); + + return buff.toString(); + } + + public static void assertEqualsByteArrays(final byte[] expected, final byte[] actual) + { + // assertEquals(expected.length, actual.length); + for (int i = 0; i < expected.length; i++) + { + Assert.assertEquals("byte at index " + i, expected[i], actual[i]); + } + } + + public static void assertEqualsTransportConfigurations(final TransportConfiguration[] expected, + final TransportConfiguration[] actual) + { + assertEquals(expected.length, actual.length); + for (int i = 0; i < expected.length; i++) + { + Assert.assertEquals("TransportConfiguration at index " + i, expected[i], actual[i]); + } + } + + public static void assertEqualsBuffers(final int size, final HornetQBuffer expected, final HornetQBuffer actual) + { + // assertEquals(expected.length, actual.length); + expected.readerIndex(0); + actual.readerIndex(0); + + for (int i = 0; i < size; i++) + { + byte b1 = expected.readByte(); + byte b2 = actual.readByte(); + Assert.assertEquals("byte at index " + i, b1, b2); + } + expected.resetReaderIndex(); + actual.resetReaderIndex(); + } + + public static void assertEqualsByteArrays(final int length, final byte[] expected, final byte[] actual) + { + // we check only for the given length (the arrays might be + // larger) + Assert.assertTrue(expected.length >= length); + Assert.assertTrue(actual.length >= length); + for (int i = 0; i < length; i++) + { + Assert.assertEquals("byte at index " + i, expected[i], actual[i]); + } + } + + public static void assertSameXids(final List<Xid> expected, final List<Xid> actual) + { + Assert.assertNotNull(expected); + Assert.assertNotNull(actual); + Assert.assertEquals(expected.size(), actual.size()); + + for (int i = 0; i < expected.size(); i++) + { + Xid expectedXid = expected.get(i); + Xid actualXid = actual.get(i); + UnitTestCase.assertEqualsByteArrays(expectedXid.getBranchQualifier(), actualXid.getBranchQualifier()); + Assert.assertEquals(expectedXid.getFormatId(), actualXid.getFormatId()); + UnitTestCase.assertEqualsByteArrays(expectedXid.getGlobalTransactionId(), actualXid.getGlobalTransactionId()); + } + } + + protected static void checkNoBinding(final Context context, final String binding) + { + try + { + context.lookup(binding); + Assert.fail("there must be no resource to look up for " + binding); + } + catch (Exception e) + { + } + } + + protected static Object checkBinding(final Context context, final String binding) throws Exception + { + Object o = context.lookup(binding); + Assert.assertNotNull(o); + return o; + } + + /** + * @param connectorConfigs + * @return + */ + protected ArrayList<String> registerConnectors(final HornetQServer server, + final List<TransportConfiguration> connectorConfigs) + { + // The connectors need to be pre-configured at main config object but this method is taking + // TransportConfigurations directly + // So this will first register them at the config and then generate a list of objects + ArrayList<String> connectors = new ArrayList<String>(); + for (TransportConfiguration tnsp : connectorConfigs) + { + String name = RandomUtil.randomString(); + + server.getConfiguration().getConnectorConfigurations().put(name, tnsp); + + connectors.add(name); + } + return connectors; + } + + protected static void checkFreePort(final int... ports) + { + for (int port : ports) + { + ServerSocket ssocket = null; + try + { + ssocket = new ServerSocket(port); + } + catch (Exception e) + { + throw new IllegalStateException("port " + port + " is already bound"); + } + finally + { + if (ssocket != null) + { + try + { + ssocket.close(); + } + catch (IOException e) + { + } + } + } + } + } + + // Constructors -------------------------------------------------- + + // Protected ----------------------------------------------------- + + /** + * @return the testDir + */ + protected String getTestDir() + { + return testDir; + } + + protected void clearData() + { + clearData(getTestDir()); + } + + protected void clearData(final String testDir) + { + // Need to delete the root + + File file = new File(testDir); + deleteDirectory(file); + file.mkdirs(); + + recreateDirectory(getJournalDir(testDir)); + recreateDirectory(getBindingsDir(testDir)); + recreateDirectory(getPageDir(testDir)); + recreateDirectory(getLargeMessagesDir(testDir)); + recreateDirectory(getClientLargeMessagesDir(testDir)); + recreateDirectory(getTemporaryDir(testDir)); + + System.out.println("deleted " + testDir); + } + + /** + * @return the journalDir + */ + public static String getJournalDir() + { + return getJournalDir(testDir); + } + + protected static String getJournalDir(final String testDir) + { + return testDir + "/journal"; + } + + protected static String getJournalDir(final int index, final boolean backup) + { + String dir = getJournalDir(testDir) + index + "-" + (backup ? "B" : "L"); + + return dir; + } + + /** + * @return the bindingsDir + */ + protected static String getBindingsDir() + { + return getBindingsDir(testDir); + } + + /** + * @return the bindingsDir + */ + protected static String getBindingsDir(final String testDir) + { + return testDir + "/bindings"; + } + + /** + * @return the bindingsDir + */ + protected static String getBindingsDir(final int index, final boolean backup) + { + return getBindingsDir(testDir) + index + "-" + (backup ? "B" : "L"); + } + + /** + * @return the pageDir + */ + protected static String getPageDir() + { + return getPageDir(testDir); + } + + /** + * @return the pageDir + */ + protected static String getPageDir(final String testDir) + { + return testDir + "/page"; + } + + protected static String getPageDir(final int index, final boolean backup) + { + return getPageDir(testDir) + index + "-" + (backup ? "B" : "L"); + } + + /** + * @return the largeMessagesDir + */ + protected static String getLargeMessagesDir() + { + return getLargeMessagesDir(testDir); + } + + /** + * @return the largeMessagesDir + */ + protected static String getLargeMessagesDir(final String testDir) + { + return testDir + "/large-msg"; + } + + protected static String getLargeMessagesDir(final int index, final boolean backup) + { + return getLargeMessagesDir(testDir) + index + "-" + (backup ? "B" : "L"); + } + + /** + * @return the clientLargeMessagesDir + */ + protected String getClientLargeMessagesDir() + { + return getClientLargeMessagesDir(testDir); + } + + /** + * @return the clientLargeMessagesDir + */ + protected String getClientLargeMessagesDir(final String testDir) + { + return testDir + "/client-large-msg"; + } + + /** + * @return the temporaryDir + */ + protected String getTemporaryDir() + { + return getTemporaryDir(testDir); + } + + /** + * @return the temporaryDir + */ + protected String getTemporaryDir(final String testDir) + { + return testDir + "/temp"; + } + + protected static void expectHornetQException(final String message, final int errorCode, final HornetQAction action) + { + try + { + action.run(); + Assert.fail(message); + } + catch (Exception e) + { + Assert.assertTrue(e instanceof HornetQException); + Assert.assertEquals(errorCode, ((HornetQException)e).getCode()); + } + } + + protected static void expectHornetQException(final int errorCode, final HornetQAction action) + { + UnitTestCase.expectHornetQException("must throw a HornetQException with the expected errorCode: " + errorCode, + errorCode, + action); + } + + protected static void expectXAException(final int errorCode, final HornetQAction action) + { + try + { + action.run(); + Assert.fail("must throw a XAException with the expected errorCode: " + errorCode); + } + catch (Exception e) + { + Assert.assertTrue(e instanceof XAException); + Assert.assertEquals(errorCode, ((XAException)e).errorCode); + } + } + + public static byte getSamplebyte(final long position) + { + return (byte)('a' + position % ('z' - 'a' + 1)); + } + + // Creates a Fake LargeStream without using a real file + public static InputStream createFakeLargeStream(final long size) throws Exception + { + return new InputStream() + { + private long count; + + private boolean closed = false; + + @Override + public void close() throws IOException + { + super.close(); + closed = true; + } + + @Override + public int read() throws IOException + { + if (closed) + { + throw new IOException("Stream was closed"); + } + if (count++ < size) + { + return UnitTestCase.getSamplebyte(count - 1); + } + else + { + return -1; + } + } + }; + + } + + /** It validates a Bean (POJO) using simple setters and getters with random values. + * You can pass a list of properties to be ignored, as some properties will have a pre-defined domain (not being possible to use random-values on them) */ + protected void validateGettersAndSetters(final Object pojo, final String... ignoredProperties) throws Exception + { + HashSet<String> ignoreSet = new HashSet<String>(); + + for (String ignore : ignoredProperties) + { + ignoreSet.add(ignore); + } + + BeanInfo info = Introspector.getBeanInfo(pojo.getClass()); + + PropertyDescriptor properties[] = info.getPropertyDescriptors(); + + for (PropertyDescriptor prop : properties) + { + Object value; + + if (prop.getPropertyType() == String.class) + { + value = RandomUtil.randomString(); + } + else if (prop.getPropertyType() == Integer.class || prop.getPropertyType() == Integer.TYPE) + { + value = RandomUtil.randomInt(); + } + else if (prop.getPropertyType() == Long.class || prop.getPropertyType() == Long.TYPE) + { + value = RandomUtil.randomLong(); + } + else if (prop.getPropertyType() == Boolean.class || prop.getPropertyType() == Boolean.TYPE) + { + value = RandomUtil.randomBoolean(); + } + else if (prop.getPropertyType() == Double.class || prop.getPropertyType() == Double.TYPE) + { + value = RandomUtil.randomDouble(); + } + else + { + System.out.println("Can't validate property of type " + prop.getPropertyType() + " on " + prop.getName()); + value = null; + } + + if (value != null && prop.getWriteMethod() != null && prop.getReadMethod() == null) + { + System.out.println("WriteOnly property " + prop.getName() + " on " + pojo.getClass()); + } + else if (value != null & prop.getWriteMethod() != null && + prop.getReadMethod() != null && + !ignoreSet.contains(prop.getName())) + { + System.out.println("Validating " + prop.getName() + " type = " + prop.getPropertyType()); + prop.getWriteMethod().invoke(pojo, value); + + Assert.assertEquals("Property " + prop.getName(), value, prop.getReadMethod().invoke(pojo)); + } + } + } + + // Package protected --------------------------------------------- + + // Protected ----------------------------------------------------- + + Map<Thread, StackTraceElement[]> previousThreads; + + @Override + protected void setUp() throws Exception + { + super.setUp(); + + OperationContextImpl.clearContext(); + + deleteDirectory(new File(getTestDir())); + + InVMRegistry.instance.clear(); + + // checkFreePort(TransportConstants.DEFAULT_PORT); + + previousThreads = Thread.getAllStackTraces(); + + UnitTestCase.log.info("###### starting test " + this.getClass().getName() + "." + getName()); + } + + @Override + protected void tearDown() throws Exception + { + cleanupPools(); + + Map<Thread, StackTraceElement[]> threadMap = Thread.getAllStackTraces(); + for (Thread thread : threadMap.keySet()) + { + StackTraceElement[] stack = threadMap.get(thread); + for (StackTraceElement stackTraceElement : stack) + { + if (stackTraceElement.getMethodName().contains("getConnectionWithRetry") && !alreadyFailedThread.contains(thread)) + { + alreadyFailedThread.add(thread); + System.out.println(threadDump(getName() + " has left threads running. Look at thread " + + thread.getName() + + " id = " + + thread.getId() + + " has running locators on test " + + getName() + + " on this following dump")); + fail("test left serverlocator running, this could effect other tests"); + // System.exit(0); + } + else if (stackTraceElement.getMethodName().contains("BroadcastGroupImpl.run") && !alreadyFailedThread.contains(thread)) + { + alreadyFailedThread.add(thread); + System.out.println(threadDump(getName() + " has left threads running. Look at thread " + + thread.getName() + + " id = " + + thread.getId() + + " is still broadcasting " + + getName() + + " on this following dump")); + fail("test left broadcastgroupimpl running, this could effect other tests"); + // System.exit(0); + } + } + } + + Map<Thread, StackTraceElement[]> postThreads = Thread.getAllStackTraces(); + + if (postThreads.size() > previousThreads.size()) + { + StringBuffer buffer = new StringBuffer(); + + + buffer.append("*********************************************************************************\n"); + buffer.append("LEAKING THREADS\n"); + + for (Thread aliveThread : postThreads.keySet()) + { + if (!aliveThread.getName().contains("SunPKCS11") && !previousThreads.containsKey(aliveThread)) + { + buffer.append("=============================================================================\n"); + buffer.append("Thread " + aliveThread + " is still alive with the following stackTrace:\n"); + StackTraceElement[] elements = postThreads.get(aliveThread); + for (StackTraceElement el : elements) + { + buffer.append(el + "\n"); + } + } + + } + buffer.append("*********************************************************************************\n"); + + System.out.println(buffer.toString()); + + } + + super.tearDown(); + } + + /** + * + */ + protected void cleanupPools() + { + OperationContextImpl.clearContext(); + + deleteDirectory(new File(getTestDir())); + + int invmSize = InVMRegistry.instance.size(); + if (invmSize > 0) + { + InVMRegistry.instance.clear(); + fail("invm registry still had acceptors registered"); + } + + if (AsynchronousFileImpl.getTotalMaxIO() != 0) + { + AsynchronousFileImpl.resetMaxAIO(); + Assert.fail("test did not close all its files " + AsynchronousFileImpl.getTotalMaxIO()); + } + + // We shutdown the global pools to give a better isolation between tests + ServerLocatorImpl.clearThreadPools(); + } + + protected byte[] autoEncode(final Object... args) + { + + int size = 0; + + for (Object arg : args) + { + if (arg instanceof Byte) + { + size++; + } + else if (arg instanceof Boolean) + { + size++; + } + else if (arg instanceof Integer) + { + size += 4; + } + else if (arg instanceof Long) + { + size += 8; + } + else if (arg instanceof Float) + { + size += 4; + } + else if (arg instanceof Double) + { + size += 8; + } + else + { + throw new IllegalArgumentException("method autoEncode doesn't know how to convert " + arg.getClass() + + " yet"); + } + } + + ByteBuffer buffer = ByteBuffer.allocate(size); + + for (Object arg : args) + { + if (arg instanceof Byte) + { + buffer.put(((Byte)arg).byteValue()); + } + else if (arg instanceof Boolean) + { + Boolean b = (Boolean)arg; + buffer.put((byte)(b.booleanValue() ? 1 : 0)); + } + else if (arg instanceof Integer) + { + buffer.putInt(((Integer)arg).intValue()); + } + else if (arg instanceof Long) + { + buffer.putLong(((Long)arg).longValue()); + } + else if (arg instanceof Float) + { + buffer.putFloat(((Float)arg).floatValue()); + } + else if (arg instanceof Double) + { + buffer.putDouble(((Double)arg).doubleValue()); + } + else + { + throw new IllegalArgumentException("method autoEncode doesn't know how to convert " + arg.getClass() + + " yet"); + } + } + + return buffer.array(); + } + + protected void recreateDirectory(final String directory) + { + File file = new File(directory); + deleteDirectory(file); + file.mkdirs(); + } + + protected boolean deleteDirectory(final File directory) + { + if (directory.isDirectory()) + { + String[] files = directory.list(); + + for (int j = 0; j < files.length; j++) + { + if (!deleteDirectory(new File(directory, files[j]))) + { + return false; + } + } + } + + return directory.delete(); + } + + protected void copyRecursive(final File from, final File to) throws Exception + { + if (from.isDirectory()) + { + if (!to.exists()) + { + to.mkdir(); + } + + String[] subs = from.list(); + + for (String sub : subs) + { + copyRecursive(new File(from, sub), new File(to, sub)); + } + } + else + { + InputStream in = null; + + OutputStream out = null; + + try + { + in = new BufferedInputStream(new FileInputStream(from)); + + out = new BufferedOutputStream(new FileOutputStream(to)); + + int b; + + while ((b = in.read()) != -1) + { + out.write(b); + } + } + finally + { + if (in != null) + { + in.close(); + } + + if (out != null) + { + out.close(); + } + } + } + } + + protected void assertRefListsIdenticalRefs(final List<MessageReference> l1, final List<MessageReference> l2) + { + if (l1.size() != l2.size()) + { + Assert.fail("Lists different sizes: " + l1.size() + ", " + l2.size()); + } + + Iterator<MessageReference> iter1 = l1.iterator(); + Iterator<MessageReference> iter2 = l2.iterator(); + + while (iter1.hasNext()) + { + MessageReference o1 = iter1.next(); + MessageReference o2 = iter2.next(); + + Assert.assertTrue("expected " + o1 + " but was " + o2, o1 == o2); + } + } + + protected ServerMessage generateMessage(final long id) + { + ServerMessage message = new ServerMessageImpl(id, 1000); + + message.setMessageID(id); + + message.getBodyBuffer().writeString(UUID.randomUUID().toString()); + + message.setAddress(new SimpleString("foo")); + + return message; + } + + protected MessageReference generateReference(final Queue queue, final long id) + { + ServerMessage message = generateMessage(id); + + return message.createReference(queue); + } + + protected int calculateRecordSize(final int size, final int alignment) + { + return (size / alignment + (size % alignment != 0 ? 1 : 0)) * alignment; + } + + protected XidImpl newXID() + { + return new XidImpl("xa1".getBytes(), 1, UUIDGenerator.getInstance().generateStringUUID().getBytes()); + } + + protected int getMessageCount(final HornetQServer service, final String address) throws Exception + { + return getMessageCount(service.getPostOffice(), address); + } + + /** + * @param address + * @param postOffice + * @return + * @throws Exception + */ + protected int getMessageCount(final PostOffice postOffice, final String address) throws Exception + { + int messageCount = 0; + + List<QueueBinding> bindings = getLocalQueueBindings(postOffice, address); + + for (QueueBinding qBinding : bindings) + { + messageCount += qBinding.getQueue().getMessageCount(); + } + + return messageCount; + } + + protected + List<QueueBinding> + getLocalQueueBindings(final PostOffice postOffice, final String address) throws Exception + { + ArrayList<QueueBinding> bindingsFound = new ArrayList<QueueBinding>(); + + Bindings bindings = postOffice.getBindingsForAddress(new SimpleString(address)); + + for (Binding binding : bindings.getBindings()) + { + if (binding instanceof LocalQueueBinding) + { + bindingsFound.add((QueueBinding)binding); + } + } + return bindingsFound; + } + + // Private ------------------------------------------------------- + + // Inner classes ------------------------------------------------- + + protected static interface HornetQAction + { + void run() throws Exception; + } + +} Modified: trunk/tests/unit-tests/pom.xml =================================================================== --- trunk/tests/unit-tests/pom.xml 2011-06-17 15:50:19 UTC (rev 10847) +++ trunk/tests/unit-tests/pom.xml 2011-06-17 15:52:20 UTC (rev 10848) @@ -16,6 +16,8 @@ <groupId>org.hornetq</groupId> <artifactId>hornetq-core</artifactId> <version>${project.version}</version> + <scope>test</scope> + <type>test-jar</type> </dependency> <dependency> <groupId>org.hornetq</groupId> Deleted: trunk/tests/unit-tests/src/test/java/org/hornetq/tests/unit/core/version/impl/VersionImplTest.java =================================================================== --- trunk/tests/unit-tests/src/test/java/org/hornetq/tests/unit/core/version/impl/VersionImplTest.java 2011-06-17 15:50:19 UTC (rev 10847) +++ trunk/tests/unit-tests/src/test/java/org/hornetq/tests/unit/core/version/impl/VersionImplTest.java 2011-06-17 15:52:20 UTC (rev 10848) @@ -1,109 +0,0 @@ -/* - * Copyright 2009 Red Hat, Inc. - * Red Hat 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.hornetq.tests.unit.core.version.impl; - -import java.io.ByteArrayInputStream; -import java.io.ByteArrayOutputStream; -import java.io.ObjectInputStream; -import java.io.ObjectOutputStream; - -import junit.framework.Assert; - -import org.hornetq.core.version.impl.VersionImpl; -import org.hornetq.tests.util.UnitTestCase; - -/** - * @author <a href="mailto:jmesnil@redhat.com">Jeff Mesnil</a> - * - * @version <tt>$Revision$</tt> - * - */ -public class VersionImplTest extends UnitTestCase -{ - // Constants ----------------------------------------------------- - - // Attributes ---------------------------------------------------- - - // Static -------------------------------------------------------- - - // Constructors -------------------------------------------------- - - // Public -------------------------------------------------------- - - public void testVersionImpl() throws Exception - { - - String versionName = "HORNETQ"; - int majorVersion = 2; - int minorVersion = 0; - int microVersion = 1; - int incrementingVersion = 10; - String versionSuffix = "suffix"; - String nettyVersion = "netty"; - int[] compatibleVersionList = {7,8,9,10}; - VersionImpl version = new VersionImpl(versionName, - majorVersion, - minorVersion, - microVersion, - incrementingVersion, - versionSuffix, - nettyVersion, - compatibleVersionList); - - Assert.assertEquals(versionName, version.getVersionName()); - Assert.assertEquals(majorVersion, version.getMajorVersion()); - Assert.assertEquals(minorVersion, version.getMinorVersion()); - Assert.assertEquals(microVersion, version.getMicroVersion()); - Assert.assertEquals(incrementingVersion, version.getIncrementingVersion()); - Assert.assertEquals(versionSuffix, version.getVersionSuffix()); - } - - public void testEquals() throws Exception - { - String nettyVersion = "netty"; - VersionImpl version = new VersionImpl("HORNETQ", 2, 0, 1, 10, "suffix", nettyVersion, new int[]{7,8,9,10}); - VersionImpl sameVersion = new VersionImpl("HORNETQ", 2, 0, 1, 10, "suffix", nettyVersion, new int[]{7,8,9,10}); - VersionImpl differentVersion = new VersionImpl("HORNETQ", 2, 0, 1, 11, "suffix", nettyVersion, new int[]{7,8,9,10,11}); - - Assert.assertFalse(version.equals(new Object())); - - Assert.assertTrue(version.equals(version)); - Assert.assertTrue(version.equals(sameVersion)); - Assert.assertFalse(version.equals(differentVersion)); - } - - public void testSerialize() throws Exception - { - String nettyVersion = "netty"; - VersionImpl version = new VersionImpl("uyiuy", 3, 7, 6, 12, "uhuhuh", nettyVersion, new int[]{9,10,11,12}); - ByteArrayOutputStream baos = new ByteArrayOutputStream(); - ObjectOutputStream oos = new ObjectOutputStream(baos); - oos.writeObject(version); - oos.flush(); - - ByteArrayInputStream bais = new ByteArrayInputStream(baos.toByteArray()); - ObjectInputStream ois = new ObjectInputStream(bais); - VersionImpl version2 = (VersionImpl)ois.readObject(); - - Assert.assertTrue(version.equals(version2)); - } - - // Package protected --------------------------------------------- - - // Protected ----------------------------------------------------- - - // Private ------------------------------------------------------- - - // Inner classes ------------------------------------------------- -} Deleted: trunk/tests/unit-tests/src/test/java/org/hornetq/tests/util/RandomUtil.java =================================================================== --- trunk/tests/unit-tests/src/test/java/org/hornetq/tests/util/RandomUtil.java 2011-06-17 15:50:19 UTC (rev 10847) +++ trunk/tests/unit-tests/src/test/java/org/hornetq/tests/util/RandomUtil.java 2011-06-17 15:52:20 UTC (rev 10848) @@ -1,176 +0,0 @@ -/* - * Copyright 2009 Red Hat, Inc. - * Red Hat 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.hornetq.tests.util; - -import java.util.Random; -import java.util.UUID; - -import javax.transaction.xa.Xid; - -import org.hornetq.api.core.HornetQBuffer; -import org.hornetq.api.core.HornetQBuffers; -import org.hornetq.api.core.SimpleString; -import org.hornetq.core.transaction.impl.XidImpl; - -/** - * @author <a href="mailto:jmesnil@redhat.com">Jeff Mesnil</a> - * - * @version <tt>$Revision$</tt> - * - */ -public final class RandomUtil -{ - // Constants ----------------------------------------------------- - - private static final Random random = new Random(System.currentTimeMillis()); - - // Attributes ---------------------------------------------------- - - // Static -------------------------------------------------------- - - public static String randomString() - { - return UUID.randomUUID().toString(); - } - - public static SimpleString randomSimpleString() - { - return new SimpleString(RandomUtil.randomString()); - } - - public static char randomChar() - { - return RandomUtil.randomString().charAt(0); - } - - public static long randomLong() - { - return RandomUtil.random.nextLong(); - } - - public static long randomPositiveLong() - { - return Math.abs(RandomUtil.randomLong()); - } - - public static int randomInt() - { - return RandomUtil.random.nextInt(); - } - - public static int randomPositiveInt() - { - return Math.abs(RandomUtil.randomInt()); - } - - - public static HornetQBuffer randomBuffer(final int size, final long... data) - { - HornetQBuffer buffer = HornetQBuffers.fixedBuffer(size + 8 * data.length); - - for (long d : data) - { - buffer.writeLong(d); - } - - for (int i = 0 ; i < size; i++) - { - buffer.writeByte(randomByte()); - } - - return buffer; - } - - - - - public static int randomInterval(final int min, final int max) - { - return min + randomMax(max - min); - } - - public static int randomMax(final int max) - { - int value = randomPositiveInt() % max; - - if (value == 0) - { - value = max; - } - - return value; - } - - public static int randomPort() - { - return RandomUtil.random.nextInt(65536); - } - - public static short randomShort() - { - return (short)RandomUtil.random.nextInt(Short.MAX_VALUE); - } - - public static byte randomByte() - { - return Integer.valueOf(RandomUtil.random.nextInt()).byteValue(); - } - - public static boolean randomBoolean() - { - return RandomUtil.random.nextBoolean(); - } - - public static byte[] randomBytes() - { - return RandomUtil.randomString().getBytes(); - } - - public static byte[] randomBytes(final int length) - { - byte[] bytes = new byte[length]; - for (int i = 0; i < bytes.length; i++) - { - bytes[i] = RandomUtil.randomByte(); - } - return bytes; - } - - public static double randomDouble() - { - return RandomUtil.random.nextDouble(); - } - - public static float randomFloat() - { - return RandomUtil.random.nextFloat(); - } - - public static Xid randomXid() - { - return new XidImpl(RandomUtil.randomBytes(), RandomUtil.randomInt(), RandomUtil.randomBytes()); - } - - // Constructors -------------------------------------------------- - - // Public -------------------------------------------------------- - - // Package protected --------------------------------------------- - - // Protected ----------------------------------------------------- - - // Private ------------------------------------------------------- - - // Inner classes ------------------------------------------------- -} Deleted: trunk/tests/unit-tests/src/test/java/org/hornetq/tests/util/UnitTestCase.java =================================================================== --- trunk/tests/unit-tests/src/test/java/org/hornetq/tests/util/UnitTestCase.java 2011-06-17 15:50:19 UTC (rev 10847) +++ trunk/tests/unit-tests/src/test/java/org/hornetq/tests/util/UnitTestCase.java 2011-06-17 15:52:20 UTC (rev 10848) @@ -1,1229 +0,0 @@ -/* - * Copyright 2009 Red Hat, Inc. - * Red Hat 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.hornetq.tests.util; - -import java.beans.BeanInfo; -import java.beans.Introspector; -import java.beans.PropertyDescriptor; -import java.io.BufferedInputStream; -import java.io.BufferedOutputStream; -import java.io.File; -import java.io.FileInputStream; -import java.io.FileOutputStream; -import java.io.IOException; -import java.io.InputStream; -import java.io.OutputStream; -import java.io.PrintWriter; -import java.io.StringWriter; -import java.lang.ref.Reference; -import java.lang.ref.WeakReference; -import java.net.ServerSocket; -import java.nio.ByteBuffer; -import java.util.ArrayList; -import java.util.HashMap; -import java.util.HashSet; -import java.util.Iterator; -import java.util.List; -import java.util.Map; -import java.util.Set; -import java.util.UUID; - -import javax.naming.Context; -import javax.transaction.xa.XAException; -import javax.transaction.xa.Xid; - -import junit.framework.Assert; -import junit.framework.TestCase; -import junit.framework.TestSuite; - -import org.hornetq.api.core.HornetQBuffer; -import org.hornetq.api.core.HornetQException; -import org.hornetq.api.core.SimpleString; -import org.hornetq.api.core.TransportConfiguration; -import org.hornetq.core.asyncio.impl.AsynchronousFileImpl; -import org.hornetq.core.client.impl.ServerLocatorImpl; -import org.hornetq.core.config.Configuration; -import org.hornetq.core.config.impl.ConfigurationImpl; -import org.hornetq.core.journal.impl.AIOSequentialFileFactory; -import org.hornetq.core.logging.Logger; -import org.hornetq.core.persistence.impl.journal.OperationContextImpl; -import org.hornetq.core.postoffice.Binding; -import org.hornetq.core.postoffice.Bindings; -import org.hornetq.core.postoffice.PostOffice; -import org.hornetq.core.postoffice.QueueBinding; -import org.hornetq.core.postoffice.impl.LocalQueueBinding; -import org.hornetq.core.remoting.impl.invm.InVMAcceptorFactory; -import org.hornetq.core.remoting.impl.invm.InVMConnectorFactory; -import org.hornetq.core.remoting.impl.invm.InVMRegistry; -import org.hornetq.core.remoting.impl.netty.NettyAcceptorFactory; -import org.hornetq.core.remoting.impl.netty.NettyConnectorFactory; -import org.hornetq.core.server.HornetQServer; -import org.hornetq.core.server.JournalType; -import org.hornetq.core.server.MessageReference; -import org.hornetq.core.server.Queue; -import org.hornetq.core.server.ServerMessage; -import org.hornetq.core.server.impl.ServerMessageImpl; -import org.hornetq.core.transaction.impl.XidImpl; -import org.hornetq.utils.UUIDGenerator; - -/** - * - * Helper base class for our unit tests - * - * @author <a href="mailto:tim.fox@jboss.com">Tim Fox</a> - * @author <a href="mailto:csuconic@redhat.com">Clebert</a> - * - */ -public class UnitTestCase extends TestCase -{ - // Constants ----------------------------------------------------- - - private static final Logger log = Logger.getLogger(UnitTestCase.class); - - public static final String INVM_ACCEPTOR_FACTORY = InVMAcceptorFactory.class.getCanonicalName(); - - public static final String INVM_CONNECTOR_FACTORY = InVMConnectorFactory.class.getCanonicalName(); - - public static final String NETTY_ACCEPTOR_FACTORY = NettyAcceptorFactory.class.getCanonicalName(); - - public static final String NETTY_CONNECTOR_FACTORY = NettyConnectorFactory.class.getCanonicalName(); - - // Attributes ---------------------------------------------------- - - private static final String testDir = System.getProperty("java.io.tmpdir", "/tmp") + "/hornetq-unit-test"; - - // There is a verification about thread leakages. We only fail a single thread when this happens - private static Set<Thread> alreadyFailedThread = new HashSet<Thread>(); - - // Static -------------------------------------------------------- - - protected Configuration createDefaultConfig() - { - return createDefaultConfig(false); - } - - protected Configuration createDefaultConfig(final boolean netty) - { - if (netty) - { - return createDefaultConfig(new HashMap<String, Object>(), INVM_ACCEPTOR_FACTORY, NETTY_ACCEPTOR_FACTORY); - } - else - { - return createDefaultConfig(new HashMap<String, Object>(), INVM_ACCEPTOR_FACTORY); - } - } - - protected static Configuration createClusteredDefaultConfig(final int index, - final Map<String, Object> params, - final String... acceptors) - { - Configuration config = createDefaultConfig(index, params, acceptors); - - config.setClustered(true); - - return config; - } - - protected static Configuration createDefaultConfig(final int index, - final Map<String, Object> params, - final String... acceptors) - { - Configuration configuration = createBasicConfig(index); - - configuration.getAcceptorConfigurations().clear(); - - for (String acceptor : acceptors) - { - TransportConfiguration transportConfig = new TransportConfiguration(acceptor, params); - configuration.getAcceptorConfigurations().add(transportConfig); - } - - return configuration; - } - - protected static ConfigurationImpl createBasicConfig() - { - return createBasicConfig(0); - } - - /** - * @param serverID - * @return - */ - protected static ConfigurationImpl createBasicConfig(final int serverID) - { - ConfigurationImpl configuration = new ConfigurationImpl(); - configuration.setSecurityEnabled(false); - configuration.setBindingsDirectory(getBindingsDir(serverID, false)); - configuration.setJournalMinFiles(2); - configuration.setJournalDirectory(getJournalDir(serverID, false)); - configuration.setJournalFileSize(100 * 1024); - configuration.setJournalType(getDefaultJournalType()); - configuration.setPagingDirectory(getPageDir(serverID, false)); - configuration.setLargeMessagesDirectory(getLargeMessagesDir(serverID, false)); - configuration.setJournalCompactMinFiles(0); - configuration.setJournalCompactPercentage(0); - return configuration; - } - - protected static Configuration createDefaultConfig(final Map<String, Object> params, final String... acceptors) - { - Configuration configuration = new ConfigurationImpl(); - configuration.setSecurityEnabled(false); - configuration.setJMXManagementEnabled(false); - configuration.setBindingsDirectory(getBindingsDir()); - configuration.setJournalMinFiles(2); - configuration.setJournalDirectory(getJournalDir()); - configuration.setJournalFileSize(100 * 1024); - configuration.setPagingDirectory(getPageDir()); - configuration.setLargeMessagesDirectory(getLargeMessagesDir()); - configuration.setJournalCompactMinFiles(0); - configuration.setJournalCompactPercentage(0); - - configuration.setFileDeploymentEnabled(false); - - configuration.setJournalType(getDefaultJournalType()); - - configuration.getAcceptorConfigurations().clear(); - - for (String acceptor : acceptors) - { - TransportConfiguration transportConfig = new TransportConfiguration(acceptor, params); - configuration.getAcceptorConfigurations().add(transportConfig); - } - - return configuration; - } - - protected static String getUDPDiscoveryAddress() - { - return System.getProperty("TEST-UDP-ADDRESS", "230.1.2.3"); - } - - protected static String getUDPDiscoveryAddress(final int variant) - { - String value = getUDPDiscoveryAddress(); - - int posPoint = value.lastIndexOf('.'); - - int last = Integer.valueOf(value.substring(posPoint + 1)); - - return value.substring(0, posPoint + 1) + (last + variant); - } - - public static int getUDPDiscoveryPort() - { - return Integer.parseInt(System.getProperty("TEST-UDP-PORT", "6750")); - } - - public static int getUDPDiscoveryPort(final int variant) - { - return getUDPDiscoveryPort() + variant; - } - - protected static JournalType getDefaultJournalType() - { - if (AsynchronousFileImpl.isLoaded()) - { - return JournalType.ASYNCIO; - } - else - { - return JournalType.NIO; - } - } - - /** - * @param name - */ - public UnitTestCase(final String name) - { - super(name); - } - - public UnitTestCase() - { - super(); - } - - public static void forceGC() - { - WeakReference<Object> dumbReference = new WeakReference<Object>(new Object()); - // A loop that will wait GC, using the minimal time as possible - while (dumbReference.get() != null) - { - System.gc(); - try - { - Thread.sleep(500); - } - catch (InterruptedException e) - { - } - } - } - - public static void forceGC(final Reference<?> ref, final long timeout) - { - long waitUntil = System.currentTimeMillis() + timeout; - // A loop that will wait GC, using the minimal time as possible - while (ref.get() != null && System.currentTimeMillis() < waitUntil) - { - ArrayList<String> list = new ArrayList<String>(); - for (int i = 0; i < 1000; i++) - { - list.add("Some string with garbage with concatenation " + i); - } - list.clear(); - list = null; - System.gc(); - try - { - Thread.sleep(500); - } - catch (InterruptedException e) - { - } - } - } - - // verify if these weak references are released after a few GCs - public static void checkWeakReferences(final WeakReference<?>... references) - { - - int i = 0; - boolean hasValue = false; - - do - { - hasValue = false; - - if (i > 0) - { - UnitTestCase.forceGC(); - } - - for (WeakReference<?> ref : references) - { - if (ref.get() != null) - { - hasValue = true; - } - } - } - while (i++ <= 30 && hasValue); - - for (WeakReference<?> ref : references) - { - Assert.assertNull(ref.get()); - } - } - - public static String threadDump(final String msg) - { - StringWriter str = new StringWriter(); - PrintWriter out = new PrintWriter(str); - - Map<Thread, StackTraceElement[]> stackTrace = Thread.getAllStackTraces(); - - out.println("*******************************************************************************"); - out.println("Complete Thread dump " + msg); - - for (Map.Entry<Thread, StackTraceElement[]> el : stackTrace.entrySet()) - { - out.println("==============================================================================="); - out.println("Thread " + el.getKey() + - " name = " + - el.getKey().getName() + - " id = " + - el.getKey().getId() + - " group = " + - el.getKey().getThreadGroup()); - out.println(); - for (StackTraceElement traceEl : el.getValue()) - { - out.println(traceEl); - } - } - - out.println("==============================================================================="); - out.println("End Thread dump " + msg); - out.println("*******************************************************************************"); - - return str.toString(); - } - - protected static TestSuite createAIOTestSuite(final Class<?> clazz) - { - TestSuite suite = new TestSuite(clazz.getName() + " testsuite"); - - if (AIOSequentialFileFactory.isSupported()) - { - suite.addTestSuite(clazz); - } - else - { - // System.out goes towards JUnit report - System.out.println("Test " + clazz.getName() + " ignored as AIO is not available"); - } - - return suite; - } - - public static String dumpBytes(final byte[] bytes) - { - StringBuffer buff = new StringBuffer(); - - buff.append(System.identityHashCode(bytes) + ", size: " + bytes.length + " ["); - - for (int i = 0; i < bytes.length; i++) - { - buff.append(bytes[i]); - - if (i != bytes.length - 1) - { - buff.append(", "); - } - } - - buff.append("]"); - - return buff.toString(); - } - - public static String dumbBytesHex(final byte[] buffer, final int bytesPerLine) - { - - StringBuffer buff = new StringBuffer(); - - buff.append("["); - - for (int i = 0; i < buffer.length; i++) - { - buff.append(String.format("%1$2X", buffer[i])); - if (i + 1 < buffer.length) - { - buff.append(", "); - } - if ((i + 1) % bytesPerLine == 0) - { - buff.append("\n "); - } - } - buff.append("]"); - - return buff.toString(); - } - - public static void assertEqualsByteArrays(final byte[] expected, final byte[] actual) - { - // assertEquals(expected.length, actual.length); - for (int i = 0; i < expected.length; i++) - { - Assert.assertEquals("byte at index " + i, expected[i], actual[i]); - } - } - - public static void assertEqualsTransportConfigurations(final TransportConfiguration[] expected, - final TransportConfiguration[] actual) - { - assertEquals(expected.length, actual.length); - for (int i = 0; i < expected.length; i++) - { - Assert.assertEquals("TransportConfiguration at index " + i, expected[i], actual[i]); - } - } - - public static void assertEqualsBuffers(final int size, final HornetQBuffer expected, final HornetQBuffer actual) - { - // assertEquals(expected.length, actual.length); - expected.readerIndex(0); - actual.readerIndex(0); - - for (int i = 0; i < size; i++) - { - byte b1 = expected.readByte(); - byte b2 = actual.readByte(); - Assert.assertEquals("byte at index " + i, b1, b2); - } - expected.resetReaderIndex(); - actual.resetReaderIndex(); - } - - public static void assertEqualsByteArrays(final int length, final byte[] expected, final byte[] actual) - { - // we check only for the given length (the arrays might be - // larger) - Assert.assertTrue(expected.length >= length); - Assert.assertTrue(actual.length >= length); - for (int i = 0; i < length; i++) - { - Assert.assertEquals("byte at index " + i, expected[i], actual[i]); - } - } - - public static void assertSameXids(final List<Xid> expected, final List<Xid> actual) - { - Assert.assertNotNull(expected); - Assert.assertNotNull(actual); - Assert.assertEquals(expected.size(), actual.size()); - - for (int i = 0; i < expected.size(); i++) - { - Xid expectedXid = expected.get(i); - Xid actualXid = actual.get(i); - UnitTestCase.assertEqualsByteArrays(expectedXid.getBranchQualifier(), actualXid.getBranchQualifier()); - Assert.assertEquals(expectedXid.getFormatId(), actualXid.getFormatId()); - UnitTestCase.assertEqualsByteArrays(expectedXid.getGlobalTransactionId(), actualXid.getGlobalTransactionId()); - } - } - - protected static void checkNoBinding(final Context context, final String binding) - { - try - { - context.lookup(binding); - Assert.fail("there must be no resource to look up for " + binding); - } - catch (Exception e) - { - } - } - - protected static Object checkBinding(final Context context, final String binding) throws Exception - { - Object o = context.lookup(binding); - Assert.assertNotNull(o); - return o; - } - - /** - * @param connectorConfigs - * @return - */ - protected ArrayList<String> registerConnectors(final HornetQServer server, - final List<TransportConfiguration> connectorConfigs) - { - // The connectors need to be pre-configured at main config object but this method is taking - // TransportConfigurations directly - // So this will first register them at the config and then generate a list of objects - ArrayList<String> connectors = new ArrayList<String>(); - for (TransportConfiguration tnsp : connectorConfigs) - { - String name = RandomUtil.randomString(); - - server.getConfiguration().getConnectorConfigurations().put(name, tnsp); - - connectors.add(name); - } - return connectors; - } - - protected static void checkFreePort(final int... ports) - { - for (int port : ports) - { - ServerSocket ssocket = null; - try - { - ssocket = new ServerSocket(port); - } - catch (Exception e) - { - throw new IllegalStateException("port " + port + " is already bound"); - } - finally - { - if (ssocket != null) - { - try - { - ssocket.close(); - } - catch (IOException e) - { - } - } - } - } - } - - // Constructors -------------------------------------------------- - - // Protected ----------------------------------------------------- - - /** - * @return the testDir - */ - protected String getTestDir() - { - return testDir; - } - - protected void clearData() - { - clearData(getTestDir()); - } - - protected void clearData(final String testDir) - { - // Need to delete the root - - File file = new File(testDir); - deleteDirectory(file); - file.mkdirs(); - - recreateDirectory(getJournalDir(testDir)); - recreateDirectory(getBindingsDir(testDir)); - recreateDirectory(getPageDir(testDir)); - recreateDirectory(getLargeMessagesDir(testDir)); - recreateDirectory(getClientLargeMessagesDir(testDir)); - recreateDirectory(getTemporaryDir(testDir)); - - System.out.println("deleted " + testDir); - } - - /** - * @return the journalDir - */ - public static String getJournalDir() - { - return getJournalDir(testDir); - } - - protected static String getJournalDir(final String testDir) - { - return testDir + "/journal"; - } - - protected static String getJournalDir(final int index, final boolean backup) - { - String dir = getJournalDir(testDir) + index + "-" + (backup ? "B" : "L"); - - return dir; - } - - /** - * @return the bindingsDir - */ - protected static String getBindingsDir() - { - return getBindingsDir(testDir); - } - - /** - * @return the bindingsDir - */ - protected static String getBindingsDir(final String testDir) - { - return testDir + "/bindings"; - } - - /** - * @return the bindingsDir - */ - protected static String getBindingsDir(final int index, final boolean backup) - { - return getBindingsDir(testDir) + index + "-" + (backup ? "B" : "L"); - } - - /** - * @return the pageDir - */ - protected static String getPageDir() - { - return getPageDir(testDir); - } - - /** - * @return the pageDir - */ - protected static String getPageDir(final String testDir) - { - return testDir + "/page"; - } - - protected static String getPageDir(final int index, final boolean backup) - { - return getPageDir(testDir) + index + "-" + (backup ? "B" : "L"); - } - - /** - * @return the largeMessagesDir - */ - protected static String getLargeMessagesDir() - { - return getLargeMessagesDir(testDir); - } - - /** - * @return the largeMessagesDir - */ - protected static String getLargeMessagesDir(final String testDir) - { - return testDir + "/large-msg"; - } - - protected static String getLargeMessagesDir(final int index, final boolean backup) - { - return getLargeMessagesDir(testDir) + index + "-" + (backup ? "B" : "L"); - } - - /** - * @return the clientLargeMessagesDir - */ - protected String getClientLargeMessagesDir() - { - return getClientLargeMessagesDir(testDir); - } - - /** - * @return the clientLargeMessagesDir - */ - protected String getClientLargeMessagesDir(final String testDir) - { - return testDir + "/client-large-msg"; - } - - /** - * @return the temporaryDir - */ - protected String getTemporaryDir() - { - return getTemporaryDir(testDir); - } - - /** - * @return the temporaryDir - */ - protected String getTemporaryDir(final String testDir) - { - return testDir + "/temp"; - } - - protected static void expectHornetQException(final String message, final int errorCode, final HornetQAction action) - { - try - { - action.run(); - Assert.fail(message); - } - catch (Exception e) - { - Assert.assertTrue(e instanceof HornetQException); - Assert.assertEquals(errorCode, ((HornetQException)e).getCode()); - } - } - - protected static void expectHornetQException(final int errorCode, final HornetQAction action) - { - UnitTestCase.expectHornetQException("must throw a HornetQException with the expected errorCode: " + errorCode, - errorCode, - action); - } - - protected static void expectXAException(final int errorCode, final HornetQAction action) - { - try - { - action.run(); - Assert.fail("must throw a XAException with the expected errorCode: " + errorCode); - } - catch (Exception e) - { - Assert.assertTrue(e instanceof XAException); - Assert.assertEquals(errorCode, ((XAException)e).errorCode); - } - } - - public static byte getSamplebyte(final long position) - { - return (byte)('a' + position % ('z' - 'a' + 1)); - } - - // Creates a Fake LargeStream without using a real file - public static InputStream createFakeLargeStream(final long size) throws Exception - { - return new InputStream() - { - private long count; - - private boolean closed = false; - - @Override - public void close() throws IOException - { - super.close(); - closed = true; - } - - @Override - public int read() throws IOException - { - if (closed) - { - throw new IOException("Stream was closed"); - } - if (count++ < size) - { - return UnitTestCase.getSamplebyte(count - 1); - } - else - { - return -1; - } - } - }; - - } - - /** It validates a Bean (POJO) using simple setters and getters with random values. - * You can pass a list of properties to be ignored, as some properties will have a pre-defined domain (not being possible to use random-values on them) */ - protected void validateGettersAndSetters(final Object pojo, final String... ignoredProperties) throws Exception - { - HashSet<String> ignoreSet = new HashSet<String>(); - - for (String ignore : ignoredProperties) - { - ignoreSet.add(ignore); - } - - BeanInfo info = Introspector.getBeanInfo(pojo.getClass()); - - PropertyDescriptor properties[] = info.getPropertyDescriptors(); - - for (PropertyDescriptor prop : properties) - { - Object value; - - if (prop.getPropertyType() == String.class) - { - value = RandomUtil.randomString(); - } - else if (prop.getPropertyType() == Integer.class || prop.getPropertyType() == Integer.TYPE) - { - value = RandomUtil.randomInt(); - } - else if (prop.getPropertyType() == Long.class || prop.getPropertyType() == Long.TYPE) - { - value = RandomUtil.randomLong(); - } - else if (prop.getPropertyType() == Boolean.class || prop.getPropertyType() == Boolean.TYPE) - { - value = RandomUtil.randomBoolean(); - } - else if (prop.getPropertyType() == Double.class || prop.getPropertyType() == Double.TYPE) - { - value = RandomUtil.randomDouble(); - } - else - { - System.out.println("Can't validate property of type " + prop.getPropertyType() + " on " + prop.getName()); - value = null; - } - - if (value != null && prop.getWriteMethod() != null && prop.getReadMethod() == null) - { - System.out.println("WriteOnly property " + prop.getName() + " on " + pojo.getClass()); - } - else if (value != null & prop.getWriteMethod() != null && - prop.getReadMethod() != null && - !ignoreSet.contains(prop.getName())) - { - System.out.println("Validating " + prop.getName() + " type = " + prop.getPropertyType()); - prop.getWriteMethod().invoke(pojo, value); - - Assert.assertEquals("Property " + prop.getName(), value, prop.getReadMethod().invoke(pojo)); - } - } - } - - // Package protected --------------------------------------------- - - // Protected ----------------------------------------------------- - - Map<Thread, StackTraceElement[]> previousThreads; - - @Override - protected void setUp() throws Exception - { - super.setUp(); - - OperationContextImpl.clearContext(); - - deleteDirectory(new File(getTestDir())); - - InVMRegistry.instance.clear(); - - // checkFreePort(TransportConstants.DEFAULT_PORT); - - previousThreads = Thread.getAllStackTraces(); - - UnitTestCase.log.info("###### starting test " + this.getClass().getName() + "." + getName()); - } - - @Override - protected void tearDown() throws Exception - { - cleanupPools(); - - Map<Thread, StackTraceElement[]> threadMap = Thread.getAllStackTraces(); - for (Thread thread : threadMap.keySet()) - { - StackTraceElement[] stack = threadMap.get(thread); - for (StackTraceElement stackTraceElement : stack) - { - if (stackTraceElement.getMethodName().contains("getConnectionWithRetry") && !alreadyFailedThread.contains(thread)) - { - alreadyFailedThread.add(thread); - System.out.println(threadDump(getName() + " has left threads running. Look at thread " + - thread.getName() + - " id = " + - thread.getId() + - " has running locators on test " + - getName() + - " on this following dump")); - fail("test left serverlocator running, this could effect other tests"); - // System.exit(0); - } - else if (stackTraceElement.getMethodName().contains("BroadcastGroupImpl.run") && !alreadyFailedThread.contains(thread)) - { - alreadyFailedThread.add(thread); - System.out.println(threadDump(getName() + " has left threads running. Look at thread " + - thread.getName() + - " id = " + - thread.getId() + - " is still broadcasting " + - getName() + - " on this following dump")); - fail("test left broadcastgroupimpl running, this could effect other tests"); - // System.exit(0); - } - } - } - - Map<Thread, StackTraceElement[]> postThreads = Thread.getAllStackTraces(); - - if (postThreads.size() > previousThreads.size()) - { - StringBuffer buffer = new StringBuffer(); - - - buffer.append("*********************************************************************************\n"); - buffer.append("LEAKING THREADS\n"); - - for (Thread aliveThread : postThreads.keySet()) - { - if (!aliveThread.getName().contains("SunPKCS11") && !previousThreads.containsKey(aliveThread)) - { - buffer.append("=============================================================================\n"); - buffer.append("Thread " + aliveThread + " is still alive with the following stackTrace:\n"); - StackTraceElement[] elements = postThreads.get(aliveThread); - for (StackTraceElement el : elements) - { - buffer.append(el + "\n"); - } - } - - } - buffer.append("*********************************************************************************\n"); - - System.out.println(buffer.toString()); - - } - - super.tearDown(); - } - - /** - * - */ - protected void cleanupPools() - { - OperationContextImpl.clearContext(); - - deleteDirectory(new File(getTestDir())); - - int invmSize = InVMRegistry.instance.size(); - if (invmSize > 0) - { - InVMRegistry.instance.clear(); - fail("invm registry still had acceptors registered"); - } - - if (AsynchronousFileImpl.getTotalMaxIO() != 0) - { - AsynchronousFileImpl.resetMaxAIO(); - Assert.fail("test did not close all its files " + AsynchronousFileImpl.getTotalMaxIO()); - } - - // We shutdown the global pools to give a better isolation between tests - ServerLocatorImpl.clearThreadPools(); - } - - protected byte[] autoEncode(final Object... args) - { - - int size = 0; - - for (Object arg : args) - { - if (arg instanceof Byte) - { - size++; - } - else if (arg instanceof Boolean) - { - size++; - } - else if (arg instanceof Integer) - { - size += 4; - } - else if (arg instanceof Long) - { - size += 8; - } - else if (arg instanceof Float) - { - size += 4; - } - else if (arg instanceof Double) - { - size += 8; - } - else - { - throw new IllegalArgumentException("method autoEncode doesn't know how to convert " + arg.getClass() + - " yet"); - } - } - - ByteBuffer buffer = ByteBuffer.allocate(size); - - for (Object arg : args) - { - if (arg instanceof Byte) - { - buffer.put(((Byte)arg).byteValue()); - } - else if (arg instanceof Boolean) - { - Boolean b = (Boolean)arg; - buffer.put((byte)(b.booleanValue() ? 1 : 0)); - } - else if (arg instanceof Integer) - { - buffer.putInt(((Integer)arg).intValue()); - } - else if (arg instanceof Long) - { - buffer.putLong(((Long)arg).longValue()); - } - else if (arg instanceof Float) - { - buffer.putFloat(((Float)arg).floatValue()); - } - else if (arg instanceof Double) - { - buffer.putDouble(((Double)arg).doubleValue()); - } - else - { - throw new IllegalArgumentException("method autoEncode doesn't know how to convert " + arg.getClass() + - " yet"); - } - } - - return buffer.array(); - } - - protected void recreateDirectory(final String directory) - { - File file = new File(directory); - deleteDirectory(file); - file.mkdirs(); - } - - protected boolean deleteDirectory(final File directory) - { - if (directory.isDirectory()) - { - String[] files = directory.list(); - - for (int j = 0; j < files.length; j++) - { - if (!deleteDirectory(new File(directory, files[j]))) - { - return false; - } - } - } - - return directory.delete(); - } - - protected void copyRecursive(final File from, final File to) throws Exception - { - if (from.isDirectory()) - { - if (!to.exists()) - { - to.mkdir(); - } - - String[] subs = from.list(); - - for (String sub : subs) - { - copyRecursive(new File(from, sub), new File(to, sub)); - } - } - else - { - InputStream in = null; - - OutputStream out = null; - - try - { - in = new BufferedInputStream(new FileInputStream(from)); - - out = new BufferedOutputStream(new FileOutputStream(to)); - - int b; - - while ((b = in.read()) != -1) - { - out.write(b); - } - } - finally - { - if (in != null) - { - in.close(); - } - - if (out != null) - { - out.close(); - } - } - } - } - - protected void assertRefListsIdenticalRefs(final List<MessageReference> l1, final List<MessageReference> l2) - { - if (l1.size() != l2.size()) - { - Assert.fail("Lists different sizes: " + l1.size() + ", " + l2.size()); - } - - Iterator<MessageReference> iter1 = l1.iterator(); - Iterator<MessageReference> iter2 = l2.iterator(); - - while (iter1.hasNext()) - { - MessageReference o1 = iter1.next(); - MessageReference o2 = iter2.next(); - - Assert.assertTrue("expected " + o1 + " but was " + o2, o1 == o2); - } - } - - protected ServerMessage generateMessage(final long id) - { - ServerMessage message = new ServerMessageImpl(id, 1000); - - message.setMessageID(id); - - message.getBodyBuffer().writeString(UUID.randomUUID().toString()); - - message.setAddress(new SimpleString("foo")); - - return message; - } - - protected MessageReference generateReference(final Queue queue, final long id) - { - ServerMessage message = generateMessage(id); - - return message.createReference(queue); - } - - protected int calculateRecordSize(final int size, final int alignment) - { - return (size / alignment + (size % alignment != 0 ? 1 : 0)) * alignment; - } - - protected XidImpl newXID() - { - return new XidImpl("xa1".getBytes(), 1, UUIDGenerator.getInstance().generateStringUUID().getBytes()); - } - - protected int getMessageCount(final HornetQServer service, final String address) throws Exception - { - return getMessageCount(service.getPostOffice(), address); - } - - /** - * @param address - * @param postOffice - * @return - * @throws Exception - */ - protected int getMessageCount(final PostOffice postOffice, final String address) throws Exception - { - int messageCount = 0; - - List<QueueBinding> bindings = getLocalQueueBindings(postOffice, address); - - for (QueueBinding qBinding : bindings) - { - messageCount += qBinding.getQueue().getMessageCount(); - } - - return messageCount; - } - - protected - List<QueueBinding> - getLocalQueueBindings(final PostOffice postOffice, final String address) throws Exception - { - ArrayList<QueueBinding> bindingsFound = new ArrayList<QueueBinding>(); - - Bindings bindings = postOffice.getBindingsForAddress(new SimpleString(address)); - - for (Binding binding : bindings.getBindings()) - { - if (binding instanceof LocalQueueBinding) - { - bindingsFound.add((QueueBinding)binding); - } - } - return bindingsFound; - } - - // Private ------------------------------------------------------- - - // Inner classes ------------------------------------------------- - - protected static interface HornetQAction - { - void run() throws Exception; - } - -}