/*
 * eXist Open Source Native XML Database
 * Copyright (C) 2001-2017 The eXist Project
 * http://exist-db.org
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public License
 * as published by the Free Software Foundation; either version 2
 * of the License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
 */

package org.exist.storage;

import org.exist.EXistException;
import org.exist.test.ExistEmbeddedServer;
import org.junit.ClassRule;
import org.junit.Test;
import org.junit.rules.TemporaryFolder;

import java.io.IOException;
import java.nio.file.Path;
import java.util.Optional;
import java.util.zip.ZipEntry;
import java.util.zip.ZipFile;

import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertTrue;

public class DataBackupTest {

    @ClassRule
    public static ExistEmbeddedServer existEmbeddedServer = new ExistEmbeddedServer();

    @ClassRule
    public static TemporaryFolder folder = new TemporaryFolder();

    @Test
    public void backup() throws InterruptedException, IOException {
        final TestableDataBackup dataBackup = new TestableDataBackup(folder.getRoot().toPath());
        existEmbeddedServer.getBrokerPool().triggerSystemTask(dataBackup);

        while(!dataBackup.isCompleted()) {
            Thread.sleep(100);
        }

        final Optional<Path> lastBackup = dataBackup.getLastBackup();
        assertTrue(lastBackup.isPresent());

        final ZipFile zipFile = new ZipFile(lastBackup.get().toFile());
        assertNotNull(zipFile.getEntry("collections.dbx"));
        assertNotNull(zipFile.getEntry("dom.dbx"));
        assertNotNull(zipFile.getEntry("structure.dbx"));
        assertNotNull(zipFile.getEntry("symbols.dbx"));
        assertNotNull(zipFile.getEntry("values.dbx"));
    }

    private class TestableDataBackup extends DataBackup {
        private volatile boolean completed = false;

        public TestableDataBackup(final Path destination) {
            super(destination);
        }

        @Override
        public void execute(final DBBroker broker) throws EXistException {
            super.execute(broker);
            completed = true;
        }

        public boolean isCompleted() {
            return completed;
        }
    }
}