ProjectTemplateDescriptionLoaderTest.java

/*******************************************************************************
 * Copyright (c) 2012-2015 Codenvy, S.A.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *   Codenvy, S.A. - initial API and implementation
 *******************************************************************************/
package org.eclipse.che.api.project.server;

import org.eclipse.che.api.project.server.type.ProjectType;
import org.eclipse.che.api.project.shared.dto.ImportSourceDescriptor;
import org.eclipse.che.api.project.shared.dto.ProjectTemplateDescriptor;
import org.eclipse.che.commons.lang.NameGenerator;
import org.junit.Assert;
import org.junit.Before;
import org.junit.Test;

import java.net.URL;
import java.util.HashSet;
import java.util.List;
import java.util.Set;

import static org.mockito.Matchers.anyListOf;
import static org.mockito.Matchers.eq;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.never;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;

/**
 * @author Vitaly Parfonov
 */
public class ProjectTemplateDescriptionLoaderTest {

    ProjectTemplateRegistry templateRegistry;

    private Set<ProjectType> pts;

    private String embedTypeId = "embed_type";

    private String projectType1 = "type1";

    private String projectType2 = "type2";

    @Before
    public void setUp() {
        templateRegistry = mock(ProjectTemplateRegistry.class);
        pts = new HashSet<>();
        ProjectType embedType = mock(ProjectType.class);
        when(embedType.getId()).thenReturn(embedTypeId);
        ProjectType type1 = mock(ProjectType.class);
        when(type1.getId()).thenReturn(projectType1);
        ProjectType type2 = mock(ProjectType.class);
        when(type2.getId()).thenReturn(projectType2);
        pts.add(type1);
        pts.add(type2);
        pts.add(embedType);
    }

    @Test
    public void testWithOutConfig() {
        ProjectTemplateDescriptionLoader loader = new ProjectTemplateDescriptionLoader(null, null, pts, templateRegistry);
        loader.start();
        verify(templateRegistry).register(eq(embedTypeId), anyListOf(ProjectTemplateDescriptor.class));
        verify(templateRegistry, never()).register(eq(projectType1), anyListOf(ProjectTemplateDescriptor.class));
        verify(templateRegistry, never()).register(eq(projectType2), anyListOf(ProjectTemplateDescriptor.class));
    }


    @Test
    //load templates from given dir
    public void testWithConfig() {
        URL resource = Thread.currentThread().getContextClassLoader().getResource("pt-tmpl");
        Assert.assertNotNull(resource);
        ProjectTemplateDescriptionLoader loader = new ProjectTemplateDescriptionLoader(resource.getFile(), null, pts, templateRegistry);
        loader.start();
        verify(templateRegistry, never()).register(eq(embedTypeId), anyListOf(ProjectTemplateDescriptor.class));
        verify(templateRegistry).register(eq(projectType1), anyListOf(ProjectTemplateDescriptor.class));
        verify(templateRegistry).register(eq(projectType2), anyListOf(ProjectTemplateDescriptor.class));
    }

    @Test
    //load templates from given dir
    public void testWithConfig2() {
        ProjectType type3 = mock(ProjectType.class);
        when(type3.getId()).thenReturn("type3");
        pts.add(type3);
        URL resource = Thread.currentThread().getContextClassLoader().getResource("pt-tmpl");
        Assert.assertNotNull(resource);
        ProjectTemplateDescriptionLoader loader = new ProjectTemplateDescriptionLoader(resource.getFile(), null, pts, templateRegistry);
        loader.start();
        verify(templateRegistry).register(eq(projectType1), anyListOf(ProjectTemplateDescriptor.class));
        verify(templateRegistry).register(eq(projectType2), anyListOf(ProjectTemplateDescriptor.class));
        verify(templateRegistry, never()).register(eq("type3"), anyListOf(ProjectTemplateDescriptor.class));
    }

    @Test
    public void testWithConfigAndReplaceLocations() {
        ProjectTemplateRegistry templateRegistry = new ProjectTemplateRegistry();
        Set<ProjectType> pts = new HashSet<>();
        ProjectType type2 = mock(ProjectType.class);
        when(type2.getId()).thenReturn(projectType2);
        pts.add(type2);
        URL resource = Thread.currentThread().getContextClassLoader().getResource("pt-tmpl");
        Assert.assertNotNull(resource);
        String location = NameGenerator.generate("location", 5);
        ProjectTemplateDescriptionLoader loader = new ProjectTemplateDescriptionLoader(resource.getFile(), location, pts, templateRegistry);
        loader.start();
        List<ProjectTemplateDescriptor> type = templateRegistry.getTemplates(projectType2);
        Assert.assertNotNull(type);
        Assert.assertEquals(1, type.size());
        ProjectTemplateDescriptor templateDescriptor = type.get(0);
        Assert.assertNotNull(templateDescriptor);
        ImportSourceDescriptor source = templateDescriptor.getSource();
        Assert.assertNotNull(source);
        Assert.assertNotNull(source.getLocation());
        Assert.assertTrue(source.getLocation().contains(location));
    }


}