AbstractMultipleJobCommandTest.java

package hudson.plugins.im.bot;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNull;
import static org.junit.Assert.assertSame;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;
import hudson.model.AbstractProject;
import hudson.model.FreeStyleProject;
import hudson.model.TopLevelItem;
import hudson.model.View;
import hudson.plugins.im.Sender;
import hudson.plugins.im.bot.AbstractMultipleJobCommand.Mode;
import hudson.plugins.im.tools.Pair;

import java.util.ArrayList;
import java.util.Collection;
import java.util.HashSet;
import java.util.List;

import org.apache.commons.lang.StringUtils;
import org.junit.Test;

public class AbstractMultipleJobCommandTest {

    @Test
    @SuppressWarnings("unchecked")
    public void testGetJobByName() throws CommandException {
        String projectName = "project name with spaces";
        
        AbstractProject project = mock(AbstractProject.class);
        
        JobProvider jobProvider = mock(JobProvider.class);
        when(jobProvider.getJobByName(projectName)).thenReturn(project);
        
        String[] projArgs = StringUtils.split(projectName);
        String[] args = new String[1 + projArgs.length];
        args[0] = "health";
        for(int i=0; i < projArgs.length; i++) {
            args[i+1] = projArgs[i];
        }
        
        HealthCommand cmd = new HealthCommand();
        cmd.setJobProvider(jobProvider);
        
        List<AbstractProject<?, ?>> projects = new ArrayList<AbstractProject<?,?>>();
        Pair<Mode, String> pair = cmd.getProjects(new Sender("sender"), args, projects);
        
        assertEquals(Mode.SINGLE, pair.getHead());
        assertNull(pair.getTail());
        
        assertEquals(1, projects.size());
        assertSame(project, projects.get(0));
    }
    
    @Test(expected=CommandException.class)
    public void testUnknownJobName() throws CommandException {
        JobProvider jobProvider = mock(JobProvider.class);
        
        String[] args = {"health", "doesnt-matter-jobname"};
        
        HealthCommand cmd = new HealthCommand();
        cmd.setJobProvider(jobProvider);
        
        List<AbstractProject<?, ?>> projects = new ArrayList<AbstractProject<?,?>>();
        cmd.getProjects(new Sender("sender"), args, projects);
    }
    
    @Test
    public void testGetByView() throws Exception {
        FreeStyleProject project = mock(FreeStyleProject.class);
        
        View mockView = mock(View.class);
        Collection<TopLevelItem> projectsForView = new HashSet<TopLevelItem>();
        projectsForView.add(project);
        when(mockView.getItems()).thenReturn(projectsForView);

        String viewName = "myView";
        
        JobProvider jobProvider = mock(JobProvider.class);
        when(jobProvider.getView(viewName)).thenReturn(mockView);
        
        String[] args = { "health", "-v", viewName };
        HealthCommand cmd = new HealthCommand();
        cmd.setJobProvider(jobProvider);
        
        List<AbstractProject<?, ?>> projects = new ArrayList<AbstractProject<?,?>>();
        Pair<Mode,String> pair = cmd.getProjects(new Sender("sender"), args, projects);

        assertEquals(Mode.VIEW, pair.getHead());
        assertEquals(viewName, pair.getTail());
        
        assertEquals(1, projects.size());
        assertSame(project, projects.get(0));
    }
}