/src/main/java/com/atlassian/bamboo/plugin/dotnet/tests/mstest/MSTestCollectorConfigurator.java

package com.atlassian.bamboo.plugin.dotnet.tests.mstest;



import com.atlassian.bamboo.collections.ActionParametersMap;

import com.atlassian.bamboo.plugin.dotnet.support.AbstractDotNetTaskConfigurator;

import com.atlassian.bamboo.task.TaskConfigConstants;

import com.atlassian.bamboo.task.TaskDefinition;

import com.atlassian.bamboo.task.TaskTestResultsSupport;

import com.atlassian.bamboo.utils.error.ErrorCollection;

import com.google.common.collect.Sets;

import org.apache.commons.lang3.StringUtils;

import org.jetbrains.annotations.NotNull;

import org.jetbrains.annotations.Nullable;



import java.util.Map;

import java.util.Set;



public class MSTestCollectorConfigurator extends AbstractDotNetTaskConfigurator implements TaskTestResultsSupport

{

    // ------------------------------------------------------------------------------------------------------- Constants



    private static final Set<String> FIELDS_TO_COPY = Sets.newHashSet(

            MSTestCollectorTaskType.TEST_DIRECTORY, TaskConfigConstants.CFG_TEST_OUTDATED_RESULTS_FILE

    );



    // ------------------------------------------------------------------------------------------------- Type Properties

    // ---------------------------------------------------------------------------------------------------- Dependencies

    // ---------------------------------------------------------------------------------------------------- Constructors

    // ----------------------------------------------------------------------------------------------- Interface Methods



    @Override

    public void validate(@NotNull final ActionParametersMap params, @NotNull final ErrorCollection errorCollection)

    {

        final String directory = params.getString(MSTestCollectorTaskType.TEST_DIRECTORY);

        if (StringUtils.isEmpty(directory))

        {

            errorCollection.addError(MSTestCollectorTaskType.TEST_DIRECTORY, textProvider.getText("mstest.testDirectory.error"));

        }

    }



    @Override

    public void populateContextForView(@NotNull final Map<String, Object> context, @NotNull final TaskDefinition taskDefinition)

    {

        super.populateContextForView(context, taskDefinition);

        taskConfiguratorHelper.populateContextWithConfiguration(context, taskDefinition, FIELDS_TO_COPY);

    }



    @Override

    public void populateContextForEdit(@NotNull final Map<String, Object> context, @NotNull final TaskDefinition taskDefinition)

    {

        super.populateContextForEdit(context, taskDefinition);

        taskConfiguratorHelper.populateContextWithConfiguration(context, taskDefinition, FIELDS_TO_COPY);

    }



    @NotNull

    @Override

    public Map<String, String> generateTaskConfigMap(@NotNull final ActionParametersMap params, @Nullable final TaskDefinition previousTaskDefinition)

    {

        final Map<String, String> map = super.generateTaskConfigMap(params, previousTaskDefinition);

        taskConfiguratorHelper.populateTaskConfigMapWithActionParameters(map, params, FIELDS_TO_COPY);

        return map;

    }



    @Override

    public boolean taskProducesTestResults(@NotNull final TaskDefinition taskDefinition)

    {

        return true;

    }

}