amazon-kinesis-client/amazon-kinesis-client-multilang/src/test/java/software/amazon/kinesis/multilang/Matchers.java

/*
 *  Copyright 2017 Amazon.com, Inc. or its affiliates. All Rights Reserved.
 *
 *  Licensed under the Amazon Software License (the "License").
 *  You may not use this file except in compliance with the License.
 *  A copy of the License is located at
 *
 *  http://aws.amazon.com/asl/
 *
 *  or in the "license" file accompanying this file. This file is distributed
 *  on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
 *  express or implied. See the License for the specific language governing
 *  permissions and limitations under the License.
 */
package software.amazon.kinesis.multilang;

import static org.hamcrest.CoreMatchers.equalTo;
import static org.hamcrest.CoreMatchers.nullValue;

import org.hamcrest.Description;
import org.hamcrest.Matcher;
import org.hamcrest.TypeSafeDiagnosingMatcher;

import software.amazon.kinesis.retrieval.kpl.ExtendedSequenceNumber;
import software.amazon.kinesis.lifecycle.events.InitializationInput;

public class Matchers {

    public static Matcher<InitializationInput> withInit(InitializationInput initializationInput) {
        return new InitializationInputMatcher(initializationInput);
    }

    public static class InitializationInputMatcher extends TypeSafeDiagnosingMatcher<InitializationInput> {

        private final Matcher<String> shardIdMatcher;
        private final Matcher<ExtendedSequenceNumber> sequenceNumberMatcher;

        public InitializationInputMatcher(InitializationInput input) {
            shardIdMatcher = equalTo(input.shardId());
            sequenceNumberMatcher = withSequence(input.extendedSequenceNumber());
        }

        @Override
        protected boolean matchesSafely(final InitializationInput item, Description mismatchDescription) {

            boolean matches = true;
            if (!shardIdMatcher.matches(item.shardId())) {
                matches = false;
                shardIdMatcher.describeMismatch(item.shardId(), mismatchDescription);
            }
            if (!sequenceNumberMatcher.matches(item.extendedSequenceNumber())) {
                matches = false;
                sequenceNumberMatcher.describeMismatch(item, mismatchDescription);
            }

            return matches;
        }

        @Override
        public void describeTo(Description description) {
            description.appendText("An InitializationInput matching: { shardId: ").appendDescriptionOf(shardIdMatcher)
                    .appendText(", sequenceNumber: ").appendDescriptionOf(sequenceNumberMatcher).appendText(" }");
        }
    }

    public static Matcher<ExtendedSequenceNumber> withSequence(ExtendedSequenceNumber extendedSequenceNumber) {
        if (extendedSequenceNumber == null) {
            return nullValue(ExtendedSequenceNumber.class);
        }
        return new ExtendedSequenceNumberMatcher(extendedSequenceNumber);
    }

    public static class ExtendedSequenceNumberMatcher extends TypeSafeDiagnosingMatcher<ExtendedSequenceNumber> {

        private final Matcher<String> sequenceNumberMatcher;
        private final Matcher<Long> subSequenceNumberMatcher;

        public ExtendedSequenceNumberMatcher(ExtendedSequenceNumber extendedSequenceNumber) {
            sequenceNumberMatcher = equalTo(extendedSequenceNumber.sequenceNumber());
            subSequenceNumberMatcher = equalTo(extendedSequenceNumber.subSequenceNumber());
        }

        @Override
        protected boolean matchesSafely(ExtendedSequenceNumber item, Description mismatchDescription) {

            boolean matches = true;
            if (!sequenceNumberMatcher.matches(item.sequenceNumber())) {
                matches = false;
                mismatchDescription.appendDescriptionOf(sequenceNumberMatcher);
            }
            if (!subSequenceNumberMatcher.matches(item.subSequenceNumber())) {
                matches = false;
                mismatchDescription.appendDescriptionOf(subSequenceNumberMatcher);
            }

            return matches;
        }

        @Override
        public void describeTo(Description description) {
            description.appendText("An ExtendedSequenceNumber matching: { sequenceNumber: ")
                    .appendDescriptionOf(sequenceNumberMatcher).appendText(", subSequenceNumber: ")
                    .appendDescriptionOf(subSequenceNumberMatcher);
        }
    }

}