/*
* ModeShape (http://www.modeshape.org)
*
* Licensed 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.modeshape.jcr;
import static org.hamcrest.core.Is.is;
import static org.junit.Assert.assertArrayEquals;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertThat;
import static org.junit.Assert.assertTrue;
import static org.junit.Assert.fail;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.util.Arrays;
import java.util.HashSet;
import java.util.Set;
import java.util.UUID;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicInteger;
import java.util.zip.GZIPInputStream;
import java.util.zip.ZipEntry;
import java.util.zip.ZipInputStream;
import javax.jcr.Binary;
import javax.jcr.ImportUUIDBehavior;
import javax.jcr.NoSuchWorkspaceException;
import javax.jcr.Node;
import javax.jcr.PathNotFoundException;
import javax.jcr.PropertyIterator;
import javax.jcr.RepositoryException;
import javax.jcr.Session;
import javax.jcr.query.Query;
import javax.jcr.query.QueryResult;
import javax.transaction.NotSupportedException;
import javax.transaction.SystemException;
import javax.transaction.TransactionManager;
import org.junit.Before;
import org.junit.Test;
import org.modeshape.common.FixFor;
import org.modeshape.common.util.FileUtil;
import org.modeshape.common.util.IoUtil;
import org.modeshape.jcr.api.BackupOptions;
import org.modeshape.jcr.api.JcrTools;
import org.modeshape.jcr.api.Problems;
import org.modeshape.jcr.api.RestoreOptions;
/**
* Tests repository backup and restore
*/
public class RepositoryBackupAndRestoreTest extends SingleUseAbstractTest {
private static final String[] BINARY_RESOURCES = new String[] { "data/large-file1.png",
"data/move-initial-data.xml",
"data/simple.json",
"data/singleNode.json" };
private File backupArea;
private File backupDirectory;
private File backupDirectory2;
private File backupRepoDir;
@Override
protected RepositoryConfiguration createRepositoryConfiguration(String repositoryName) throws Exception {
return RepositoryConfiguration.read("config/backup-repo-config.json");
}
@Before
@Override
public void beforeEach() throws Exception {
backupArea = new File("target/backupArea");
FileUtil.delete(backupArea.getPath());
//use a UUID for the backup folder to prevent some file locks lingering between tests
String folderId = UUID.randomUUID().toString();
backupDirectory = new File(backupArea, "repoBackups_" + folderId);
backupDirectory2 = new File(backupArea, "repoBackupsAfter_" + folderId);
backupDirectory.mkdirs();
backupDirectory2.mkdirs();
backupRepoDir = new File(backupArea, "backupRepo");
backupRepoDir.mkdirs();
new File(backupArea, "restoreRepo").mkdirs();
super.beforeEach();
}
@Test
@FixFor( "MODE-2440" )
public void shouldBackupRepositoryWhichIncludesBinaryValuesCompressed() throws Exception {
loadBinaryContent();
makeBackup(BackupOptions.DEFAULT);
wipeRepository();
restoreBackup();
verifyBinaryContent();
}
@Test
@FixFor( "MODE-2559" )
public void shouldBackupRepositoryWhichIncludesBinaryValuesUncompressed() throws Exception {
BackupOptions backupOptions = new BackupOptions() {
@Override
public boolean compress() {
return false;
}
};
loadBinaryContent();
makeBackup(backupOptions);
wipeRepository();
restoreBackup();
verifyBinaryContent();
}
@Test
@FixFor( "MODE-2584" )
public void shouldPreserveBinariesFromRestoredBackup() throws Exception {
loadBinaryContent();
makeBackup(BackupOptions.DEFAULT);
wipeRepository();
restoreBackup();
makeBackup(BackupOptions.DEFAULT);
wipeRepository();
restoreBackup();
verifyBinaryContent();
}
private void makeBackup(BackupOptions options) throws RepositoryException {
TestingUtil.waitUntilFolderCleanedUp(backupDirectory.getPath());
JcrSession session = repository().login();
try {
Problems problems = session.getWorkspace().getRepositoryManager().backupRepository(backupDirectory, options);
assertNoProblems(problems);
} finally {
session.logout();
}
}
private void wipeRepository() {
// shutdown the repo and remove all repo data (stored on disk)
repository().doShutdown(false);
assertTrue(FileUtil.delete(backupRepoDir));
}
private void restoreBackup() throws Exception {
startRepositoryWithConfiguration(resourceStream("config/backup-repo-config.json"));
JcrSession session = repository().login();
Problems problems = session.getWorkspace().getRepositoryManager().restoreRepository(backupDirectory);
assertNoProblems(problems);
session.logout();
}
private void verifyBinaryContent() throws Exception {
assertFilesInWorkspace("default");
assertFilesInWorkspace("ws2");
assertFilesInWorkspace("ws3");
}
/**
* Test that a repository containing the same binary data (files) as the ones from this test backed up using ModeShape 3.8.1
* is restored correctly.
*/
@Test
@FixFor( "MODE-2440" )
public void shouldRestoreLegacy381RepositoryWithSameName() throws Exception {
// extract the old repo backup
File legacyBackupDir = extractZip("legacy_backup/repoBackups381.zip", this.backupArea);
assertTrue("Zip content not extracted correctly", legacyBackupDir.exists() && legacyBackupDir.canRead() && legacyBackupDir.isDirectory());
Problems problems = session().getWorkspace().getRepositoryManager().restoreRepository(legacyBackupDir);
assertNoProblems(problems);
verifyBinaryContent();
}
/**
* Test that a repository backed up using ModeShape 3.8.1 is restored correctly using another repository name and cache name.
*/
@Test
@FixFor("MODE-2539")
public void shouldRestoreLegacy381RepositoryWithDifferentName() throws Exception {
// extract the old repo backup
File legacyBackupDir = extractZip("legacy_backup/repoBackups381_2.zip", this.backupArea);
assertTrue("Zip content not extracted correctly", legacyBackupDir.exists() && legacyBackupDir.canRead() && legacyBackupDir.isDirectory());
Problems problems = session().getWorkspace().getRepositoryManager().restoreRepository(legacyBackupDir);
assertNoProblems(problems);
final Session session = repository.login();
final AtomicInteger totalCount = new AtomicInteger(0);
tools.onEachNode(session, false, new JcrTools.NodeOperation() {
@Override
public void run(Node node) throws Exception {
if (node.getPath().equals("/")) {
return;
}
totalCount.incrementAndGet();
// the backup created 7 properties per node + the primary type
PropertyIterator properties = node.getProperties();
assertEquals(8, properties.getSize());
while (properties.hasNext()) {
assertNotNull(properties.nextProperty().getString());
}
}
});
// the backup should have 1110 nodes without the root...
assertEquals(1110, totalCount.get());
}
@Test
@FixFor( "MODE-2440" )
public void shouldRestoreBinaryReferencesWhenExcludedFromBackup() throws Exception {
loadBinaryContent();
verifyBinaryContent();
// Make the backup, and check that there are no problems ...
BackupOptions backupOptions = new BackupOptions() {
@Override
public boolean includeBinaries() {
return false;
}
};
Problems problems = session().getWorkspace().getRepositoryManager().backupRepository(backupDirectory, backupOptions);
assertNoProblems(problems);
// shutdown the repo and remove just the repo main store (not the binary store)
assertTrue(repository().shutdown().get());
FileUtil.delete(backupRepoDir.getPath() + "/store");
// start a fresh empty repo and then restore just the data without binaries
startRepositoryWithConfigurationFrom("config/backup-repo-config.json");
RestoreOptions restoreOptions = new RestoreOptions() {
@Override
public boolean includeBinaries() {
return false;
}
};
problems = session().getWorkspace().getRepositoryManager().restoreRepository(backupDirectory, restoreOptions);
assertNoProblems(problems);
verifyBinaryContent();
}
@Test
public void shouldBackupRepositoryWithMultipleWorkspaces() throws Exception {
loadContent();
Problems problems = session().getWorkspace().getRepositoryManager().backupRepository(backupDirectory);
assertNoProblems(problems);
// Make some changes that will not be in the backup ...
session().getRootNode().addNode("node-not-in-backup");
session().save();
assertContentInWorkspace(repository(), "default", "/node-not-in-backup");
assertContentInWorkspace(repository(), "ws2");
assertContentInWorkspace(repository(), "ws3");
// Start up a new repository
RepositoryConfiguration config = RepositoryConfiguration.read("config/restore-repo-config.json");
JcrRepository newRepository = new JcrRepository(config);
try {
newRepository.start();
// And restore it from the contents ...
JcrSession newSession = newRepository.login();
try {
Problems restoreProblems = newSession.getWorkspace().getRepositoryManager().restoreRepository(backupDirectory);
assertNoProblems(restoreProblems);
} finally {
newSession.logout();
}
// Check that the node that was added *after* the backup is not there ...
assertContentNotInWorkspace(newRepository, "default", "/node-not-in-backup");
// Before we assert the content, create a backup of it (for comparison purposes when debugging) ...
newSession = newRepository.login();
try {
Problems backupProblems = newSession.getWorkspace().getRepositoryManager().backupRepository(backupDirectory2);
assertNoProblems(backupProblems);
} finally {
newSession.logout();
}
assertWorkspaces(newRepository, "default", "ws2", "ws3");
assertContentInWorkspace(newRepository, null);
assertContentInWorkspace(newRepository, "ws2");
assertContentInWorkspace(newRepository, "ws3");
queryContentInWorkspace(newRepository, null);
} finally {
newRepository.shutdown().get(10, TimeUnit.SECONDS);
}
}
@Test
public void shouldBackupAndRestoreRepositoryWithMultipleWorkspaces() throws Exception {
// Load the content and verify it's there ...
loadContent();
assertContentInWorkspace(repository(), "default");
assertContentInWorkspace(repository(), "ws2");
assertContentInWorkspace(repository(), "ws3");
// Make the backup, and check that there are no problems ...
Problems problems = session().getWorkspace().getRepositoryManager().backupRepository(backupDirectory);
assertNoProblems(problems);
// Make some changes that will not be in the backup ...
session().getRootNode().addNode("node-not-in-backup");
session().save();
// Check the content again ...
assertContentInWorkspace(repository(), "default", "/node-not-in-backup");
assertContentInWorkspace(repository(), "ws2");
assertContentInWorkspace(repository(), "ws3");
// Restore the content from the backup into our current repository ...
JcrSession newSession = repository().login();
try {
Problems restoreProblems = newSession.getWorkspace().getRepositoryManager().restoreRepository(backupDirectory);
assertNoProblems(restoreProblems);
} finally {
newSession.logout();
}
assertWorkspaces(repository(), "default", "ws2", "ws3");
// Check the content again ...
assertContentInWorkspace(repository(), "default");
assertContentInWorkspace(repository(), "ws2");
assertContentInWorkspace(repository(), "ws3");
assertContentNotInWorkspace(repository(), "default", "/node-not-in-backup");
queryContentInWorkspace(repository(), null);
}
@FixFor( "MODE-2309" )
@Test
public void shouldBackupAndRestoreRepositoryWithLineBreaksInPropertyValues() throws Exception {
// Load the content and verify it's there ...
importIntoWorkspace("default", "io/cars-system-view.xml");
assertWorkspaces(repository(), "default");
assertContentInWorkspace(repository(), "default");
print = true;
Node prius = session().getNode("/Cars/Hybrid/Toyota Prius");
prius.setProperty("crlfproperty", "test\r\ntest\r\ntest");
prius.setProperty("lfprop", "value\nvalue\nvalue");
session().save();
// Make the backup, and check that there are no problems ...
Problems problems = session().getWorkspace().getRepositoryManager().backupRepository(backupDirectory);
assertNoProblems(problems);
// Make some changes that will not be in the backup ...
session().getRootNode().addNode("node-not-in-backup");
session().save();
// Check the content again ...
assertContentInWorkspace(repository(), "default", "/node-not-in-backup");
// Restore the content from the backup into our current repository ...
JcrSession newSession = repository().login();
try {
Problems restoreProblems = newSession.getWorkspace().getRepositoryManager().restoreRepository(backupDirectory);
assertNoProblems(restoreProblems);
} finally {
newSession.logout();
}
assertWorkspaces(repository(), "default");
// Check the content again ...
assertContentInWorkspace(repository(), "default");
assertContentNotInWorkspace(repository(), "default", "/node-not-in-backup");
queryContentInWorkspace(repository(), null);
}
@Test
@FixFor( "MODE-2253" )
public void shouldBackupAndRestoreWithExistingUserTransaction() throws Exception {
loadContent();
startTransaction();
Problems problems = session().getWorkspace().getRepositoryManager().backupRepository(backupDirectory);
assertNoProblems(problems);
problems = session().getWorkspace().getRepositoryManager().restoreRepository(backupDirectory);
assertNoProblems(problems);
rollbackTransaction();
assertContentInWorkspace(repository(), "default");
assertContentInWorkspace(repository(), "ws2");
assertContentInWorkspace(repository(), "ws3");
}
@Test
@FixFor( "MODE-2440" )
public void shouldRestoreLegacy450BackupWithCompressedBinaries() throws Exception {
// extract the old repo backup
File legacyBackupDir = extractZip("legacy_backup/repoBackups450.zip", this.backupArea);
assertTrue("Zip content not extracted correctly", legacyBackupDir.exists() && legacyBackupDir.canRead() && legacyBackupDir.isDirectory());
Problems problems = session().getWorkspace().getRepositoryManager().restoreRepository(legacyBackupDir);
assertNoProblems(problems);
verifyBinaryContent();
}
@Test
@FixFor( "MODE-2611" )
public void backupShouldHaveUniqueIDs() throws Exception {
loadContent();
makeBackup(BackupOptions.DEFAULT);
File backup = backupDirectory.listFiles((dir, name) -> name.contains("documents"))[0];
String idPrefix = "{ \"metadata\" : { \"id\" : \"";
try (GZIPInputStream gzipInputStream = new GZIPInputStream(new FileInputStream(backup))) {
String fileContent = IoUtil.read(gzipInputStream);
String lines[] = fileContent.split("\\r?\\n");
Set<String> ids = Arrays.stream(lines)
.map(s -> s.substring(s.indexOf(idPrefix) + idPrefix.length(), s.indexOf("\" }")))
.collect(HashSet::new, (set, id) -> {
if (!set.add(id)) {
fail("Duplicate id found: " + id);
}
}, HashSet::addAll);
assertEquals(lines.length, ids.size());
}
}
private File extractZip( String zipFile, File destination ) throws IOException {
File backupDir = null;
final int bufferSize = 2048;
File currentFile = destination;
try (ZipInputStream zipInputStream = new ZipInputStream(resourceStream(zipFile))) {
ZipEntry entry = zipInputStream.getNextEntry();
while (entry != null) {
currentFile = new File(destination, entry.getName());
if (backupDir == null) {
backupDir = currentFile;
}
if (entry.isDirectory()) {
currentFile.mkdirs();
} else {
FileOutputStream fos = new FileOutputStream(currentFile);
byte[] buffer = new byte[bufferSize];
int numRead = 0;
while ((numRead = zipInputStream.read(buffer)) > -1) {
fos.write(buffer, 0, numRead);
}
}
entry = zipInputStream.getNextEntry();
}
}
return backupDir;
}
private void startTransaction() throws NotSupportedException, SystemException {
TransactionManager txnMgr = session.repository.transactionManager();
txnMgr.begin();
}
private void rollbackTransaction() throws SystemException, SecurityException, IllegalStateException {
TransactionManager txnMgr = session.repository.transactionManager();
txnMgr.rollback();
}
private void assertWorkspaces( JcrRepository newRepository,
String... workspaceNames ) throws RepositoryException {
Set<String> expectedNames = new HashSet<String>();
for (String expectedName : workspaceNames) {
expectedNames.add(expectedName);
}
Set<String> actualNames = new HashSet<String>();
JcrSession session = newRepository.login();
try {
for (String actualName : session.getWorkspace().getAccessibleWorkspaceNames()) {
actualNames.add(actualName);
}
} finally {
session.logout();
}
assertThat(actualNames, is(expectedNames));
}
private void queryContentInWorkspace( JcrRepository newRepository,
String workspaceName ) throws RepositoryException {
JcrSession session = newRepository.login();
try {
String statement = "SELECT [car:model], [car:year], [car:msrp] FROM [car:Car] AS car";
Query query = session.getWorkspace().getQueryManager().createQuery(statement, Query.JCR_SQL2);
QueryResult results = query.execute();
assertThat(results.getRows().getSize(), is(13L));
} finally {
session.logout();
}
}
private void assertContentInWorkspace( JcrRepository newRepository,
String workspaceName,
String... paths ) throws RepositoryException {
JcrSession session = workspaceName != null ? newRepository.login(workspaceName) : newRepository.login();
try {
session.getRootNode();
session.getNode("/Cars");
session.getNode("/Cars/Hybrid");
session.getNode("/Cars/Hybrid/Toyota Prius");
session.getNode("/Cars/Hybrid/Toyota Highlander");
session.getNode("/Cars/Hybrid/Nissan Altima");
session.getNode("/Cars/Sports/Aston Martin DB9");
session.getNode("/Cars/Sports/Infiniti G37");
session.getNode("/Cars/Luxury/Cadillac DTS");
session.getNode("/Cars/Luxury/Bentley Continental");
session.getNode("/Cars/Luxury/Lexus IS350");
session.getNode("/Cars/Utility/Land Rover LR2");
session.getNode("/Cars/Utility/Land Rover LR3");
session.getNode("/Cars/Utility/Hummer H3");
session.getNode("/Cars/Utility/Ford F-150");
session.getNode("/Cars/Utility/Toyota Land Cruiser");
for (String path : paths) {
session.getNode(path);
}
} finally {
session.logout();
}
}
private void assertContentNotInWorkspace( JcrRepository newRepository,
String workspaceName,
String... paths ) throws RepositoryException {
JcrSession session = workspaceName != null ? newRepository.login(workspaceName) : newRepository.login();
try {
session.getRootNode();
for (String path : paths) {
try {
session.getNode(path);
fail("Should not have found '" + path + "'");
} catch (PathNotFoundException e) {
// expected
}
}
} finally {
session.logout();
}
}
protected void assertNoProblems( Problems problems ) {
if (problems.hasProblems()) {
System.out.println(problems);
}
assertThat(problems.hasProblems(), is(false));
}
protected void loadContent() throws Exception {
importIntoWorkspace("default", "io/cars-system-view.xml");
importIntoWorkspace("ws2", "io/cars-system-view.xml");
importIntoWorkspace("ws3", "io/cars-system-view.xml");
}
protected void loadBinaryContent() throws Exception {
addFilesToWorkspace("default");
addFilesToWorkspace("ws2");
addFilesToWorkspace("ws3");
}
protected void addFilesToWorkspace(String workspaceName) throws Exception {
Session session = null;
try {
session = repository().login(workspaceName);
} catch (NoSuchWorkspaceException e) {
// Create the workspace ...
session().getWorkspace().createWorkspace(workspaceName);
// Create a new session ...
session = repository().login(workspaceName);
}
try {
for (int i = 0; i < BINARY_RESOURCES.length; i++) {
tools.uploadFile(session, "file_" + i, resourceStream(BINARY_RESOURCES[i]));
}
session.save();
} finally {
session.logout();
}
}
protected void assertFilesInWorkspace(String workspaceName) throws Exception {
Session session = repository().login(workspaceName);
try {
for (int i = 0; i < BINARY_RESOURCES.length; i++) {
String fileName = "/file_" + i;
Node file = session.getNode(fileName).getNode("jcr:content");
Binary binary = file.getProperty("jcr:data").getBinary();
assertNotNull(binary);
byte[] expectedContent = IoUtil.readBytes(resourceStream(BINARY_RESOURCES[i]));
byte[] actualContent = IoUtil.readBytes(binary.getStream());
assertArrayEquals("Binary content to valid for " + fileName, expectedContent, actualContent);
}
} finally {
session.logout();
}
}
protected void importIntoWorkspace( String workspaceName,
String resourcePath ) throws IOException, RepositoryException {
Session session = null;
try {
session = repository().login(workspaceName);
} catch (NoSuchWorkspaceException e) {
// Create the workspace ...
session().getWorkspace().createWorkspace(workspaceName);
// Create a new session ...
session = repository().login(workspaceName);
}
try {
importContent(session.getRootNode(), resourcePath, ImportUUIDBehavior.IMPORT_UUID_COLLISION_THROW);
} finally {
session.logout();
}
}
}