com.addthis.meshy.service.file.TestVFS Maven / Gradle / Ivy

Go to download
/*
 * 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 com.addthis.meshy.service.file;

import java.io.File;

import java.util.Iterator;
import java.util.LinkedList;
import java.util.Map;

import java.nio.file.PathMatcher;

import com.addthis.meshy.LocalFileHandler;
import com.addthis.meshy.LocalFileSystem;
import com.addthis.meshy.MeshyClient;
import com.addthis.meshy.MeshyServer;
import com.addthis.meshy.TestMesh;
import com.addthis.meshy.VirtualFileInput;
import com.addthis.meshy.VirtualFileReference;

import org.junit.After;
import org.junit.Before;
import org.junit.Test;


public class TestVFS extends TestMesh {

    @Before
    public void setup() {
        System.setProperty("mesh.local.handlers", DummyHandler.class.getName());
        LocalFileSystem.reloadHandlers();
        MeshyServer.resetFileSystems();
    }

    @After @Override
    public void cleanup() {
        super.cleanup();
        System.setProperty("mesh.local.handlers", "");
        LocalFileSystem.reloadHandlers();
        MeshyServer.resetFileSystems();
    }

    @Test
    public void testVFS() throws Exception {
        final MeshyServer server = getServer("src/test");
        final MeshyClient client = getClient(server);
        FileSource files = new FileSource(client, new String[]{"*"});
        files.waitComplete();
        Map map = files.getFileMap();
        log.info("map={}", map);
        checkFile(map, new FileReference("/dummy", 0, 0).setHostUUID(server.getUUID()));
    }

    public static class DummyHandler implements LocalFileHandler {

        LinkedList list = new LinkedList<>();
        VirtualFileReference ref = new DummyReference();

        public DummyHandler() {
            list.add(ref);
        }

        @Override
        public boolean canHandleDirectory(File dir) {
            return true;
        }

        @Override
        public Iterator listFiles(File dir, PathMatcher filter) {
            return list.iterator();
        }

        @Override
        public VirtualFileReference getFile(File dir, String name) {
            return ref;
        }
    }

    static class DummyReference implements VirtualFileReference {

        @Override
        public String getName() {
            return "dummy";
        }

        @Override
        public long getLastModified() {
            return 0;
        }

        @Override
        public long getLength() {
            return 0;
        }

        @Override
        public Iterator listFiles(PathMatcher filter) {
            return null;
        }

        @Override
        public VirtualFileReference getFile(String name) {
            return null;
        }

        @Override
        public VirtualFileInput getInput(Map options) {
            return null;
        }
    }
}