io.split.qos.server.integrations.slack.commands.SlackTestsCommand Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of qosrunner Show documentation
Framework for running JUnit Tests as a Continous Service (QoS)
There is a newer version: 22.4.3
package io.split.qos.server.integrations.slack.commands;

import com.google.common.base.Preconditions;
import com.google.inject.Inject;
import com.google.inject.Singleton;
import com.google.inject.name.Named;
import com.ullink.slack.simpleslackapi.SlackAttachment;
import com.ullink.slack.simpleslackapi.SlackSession;
import com.ullink.slack.simpleslackapi.events.SlackMessagePosted;
import io.split.qos.server.QOSServerState;
import io.split.qos.server.integrations.slack.commandintegration.SlackCommand;
import io.split.qos.server.integrations.slack.commandintegration.SlackCommandGetter;
import io.split.qos.server.modules.QOSServerModule;
import io.split.qos.server.util.SlackMessageSender;
import io.split.qos.server.util.TestId;
import io.split.testrunner.util.DateFormatter;
import io.split.testrunner.util.SlackColors;

import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

/**
 * Shows the tests that matches the input.
 */
@Singleton
public class SlackTestsCommand extends SlackAbstractCommand {
    private final DateFormatter dateFormatter;
    private final QOSServerState state;

    @Inject
    public SlackTestsCommand(
            SlackColors slackColors,
            SlackCommandGetter slackCommandGetter,
            DateFormatter dateFormatter,
            SlackMessageSender slackMessageSender,
            QOSServerState state,
            @Named(QOSServerModule.QOS_SERVER_NAME) String serverName) {
        super(slackColors, serverName, slackMessageSender, slackCommandGetter);
        this.dateFormatter = Preconditions.checkNotNull(dateFormatter);
        this.state = Preconditions.checkNotNull(state);
    }

    @Override
    public boolean test(SlackMessagePosted messagePosted, SlackSession session) {
        SlackCommand slackCommand = command(messagePosted);
        List arguments = slackCommand.arguments();

        Map tests = null;
        if (arguments.size() == 0) {
            tests = state.tests();
        } else if (arguments.size() == 1) {
            tests = state.tests(arguments.get(0));
        } else {
            tests = state.tests(arguments.get(0), arguments.get(1));
        }

        List toBeAdded = tests
                .entrySet()
                .stream()
                .sorted((o1, o2) -> {
                    QOSServerState.TestStatus statusO1 = o1.getValue();
                    QOSServerState.TestStatus statusO2 = o2.getValue();
                    // First finished, second did not, so second always goes up.
                    if ((statusO1.hasFinished()) && (!statusO2.hasFinished())) {
                        return 1;
                    }
                    // Second finished, first did not, so first always goes up.
                    if ((!statusO1.hasFinished()) && (statusO2.hasFinished())) {
                        return -1;
                    }
                    // None finished, order by name
                    if ((!statusO1.hasFinished() && !statusO2.hasFinished())) {
                        return o1.getKey().compareTo(o2.getKey());
                    }
                    // Both finished.
                    // First order by failure (failed first)
                    // If same status order by name.
                    int comparison = statusO1.succeeded().compareTo(statusO2.succeeded());
                    if (comparison != 0) {
                        return comparison;
                    } else {
                        return o1.getKey().compareTo(o2.getKey());
                    }
                })
                .map(value -> {
                    TestId id = value.getKey();
                    SlackAttachment testAttachment = new SlackAttachment("", "", id.toString(), null);
                    QOSServerState.TestStatus status = value.getValue();
                    if (status.succeeded() == null) {
                        testAttachment.setColor(colors().getWarning());
                    } else if (status.succeeded()) {
                        testAttachment.setColor(colors().getSuccess());
                    } else {
                        testAttachment.setColor(colors().getFailed());
                    }
                    return testAttachment;
                })
                .collect(Collectors.toList());
        messageSender().sendPartition(slackCommand.command(), session, messagePosted.getChannel(), toBeAdded);
        return true;
    }

    @Override
    public String description() {
        return "Finds the tests that matches the parameters. No parameters gets all tests.";
    }

    @Override
    public String arguments() {
        return "[server-name (optional)] tests [class name (optional)] [test name (optional)]: ";
    }

    @Override
    public boolean acceptsArguments(List arguments) {
        Preconditions.checkNotNull(arguments);
        return arguments.size() < 3;
    }
}