/*******************************************************************************
* Copyright (c) 2004, 2010 Tasktop Technologies 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:
* Tasktop Technologies - initial API and implementation
*******************************************************************************/
package org.eclipse.mylyn.tasks.tests.connector;
import java.io.ByteArrayInputStream;
import java.io.InputStream;
import org.eclipse.core.runtime.CoreException;
import org.eclipse.core.runtime.IProgressMonitor;
import org.eclipse.mylyn.tasks.core.ITask;
import org.eclipse.mylyn.tasks.core.TaskRepository;
import org.eclipse.mylyn.tasks.core.data.AbstractTaskAttachmentHandler;
import org.eclipse.mylyn.tasks.core.data.AbstractTaskAttachmentSource;
import org.eclipse.mylyn.tasks.core.data.TaskAttribute;
/**
* @author Steffen Pingel
*/
public class MockAttachmentHandler extends AbstractTaskAttachmentHandler {
private CoreException exception;
private byte[] data;
public void setException(CoreException exception) {
this.exception = exception;
}
public void setAttachmentData(byte[] data) {
this.data = data;
}
@Override
public boolean canGetContent(TaskRepository repository, ITask task) {
return true;
}
@Override
public boolean canPostContent(TaskRepository repository, ITask task) {
return true;
}
@Override
public InputStream getContent(TaskRepository repository, ITask task, TaskAttribute attachmentAttribute,
IProgressMonitor monitor) throws CoreException {
if (exception != null) {
throw exception;
}
return new ByteArrayInputStream(data);
}
@Override
public void postContent(TaskRepository repository, ITask task, AbstractTaskAttachmentSource source, String comment,
TaskAttribute attachmentAttribute, IProgressMonitor monitor) throws CoreException {
}
}