io.sphere.sdk.test.ReferenceAssert Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of test-lib Show documentation

test-lib

There is a newer version: 1.0.0-M12

package io.sphere.sdk.test;

import io.sphere.sdk.models.DefaultModel;
import io.sphere.sdk.models.Reference;
import io.sphere.sdk.models.Referenceable;
import org.fest.assertions.Assertions;
import org.fest.assertions.GenericAssert;

public class ReferenceAssert extends GenericAssert> {

    protected ReferenceAssert(final Reference actual) {
        super(ReferenceAssert.class, actual);
    }

    public static ReferenceAssert assertThat(final Reference actual) {
        return new ReferenceAssert(actual);
    }

    public ReferenceAssert hasId(final String id) {
        Assertions.assertThat(actual.getId()).isEqualTo(id);
        return this;
    }

    public ReferenceAssert hasAnExpanded(final DefaultModel model) {
        checkIsExpanded();
        if (!actual.getObj().get().equals(model)) {
            failIfCustomMessageIsSet();
            throw failure(String.format("%s does not contain an expanded %s.", actual, model));
        }
        return this;
    }

    public void checkIsExpanded() {
        if (!actual.getObj().isPresent()) {
            failIfCustomMessageIsSet();
            throw failure(String.format("The reference %s is not expanded.", actual));
        }
    }

    public ReferenceAssert isExpanded() {
        checkIsExpanded();
        return this;
    }

    @SuppressWarnings({"rawtypes", "unchecked"})
    public ReferenceAssert references(final Referenceable counterpart) {
        if (!actual.referencesSameResource(counterpart)) {
            failIfCustomMessageIsSet();
            throw failure(String.format("%s does not reference the same as %s.", actual, counterpart));
        }
        return this;
    }
}