ASSIGNED - bug 387492: [api] provide iterators and other services for

TaskSchema 
https://bugs.eclipse.org/bugs/show_bug.cgi?id=387492

Change-Id: I00a0c80eb8b03e71d71cc3d205c4608b1ebf38bb
Signed-off-by: Miles Parker <milesparker@gmail.com>

1 files changed, 38 insertions, 0 deletions

diff --git a/org.eclipse.mylyn.tasks.tests/src/org/eclipse/mylyn/tasks/tests/core/DefaultTaskSchemaTest.java b/org.eclipse.mylyn.tasks.tests/src/org/eclipse/mylyn/tasks/tests/core/DefaultTaskSchemaTest.java
index bb6942bbc..f4215afb0 100644
--- a/org.eclipse.mylyn.tasks.tests/src/org/eclipse/mylyn/tasks/tests/core/DefaultTaskSchemaTest.java
+++ b/org.eclipse.mylyn.tasks.tests/src/org/eclipse/mylyn/tasks/tests/core/DefaultTaskSchemaTest.java
@@ -11,14 +11,23 @@
 
 package org.eclipse.mylyn.tasks.tests.core;
 
+import java.util.HashSet;
+import java.util.Iterator;
+import java.util.Set;
+
 import junit.framework.TestCase;
 
+import org.eclipse.mylyn.tasks.core.TaskRepository;
 import org.eclipse.mylyn.tasks.core.data.AbstractTaskSchema;
+import org.eclipse.mylyn.tasks.core.data.AbstractTaskSchema.Field;
 import org.eclipse.mylyn.tasks.core.data.DefaultTaskSchema;
 import org.eclipse.mylyn.tasks.core.data.TaskAttribute;
+import org.eclipse.mylyn.tasks.core.data.TaskAttributeMapper;
+import org.eclipse.mylyn.tasks.core.data.TaskData;
 
 /**
  * @author Steffen Pingel
+ * @author Miles Parker
  */
 public class DefaultTaskSchemaTest extends TestCase {
 
@@ -55,4 +64,33 @@ public class DefaultTaskSchemaTest extends TestCase {
 		assertFalse(schema.REPORTER_MODIFIED.isReadOnly());
 	}
 
+	public void testIterator() {
+		AbstractTaskSchema schema = new DefaultTaskSchema();
+		Iterator<Field> fields = schema.getFields().iterator();
+		int i = 0;
+		Set<String> attributeIds = new HashSet<String>();
+		while (fields.hasNext()) {
+			Field next = fields.next();
+			attributeIds.add(next.getKey());
+			i++;
+		}
+		//Let's allow for adding fields to default schema without breaking test, but assume that no-existing attributes will be removed
+		assertTrue("Actual Attribute Count: " + i, i >= 40);
+		assertTrue(attributeIds.contains(TaskAttribute.ADD_SELF_CC));
+		assertTrue(attributeIds.contains(TaskAttribute.ATTACHMENT_AUTHOR));
+		assertTrue(attributeIds.contains(TaskAttribute.SUMMARY));
+	}
+
+	public void testInitializeTaskData() {
+		AbstractTaskSchema schema = new DefaultTaskSchema();
+		TaskData testData = new TaskData(new TaskAttributeMapper(new TaskRepository("mock", "http://mock")), "mock",
+				"http://mock", "-1");
+		schema.initialize(testData);
+		int size = testData.getRoot().getAttributes().values().size();
+		//Let's allow for adding fields to default schema without breaking test, but assume that no-existing attributes will be removed
+		assertTrue("Actual Attribute Count: " + size, size >= 40);
+		assertNotNull(testData.getRoot().getAttribute(TaskAttribute.ADD_SELF_CC));
+		assertNotNull(testData.getRoot().getAttribute(TaskAttribute.ATTACHMENT_ID));
+		assertNotNull(testData.getRoot().getAttribute(TaskAttribute.SUMMARY));
+	}
 }