source: src/test/java/geniusweb/partiesserver/AvailablePartiesUpdaterTest.java@ 25

package geniusweb.partiesserver;

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

import java.io.File;
import java.io.IOException;
import java.nio.charset.Charset;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.Arrays;
import java.util.Collection;
import java.util.HashMap;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;
import java.util.logging.Handler;
import java.util.logging.Level;
import java.util.logging.LogRecord;
import java.util.logging.Logger;

import org.junit.Test;

import geniusweb.partiesserver.repository.AvailableParty;
import tudelft.utilities.files.FileInfo;
import tudelft.utilities.files.FileWatcher;
import tudelft.utilities.immutablelist.Tuple;
import tudelft.utilities.listener.Listenable;
import tudelft.utilities.repository.Repository;

public class AvailablePartiesUpdaterTest {
        private static final String RANDOMPARTY = "target/jars/randomparty-1.5.6.jar";
        private static final int TESTRATE = 200; // check file changes every 200ms
        @SuppressWarnings("unchecked")
        private Repository<String, AvailableParty> repo = new TestRepo();
        private AvailablePartiesUpdater updater;
        private Path tmpdir;
        private final List<LogRecord> logrecords = new LinkedList<>();
        private Path RNDPARTY;

        private void init() throws IOException {
                tmpdir = Files.createTempDirectory("geniuswebparties");
                RNDPARTY = tmpdir.resolve("randomparty.jar");
                logrecords.clear();
                Logger.getLogger("partiesserver").addHandler(new Handler() {

                        @Override
                        public void publish(LogRecord record) {
                                if (record.getLevel().intValue() >= Level.WARNING.intValue()) {
                                        logrecords.add(record);
                                }
                        }

                        @Override
                        public void flush() {
                                // TODO Auto-generated method stub

                        }

                        @Override
                        public void close() throws SecurityException {
                                // TODO Auto-generated method stub

                        }
                });

                updater = new AvailablePartiesUpdater(repo) {
                        @Override
                        protected Path getRootDir() {
                                return tmpdir;
                        }

                        @Override
                        protected Listenable<Tuple<FileInfo, FileInfo>> getFileWatcher(
                                        File directory) {
                                return new FileWatcher(directory, TESTRATE, 1);
                        }
                };
        }

        @Test
        public void smokeTest() throws IOException {
                init();
        }

        @Test
        public void initiallyEmptyTest() throws IOException, InterruptedException {
                init();
                Thread.sleep(2 * TESTRATE);
                assertTrue(repo.list().isEmpty());
                assertTrue(logrecords.isEmpty());
        }

        @Test
        public void IgnoreBadFileTest() throws IOException, InterruptedException {
                init();

                Path file = tmpdir.resolve("file1.txt");
                Files.write(file, Arrays.asList("blah blah"), Charset.forName("UTF-8"));

                Thread.sleep(2 * TESTRATE);
                assertTrue(repo.list().isEmpty());
                assertEquals(1, logrecords.size());
                assertTrue(logrecords.get(0).getMessage()
                                .contains("Can not load class from file"));
                assertTrue(logrecords.get(0).getThrown().getMessage()
                                .contains("must be a file and have extension .jar"));

        }

        @Test
        public void IgnoreBadJarFileTest()
                        throws IOException, InterruptedException {
                init();

                Path file = tmpdir.resolve("file1.jar");
                Files.write(file, Arrays.asList("blah blah"), Charset.forName("UTF-8"));

                Thread.sleep(2 * TESTRATE);
                assertTrue(repo.list().isEmpty());
                assertEquals(1, logrecords.size());
                assertTrue(logrecords.get(0).getMessage()
                                .contains("Can not load class from file"));
                assertTrue(logrecords.get(0).getThrown().getMessage()
                                .contains("error in opening zip file"));

        }

        @Test
        public void ignoreEmptyJarFileTest()
                        throws IOException, InterruptedException {
                init();
                Files.copy(getClass().getResourceAsStream("/empty.jar"),
                                tmpdir.resolve("empty.jar"));

                Thread.sleep(2 * TESTRATE);
                assertTrue(repo.list().isEmpty());
                assertEquals(1, logrecords.size());
                assertTrue(logrecords.get(0).getMessage()
                                .contains("Can not load class from file"));
                assertTrue(logrecords.get(0).getThrown().getMessage().contains(
                                "jar file does not contain manifest with main class"));
        }

        @Test
        public void randomPartyJarTest() throws IOException, InterruptedException {
                init();
                Files.copy(Paths.get(RANDOMPARTY), tmpdir.resolve(RNDPARTY));

                Thread.sleep(2 * TESTRATE);
                assertTrue(logrecords.isEmpty());
                assertEquals(1, repo.list().size());
        }

        @Test
        public void removeJarTest() throws IOException, InterruptedException {
                init();
                Files.copy(Paths.get(RANDOMPARTY), RNDPARTY);
                Thread.sleep(2 * TESTRATE);
                assertEquals(1, repo.list().size());

                Files.delete(RNDPARTY);
                Thread.sleep(2 * TESTRATE);
                assertEquals(0, repo.list().size());
        }

        @Test
        public void removeRootTest() throws IOException, InterruptedException {
                init();
                Files.copy(Paths.get(RANDOMPARTY), RNDPARTY);
                Thread.sleep(2 * TESTRATE);
                assertEquals(1, repo.list().size());

                Files.delete(RNDPARTY);
                Files.delete(tmpdir);
                Thread.sleep(2 * TESTRATE);
                assertEquals(0, repo.list().size());
                assertEquals(1, logrecords.size());
                assertTrue(logrecords.get(0).getMessage()
                                .contains("Can not read repository root directory"));

        }

}

/**
 * Basic repo to track what the updater is doing.
 */
class TestRepo implements Repository<String, AvailableParty> {
        private Map<String, AvailableParty> map = new HashMap<String, AvailableParty>();

        @Override
        public Collection<AvailableParty> list() {
                return map.values();
        }

        @Override
        public AvailableParty get(String id) {
                return map.get(id);
        }

        @Override
        public void put(AvailableParty entity) {
                if (map.containsKey(entity.getID())) {
                        throw new IllegalStateException("repo already contains entity");
                }
                map.put(entity.getID(), entity);
        }

        @Override
        public void remove(String id) {
                map.remove(id);
        }

        @Override
        public void replace(AvailableParty entity) {
                if (!map.containsKey(entity.getID())) {
                        throw new IllegalStateException("repo does not contain entity");
                }
                map.put(entity.getID(), entity);
        }

}