Minor refactor of migration test

pull/14/head
Sam Bosley 12 years ago
parent a7bac3e142
commit 7d27cbef8d

@ -1,6 +1,6 @@
package com.todoroo.astrid.sync; package com.todoroo.astrid.sync;
import java.math.BigInteger; import java.util.ArrayList;
import android.text.TextUtils; import android.text.TextUtils;
@ -94,11 +94,11 @@ public class Astrid44MigrationTest extends NewSyncTestCase {
} }
private void assertAllModelsHaveUUID() { private void assertAllModelsHaveUUID() {
assertRemoteIds(Task.TABLE, new Task(), taskDao, Task.ID, Task.UUID); assertUUIDs(Task.TABLE, new Task(), taskDao, Task.ID, Task.UUID);
assertRemoteIds(TagData.TABLE, new TagData(), tagDataDao, TagData.ID, TagData.UUID); assertUUIDs(TagData.TABLE, new TagData(), tagDataDao, TagData.ID, TagData.UUID);
} }
private <TYPE extends RemoteModel> void assertRemoteIds(Table table, TYPE instance, RemoteModelDao<TYPE> dao, Property<?>... properties) { private <TYPE extends RemoteModel> void assertUUIDs(Table table, TYPE instance, RemoteModelDao<TYPE> dao, Property<?>... properties) {
TodorooCursor<TYPE> cursor = dao.query(Query.select(properties).from(table)); TodorooCursor<TYPE> cursor = dao.query(Query.select(properties).from(table));
try { try {
for (cursor.moveToFirst(); !cursor.isAfterLast(); cursor.moveToNext()) { for (cursor.moveToFirst(); !cursor.isAfterLast(); cursor.moveToNext()) {
@ -115,21 +115,23 @@ public class Astrid44MigrationTest extends NewSyncTestCase {
} }
private void assertAllTagsHaveTagData() { private void assertAllTagsHaveTagData() {
for (int i = 1; i <= 5; i++) { ArrayList<String> names = new ArrayList<String>();
for (int i = 1; i <= 5; i++) {
String name = "Tag " + i; String name = "Tag " + i;
assertTagForName(name); names.add(name);
} }
assertTagForName("New tag"); names.add("New tag");
assertTagForName("New tag 2"); names.add("New tag 2");
}
String[] namesArray = names.toArray(new String[names.size()]);
private void assertTagForName(String name) { TodorooCursor<TagData> tagData = tagDataDao.query(Query.select(TagData.NAME).where(TagData.NAME.in(namesArray)));
TodorooCursor<TagData> tagData = tagDataDao.query(Query.select(TagData.NAME).where(TagData.NAME.eq(name)));
try { try {
assertEquals(tagData.getCount(), 1); assertEquals(namesArray.length, tagData.getCount());
} finally { } finally {
tagData.close(); tagData.close();
} }
} }
private void assertAllMetadataHasAllFields() { private void assertAllMetadataHasAllFields() {

Loading…
Cancel
Save