TaskStoreTest.java

/*******************************************************************************
 * Copyright (c) 2011 IBM Corporation and others.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 * 
 * Contributors:
 *     IBM Corporation - initial API and implementation
 *******************************************************************************/
package org.eclipse.orion.server.tests.tasks;

import java.io.File;
import java.io.IOException;
import java.util.HashSet;
import java.util.Set;

import junit.framework.TestCase;

import org.eclipse.core.runtime.IProgressMonitor;
import org.eclipse.core.runtime.IStatus;
import org.eclipse.core.runtime.Path;
import org.eclipse.core.runtime.Status;
import org.eclipse.core.runtime.jobs.Job;
import org.eclipse.orion.internal.server.core.tasks.TaskDescription;
import org.eclipse.orion.internal.server.core.tasks.TaskService;
import org.eclipse.orion.internal.server.core.tasks.TaskStore;
import org.eclipse.orion.server.core.tasks.CorruptedTaskException;
import org.eclipse.orion.server.core.tasks.ITaskService;
import org.eclipse.orion.server.core.tasks.TaskInfo;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;

/**
 * Tests for {@link TaskStore}.
 */
public class TaskStoreTest extends TestCase {
	File tempDir;

	@Test
	public void testRead() {
		TaskStore store = new TaskStore(tempDir);
		String task = store.readTask(new TaskDescription("Userdoesnotexist", "Doesnotexist", true));
		assertNull(task);
	}

	@Test
	public void testRoundTrip() throws CorruptedTaskException {
		TaskInfo task = AllTaskTests.createTestTask("test");
		TaskStore store = new TaskStore(tempDir);
		store.writeTask(new TaskDescription(task.getUserId(), task.getId(), true), task.toJSON().toString());

		TaskInfo task2 = TaskInfo.fromJSON(new TaskDescription(task.getUserId(), task.getId(), true), store.readTask(new TaskDescription(task.getUserId(), task.getId(), true)));
		AllTaskTests.assertEqualTasks(task, task2);
	}

	@Test
	public void testDeleteTask() {
		TaskInfo task = AllTaskTests.createTestTask("test");
		task.done(Status.OK_STATUS);
		TaskStore store = new TaskStore(tempDir);
		store.writeTask(new TaskDescription(task.getUserId(), task.getId(), true), task.toJSON().toString());
		assertNotNull(store.readTask(new TaskDescription(task.getUserId(), task.getId(), true)));
		assertTrue(store.removeTask(new TaskDescription(task.getUserId(), task.getId(), true)));
		assertNull(store.readTask(new TaskDescription(task.getUserId(), task.getId(), true)));
	}

	@Test
	public void readAllTasksTest() {
		TaskInfo task1 = new TaskInfo("test", "taskid1", true);
		task1.done(Status.OK_STATUS);
		TaskInfo task2 = new TaskInfo("test", "taskid2", true);
		task2.done(Status.OK_STATUS);
		TaskStore store = new TaskStore(tempDir);
		store.writeTask(new TaskDescription("test", task1.getId(), true), task1.toJSON().toString());
		assertEquals(1, store.readAllTasks("test"));
		store.writeTask(new TaskDescription("test", task2.getId(), true), task2.toJSON().toString());
		assertEquals(2, store.readAllTasks("test"));
		store.removeTask(new TaskDescription("test", task1.getId(), true));
		assertEquals(1, store.readAllTasks("test"));
	}

	@Before
	public void setUp() throws IOException {
		tempDir = new File(new File(System.getProperty("java.io.tmpdir")), "eclipse.TaskStoreTest");
		tearDown();
		tempDir.mkdir();
	}

	@After
	public void tearDown() {
		File[] children = tempDir.listFiles();
		if (children != null) {
			for (File child : children) {
				if (child.isDirectory()) {
					File[] directoryChildren = child.listFiles();
					for (File grandchild : directoryChildren) {
						grandchild.delete();
					}
				}
				child.delete();
			}
		}
		tempDir.delete();
	}

	private class TaskStoreTestJob extends Job {

		private ITaskService taskService;
		private TaskInfo[] taskInfoTable;
		private int jobNum;

		public TaskStoreTestJob(ITaskService taskService, TaskInfo[] taskInfoTable, int jobNum) {
			super("Test task job number " + jobNum);
			this.taskService = taskService;
			this.taskInfoTable = taskInfoTable;
			this.jobNum = jobNum;
		}

		@Override
		protected IStatus run(IProgressMonitor monitor) {
			taskInfoTable[jobNum] = taskService.createTask("test", true);
			return Status.OK_STATUS;
		}
	}

	@Test
	public void testUniqueTaskInfo() throws InterruptedException { //see Bug 370729
		final ITaskService taskService = new TaskService(new Path(tempDir.getAbsolutePath()));
		int numberOfChecks = 100;
		Job[] jobs = new Job[numberOfChecks];
		final TaskInfo[] taskInfos = new TaskInfo[numberOfChecks];
		for (int i = 0; i < numberOfChecks; i++) {
			jobs[i] = new TaskStoreTestJob(taskService, taskInfos, i);
		}
		for (int i = 0; i < numberOfChecks; i++) {
			jobs[i].schedule(); //start jobs fast, so they run parallelly
		}
		for (int i = 0; i < numberOfChecks; i++) {
			jobs[i].join(); //wait for all jobs to finish
		}
		Set<String> values = new HashSet<String>();
		for (int i = 0; i < numberOfChecks; i++) { //check if task ids are unique 
			String taskInfoId = taskInfos[i].getId();
			assertFalse("Bad value number " + i + " value: " + taskInfoId, values.contains(taskInfoId));
			values.add(taskInfoId);
		}
	}
}