diff --git a/src/main/java/com/omertron/themoviedbapi/model2/artwork/ArtworkMedia.java b/src/main/java/com/omertron/themoviedbapi/model2/artwork/ArtworkMedia.java index 5905ea6f9..75ac96bd6 100644 --- a/src/main/java/com/omertron/themoviedbapi/model2/artwork/ArtworkMedia.java +++ b/src/main/java/com/omertron/themoviedbapi/model2/artwork/ArtworkMedia.java @@ -28,6 +28,8 @@ import com.omertron.themoviedbapi.model2.MediaBasic; import com.omertron.themoviedbapi.model2.movie.MovieBasic; import com.omertron.themoviedbapi.model2.tv.TVBasic; import com.omertron.themoviedbapi.model2.tv.TVEpisodeBasic; +import org.apache.commons.lang3.builder.EqualsBuilder; +import org.apache.commons.lang3.builder.HashCodeBuilder; /** * @@ -71,4 +73,25 @@ public class ArtworkMedia extends Artwork { this.media = media; } + @Override + public int hashCode() { + return new HashCodeBuilder() + .appendSuper(super.hashCode()) + .append(mediaType) + .toHashCode(); + } + + @Override + public boolean equals(Object obj) { + if (obj instanceof ArtworkMedia) { + final ArtworkMedia other = (ArtworkMedia) obj; + return new EqualsBuilder() + .appendSuper(super.equals(obj)) + .append(mediaType, other.mediaType) + .isEquals(); + } else { + return false; + } + } + }