com.chutneytesting.action.ssh.sshd.SshServerMock Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of action-impl Show documentation

There is a newer version: 2.9.0

package com.chutneytesting.action.ssh.sshd;

import static java.util.Collections.unmodifiableList;
import static java.util.Optional.empty;
import static java.util.Optional.of;
import static java.util.stream.Collectors.toList;

import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
import java.util.Optional;
import org.apache.sshd.server.SshServer;

public class SshServerMock {

    private final SshServer sshServer;

    private final List stubs;
    private final List usedStubs;
    private final List commands = new ArrayList<>();

    public SshServerMock(SshServer sshServer, List stubs) {
        this.sshServer = sshServer;
        this.stubs = stubs;
        this.usedStubs = stubs.stream().map(s -> false).collect(toList());
    }

    public Optional addCommand(String command) {
        commands.add(command);
        int lastCommandIndex = commands.size() - 1;
        if (lastCommandIndex < stubs.size()) {
            usedStubs.set(lastCommandIndex, true);
            return of(stubs.get(lastCommandIndex));
        }
        return empty();
    }

    public List commands() {
        return unmodifiableList(commands);
    }

    public List stubs() {
        return unmodifiableList(stubs);
    }

    public String command(int i) {
        return commands.get(i);
    }

    public boolean allStubsUsed() {
        return usedStubs.stream().reduce((b1, b2) -> b1 && b2).orElse(false);
    }

    public void start() throws IOException {
        sshServer.start();
    }

    public void stop() throws IOException {
        sshServer.stop();
    }

    public String host() {
        return sshServer.getHost();
    }

    public int port() {
        return sshServer.getPort();
    }

    public boolean isStarted() {
        return sshServer.isStarted();
    }

    public boolean isClosed() {
        return sshServer.isClosed();
    }

    public boolean isOpen() {
        return sshServer.isOpen();
    }
}