add equals and hashcode to ArtworkMedia

2015-03-03 12:38:25 +00:00
parent d0ec5c3b18
commit aa6bb1e252
1 changed files with 23 additions and 0 deletions
@@ -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;
+        }
+    }
+
 }