SourceServicesImplTest.java

/*
 * Copyright 2013 Red Hat, Inc. and/or its affiliates.
 *
 * 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.kie.workbench.common.services.backend.source;

import java.net.URI;
import java.net.URISyntaxException;
import java.util.ArrayList;
import java.util.List;
import javax.enterprise.inject.Instance;

import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import org.uberfire.java.nio.file.Files;
import org.uberfire.java.nio.file.Path;

import static org.junit.Assert.*;
import static org.mockito.Mockito.*;

public class SourceServicesImplTest {

    private Instance instance;
    private List<SourceService> sourceServices;
    private List<Path> pathsToDelete;

    @Before
    public void setUp() throws Exception {
        instance = mock(Instance.class);
        sourceServices = new ArrayList<SourceService>();
        pathsToDelete = new ArrayList<Path>();
    }

    @After
    public void clearDown() {
        for (final Path p : pathsToDelete) {
            Files.delete(p);
        }
    }

    @Test
    public void testSomethingSimple() throws Exception {
        addToList(getSourceService(".drl"));
        assertTrue(new SourceServicesImpl(instance).hasServiceFor(makePath("myFile",
                ".drl")));
    }

    @Test
    public void testMissing() throws Exception {
        addToList(getSourceService(".notHere"));
        assertFalse(new SourceServicesImpl(instance).hasServiceFor(makePath("myFile",
                ".drl")));
    }

    @Test
    public void testShorter() throws Exception {
        SourceService DRL = getSourceService(".drl");
        SourceService modelDRL = getSourceService(".model.drl");
        addToList(DRL, modelDRL);

        assertEquals(DRL, new SourceServicesImpl(instance).getServiceFor(makePath("myFile",
                ".drl")));

        sourceServices.clear();

        addToList(modelDRL, DRL);

        assertEquals(DRL, new SourceServicesImpl(instance).getServiceFor(makePath("myFile",
                ".drl")));
    }

    @Test
    public void testLonger() throws Exception {
        SourceService DRL = getSourceService(".drl");
        SourceService modelDRL = getSourceService(".model.drl");
        addToList(DRL,
                modelDRL);

        assertEquals(modelDRL, new SourceServicesImpl(instance).getServiceFor(makePath("myFile",
                ".model.drl")));

        sourceServices.clear();

        addToList(modelDRL,
                DRL);

        assertEquals(modelDRL, new SourceServicesImpl(instance).getServiceFor(makePath("myFile",
                ".model.drl")));
    }

    private SourceService getSourceService(final String extension) {
        return new SourceService() {

            @Override
            public boolean accepts(final Path path) {
                final String uri = path.toUri().toString();
                return uri.substring(uri.length() - extension.length()).equals(extension);
            }

            @Override
            public String getSource(Path path, Object model) {
                return null;
            }

            @Override
            public String getSource(Path path) {
                return null;
            }

            @Override
            public String getPattern() {
                return extension;
            }
        };
    }

    private void addToList(SourceService... services) {

        for (SourceService service : services) {
            sourceServices.add(service);
        }

        when(
                instance.iterator()
        ).thenReturn(
                sourceServices.iterator()
        );
    }

    private Path makePath(final String prefix,
                          final String suffix) throws URISyntaxException {
        Path path = mock(Path.class);


        when(
                path.toUri()
        ).thenReturn(
                new URI(prefix + suffix)
        );
        return path;
    }

}