diff -r 000000000000 -r 6474c204b198 mobile/android/tests/background/junit3/src/sync/helpers/ExpectFetchDelegate.java --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/mobile/android/tests/background/junit3/src/sync/helpers/ExpectFetchDelegate.java Wed Dec 31 06:09:35 2014 +0100 @@ -0,0 +1,32 @@ +/* Any copyright is dedicated to the Public Domain. + http://creativecommons.org/publicdomain/zero/1.0/ */ + +package org.mozilla.gecko.background.sync.helpers; + +import java.util.HashMap; + +import org.mozilla.gecko.sync.repositories.domain.Record; + +public class ExpectFetchDelegate extends DefaultFetchDelegate { + private HashMap expect = new HashMap(); + + public ExpectFetchDelegate(Record[] records) { + for(int i = 0; i < records.length; i++) { + expect.put(records[i].guid, records[i]); + } + } + + @Override + public void onFetchedRecord(Record record) { + this.records.add(record); + } + + @Override + public void onFetchCompleted(final long fetchEnd) { + super.onDone(this.records, this.expect, fetchEnd); + } + + public Record recordAt(int i) { + return this.records.get(i); + } +}