package com.novoda.downloadmanager;

import java.util.List;

class Migration {

    private Batch batch;
    private final List<String> originalFileLocations;
    private final List<FileSize> fileSizes;

    Migration(Batch batch, List<String> originalFileLocations, List<FileSize> fileSizes) {
        this.batch = batch;
        this.originalFileLocations = originalFileLocations;
        this.fileSizes = fileSizes;
    }

    void add(String originalFileLocation, FileSize fileSize) {
        originalFileLocations.add(originalFileLocation);
        fileSizes.add(fileSize);
    }

    void setBatch(Batch batch) {
        this.batch = batch;
    }

    List<String> originalFileLocations() {
        return originalFileLocations;
    }

    List<FileSize> fileSizes() {
        return fileSizes;
    }

    Batch batch() {
        return batch;
    }
}

@@ -848,11 +848,999 @@
 h;%0A    %7D
+%0A%0A    @Override%0A    public boolean equals(Object o) %7B%0A        if (this == o) %7B%0A            return true;%0A        %7D%0A        if (o == null %7C%7C getClass() != o.getClass()) %7B%0A            return false;%0A        %7D%0A%0A        Migration migration = (Migration) o;%0A%0A        if (!batch.equals(migration.batch)) %7B%0A            return false;%0A        %7D%0A        if (!originalFileLocations.equals(migration.originalFileLocations)) %7B%0A            return false;%0A        %7D%0A        return fileSizes.equals(migration.fileSizes);%0A    %7D%0A%0A    @Override%0A    public int hashCode() %7B%0A        int result = batch.hashCode();%0A        result = 31 * result + originalFileLocations.hashCode();%0A        result = 31 * result + fileSizes.hashCode();%0A        return result;%0A    %7D%0A%0A    @Override%0A    public String toString() %7B%0A        return %22Migration%7B%22 +%0A                %22batch=%22 + batch +%0A                %22, originalFileLocations=%22 + originalFileLocations +%0A                %22, fileSizes=%22 + fileSizes +%0A                '%7D';%0A    %7D
 %0A%7D%0A

Add equals and hashcode to Migration.