/* This file is part of VoltDB.
* Copyright (C) 2008-2017 VoltDB Inc.
*
* Permission is hereby granted, free of charge, to any person obtaining
* a copy of this software and associated documentation files (the
* "Software"), to deal in the Software without restriction, including
* without limitation the rights to use, copy, modify, merge, publish,
* distribute, sublicense, and/or sell copies of the Software, and to
* permit persons to whom the Software is furnished to do so, subject to
* the following conditions:
*
* The above copyright notice and this permission notice shall be
* included in all copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
* IN NO EVENT SHALL THE AUTHORS BE LIABLE FOR ANY CLAIM, DAMAGES OR
* OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
* ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
* OTHER DEALINGS IN THE SOFTWARE.
*/
package org.voltdb.utils;
import java.io.CharArrayReader;
import java.io.File;
import java.io.IOException;
import java.io.LineNumberReader;
import java.io.UnsupportedEncodingException;
import java.net.URL;
import java.net.URLDecoder;
import java.util.Arrays;
import junit.framework.TestCase;
import org.voltdb.benchmark.tpcc.TPCCProjectBuilder;
import org.voltdb.catalog.Catalog;
import org.voltdb.catalog.Database;
import org.voltdb.compiler.VoltCompiler;
import org.voltdb.compiler.VoltProjectBuilder;
import org.voltdb.utils.InMemoryJarfile.JarLoader;
public class TestInMemoryJarfile extends TestCase {
protected File m_jarPath;
protected Catalog m_catalog;
protected Database m_catalogDb;
private Catalog createTestJarFile(String jarFileName, boolean adhoc)
{
String schemaPath = "";
try {
URL url = TPCCProjectBuilder.class.getResource("tpcc-ddl.sql");
schemaPath = URLDecoder.decode(url.getPath(), "UTF-8");
} catch (UnsupportedEncodingException e) {
e.printStackTrace();
System.exit(-1);
}
String schema2 = "partition table WAREHOUSE on column W_ID;\n" +
"create procedure from class org.voltdb.compiler.procedures.TPCCTestProc;\n";
if (adhoc) {
schema2 += "create role default with sql, adhoc;";
}
final File schema2File = VoltProjectBuilder.writeStringToTempFile(schema2);
final String schema2Path = schema2File.getPath();
VoltCompiler compiler = new VoltCompiler(false);
assertTrue(compiler.compileFromDDL(jarFileName, schemaPath, schema2Path));
return compiler.getCatalog();
}
@Override
protected void setUp() throws Exception {
System.out.print("START: " + System.currentTimeMillis());
super.setUp();
m_catalog = createTestJarFile("testout.jar", true);
assertNotNull(m_catalog);
m_catalogDb = m_catalog.getClusters().get("cluster").getDatabases().get("database");
assertNotNull(m_catalogDb);
m_jarPath = new File("testout.jar");
}
@Override
protected void tearDown() throws Exception {
super.tearDown();
if (m_jarPath != null) {
assertTrue(m_jarPath.delete());
}
File dupeFile = new File("testout-dupe.jar");
dupeFile.delete();
}
/**
*
*/
public void testReadFileFromJarfile() throws IOException {
String catalog0 = this.m_catalog.serialize();
assertTrue(catalog0.length() > 0);
InMemoryJarfile jarfile = new InMemoryJarfile(m_jarPath.getAbsolutePath());
byte[] catalogBytes = jarfile.get(CatalogUtil.CATALOG_FILENAME);
String catalog1 = new String(catalogBytes, "UTF-8");
assertTrue(catalog1.length() > 0);
assertEquals(catalog0.length(), catalog1.length());
LineNumberReader reader0 = new LineNumberReader(new CharArrayReader(catalog0.toCharArray()));
LineNumberReader reader1 = new LineNumberReader(new CharArrayReader(catalog1.toCharArray()));
try {
int lines = 0;
while (reader0.ready()) {
assertEquals(reader0.ready(), reader1.ready());
assertEquals(reader0.readLine(), reader1.readLine());
lines++;
}
assertTrue(lines > 0);
reader0.close();
reader1.close();
} catch (Exception ex) {
ex.printStackTrace();
assertTrue(false);
}
}
public void testIdenticalJarContentsMatchCRCs()
throws IOException, InterruptedException
{
// Create a second jarfile with identical contents
// Sleep for 5 seconds so the timestamps will differ
// and cause different global CRCs
// Use "group*" element names for backward compatibility test.
Thread.sleep(5000);
createTestJarFile("testout-dupe.jar", true);
long crc1 = new InMemoryJarfile(m_jarPath).getCRC();
long crc2 = new InMemoryJarfile("testout-dupe.jar").getCRC();
assertEquals(crc1, crc2);
byte[] sha1 = new InMemoryJarfile(m_jarPath).getSha1Hash();
byte[] sha2 = new InMemoryJarfile("testout-dupe.jar").getSha1Hash();
assertTrue(Arrays.equals(sha1, sha2));
}
public void testDifferentJarContentsDontMatchCRCs()
throws IOException, InterruptedException
{
// Create a second jarfile with different contents
createTestJarFile("testout-dupe.jar", false);
long crc1 = new InMemoryJarfile("testout.jar").getCRC();
long crc2 = new InMemoryJarfile("testout-dupe.jar").getCRC();
assertFalse(crc1 == crc2);
byte[] sha1 = new InMemoryJarfile(m_jarPath).getSha1Hash();
byte[] sha2 = new InMemoryJarfile("testout-dupe.jar").getSha1Hash();
assertFalse(Arrays.equals(sha1, sha2));
}
public void testJarfileRemoveClassRemovesInnerClasses() throws Exception
{
InMemoryJarfile dut = new InMemoryJarfile();
// Add a class file that we know has inner classes
// Someday this seems like it should be an operation directly on InMemoryJarfile
VoltCompiler comp = new VoltCompiler(false);
// This will pull in all the inner classes (currently 4 of them), but check anyway
comp.addClassToJar(dut, org.voltdb_testprocs.updateclasses.InnerClassesTestProc.class);
JarLoader loader = dut.getLoader();
assertEquals(5, loader.getClassNames().size());
System.out.println(loader.getClassNames());
assertTrue(loader.getClassNames().contains("org.voltdb_testprocs.updateclasses.InnerClassesTestProc$InnerNotPublic"));
assertTrue(dut.get("org/voltdb_testprocs/updateclasses/InnerClassesTestProc$InnerNotPublic.class") != null);
// Now, remove the outer class and verify that all the inner classes go away.
dut.removeClassFromJar("org.voltdb_testprocs.updateclasses.InnerClassesTestProc");
assertTrue(loader.getClassNames().isEmpty());
assertTrue(dut.get("org/voltdb_testprocs/updateclasses/InnerClassesTestProc$InnerNotPublic.class") == null);
}
}