com.github.mikesafonov.smpp.assertj.MockSmppServerAssert Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of smpp-server-mock-assertj Show documentation
AssertJ asserts for smpp-server-mock
The newest version!
package com.github.mikesafonov.smpp.assertj;

import com.cloudhopper.smpp.pdu.CancelSm;
import com.cloudhopper.smpp.pdu.PduRequest;
import com.cloudhopper.smpp.pdu.SubmitSm;
import com.github.mikesafonov.smpp.server.MockSmppServer;
import org.assertj.core.api.AbstractAssert;

import java.util.List;

public class MockSmppServerAssert extends AbstractAssert {
    public MockSmppServerAssert(MockSmppServer mockSmppServer) {
        super(mockSmppServer, MockSmppServerAssert.class);
    }

    public MockSmppServerAssert receiveRequestsCount(int size) {
        isNotNull();
        checkMessagesCount(actual.getRequests(), size);
        return this;
    }

    public MockSmppServerAssert hasName(String name) {
        isNotNull();
        if(!actual.getName().equals(name)){
            failWithMessage("Expected name to be <%s> but was <%s>", name, actual.getName());
        }
        return this;
    }

    public MockSmppServerAssert hasPort(int port){
        isNotNull();
        if(actual.getPort() != port){
            failWithMessage("Expected port to be <%s> but was <%s>", port, actual.getPort());
        }
        return this;
    }

    public MockSmppServerAssert hasSystemId(String systemId){
        isNotNull();
        if(!actual.getSystemId().equals(systemId)){
            failWithMessage("Expected systemId to be <%s> but was <%s>", systemId, actual.getSystemId());
        }
        return this;
    }

    public SubmitSmListAssert messages() {
        return new SubmitSmListAssert(actual.getSubmitSmMessages());
    }

    public CancelSmListAssert cancelMessages() {
        return new CancelSmListAssert(actual.getCancelSmMessages());
    }

    public SubmitSmAssert hasSingleMessage() {
        return new SubmitSmAssert(checkAndGetSubmitSm());
    }

    public CancelSmAssert hasSingleCancelMessage() {
        return new CancelSmAssert(checkAndGetCancelSm());
    }

    public PduRequestListAssert requests() {
        return new PduRequestListAssert(actual.getRequests());
    }


    private void checkMessagesCount(List requests, int expectedCount) {
        if (requests.size() != expectedCount) {
            failWithMessage("Expected messages size to be <%s> but was <%s>", expectedCount, requests.size());
        }
    }

    private SubmitSm checkAndGetSubmitSm() {
        isNotNull();
        List submitSmMessages = actual.getSubmitSmMessages();
        checkMessagesCount(submitSmMessages, 1);
        return submitSmMessages.get(0);
    }

    private CancelSm checkAndGetCancelSm() {
        isNotNull();
        List cancelSmMessages = actual.getCancelSmMessages();
        checkMessagesCount(cancelSmMessages, 1);
        return cancelSmMessages.get(0);
    }
}