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 sphere-test-lib Show documentation

sphere-test-lib

There is a newer version: 1.0.0-M26

package io.sphere.sdk.test;

import io.sphere.sdk.models.Resource;
import io.sphere.sdk.models.Reference;
import io.sphere.sdk.models.Referenceable;
import org.assertj.core.api.AbstractAssert;
import org.assertj.core.api.Assertions;

import static java.lang.String.format;

public class ReferenceAssert extends AbstractAssert> {

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

    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 Resource model) {
        checkIsExpanded();
        if (!actual.getObj().equals(model)) {
            failWithMessage(format("%s does not contain an expanded %s.", actual, model));
        }
        return this;
    }

    public void checkIsExpanded() {
        if (actual.getObj() == null) {
            failWithMessage(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)) {
            failWithMessage(format("%s does not reference the same as %s.", actual, counterpart));
        }
        return this;
    }
}