1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
|
/*******************************************************************************
* Copyright (c) 2004, 2008 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;
import java.io.File;
import java.io.RandomAccessFile;
import junit.framework.TestCase;
import org.eclipse.core.runtime.CoreException;
import org.eclipse.core.runtime.Status;
import org.eclipse.mylyn.internal.tasks.core.TaskAttachment;
import org.eclipse.mylyn.internal.tasks.core.TaskRepositoryManager;
import org.eclipse.mylyn.internal.tasks.ui.TasksUiPlugin;
import org.eclipse.mylyn.internal.tasks.ui.util.CopyAttachmentToClipboardJob;
import org.eclipse.mylyn.internal.tasks.ui.util.DownloadAttachmentJob;
import org.eclipse.mylyn.tasks.core.ITaskAttachment;
import org.eclipse.mylyn.tasks.core.TaskRepository;
import org.eclipse.mylyn.tasks.core.data.TaskAttributeMapper;
import org.eclipse.mylyn.tasks.core.data.TaskData;
import org.eclipse.mylyn.tasks.tests.connector.MockAttachmentHandler;
import org.eclipse.mylyn.tasks.tests.connector.MockRepositoryConnector;
import org.eclipse.mylyn.tasks.tests.connector.MockTask;
import org.eclipse.swt.dnd.Clipboard;
import org.eclipse.swt.dnd.TextTransfer;
import org.eclipse.ui.PlatformUI;
/**
* Test task attachment jobs.
*
* @author Steffen Pingel
*/
public class TaskAttachmentTest extends TestCase {
private TaskRepositoryManager manager;
private MockRepositoryConnector connector;
private MockAttachmentHandler attachmentHandler;
private TaskRepository repository;
private ITaskAttachment attachment;
@Override
protected void setUp() throws Exception {
super.setUp();
manager = TasksUiPlugin.getRepositoryManager();
repository = new TaskRepository(MockRepositoryConnector.CONNECTOR_KIND, MockRepositoryConnector.REPOSITORY_URL);
manager.addRepository(repository);
attachmentHandler = new MockAttachmentHandler();
connector = new MockRepositoryConnector();
connector.setTaskAttachmentHandler(attachmentHandler);
manager.addRepositoryConnector(connector);
TaskData taskData = new TaskData(new TaskAttributeMapper(repository), MockRepositoryConnector.CONNECTOR_KIND,
MockRepositoryConnector.REPOSITORY_URL, "1");
attachment = new TaskAttachment(repository, new MockTask("1"), taskData.getRoot().createAttribute("attachment"));
}
@Override
protected void tearDown() throws Exception {
manager.removeRepository(repository);
}
public void testCopyToClipboardAction() throws Exception {
String expected = "attachment content";
attachmentHandler.setAttachmentData(expected.getBytes());
CopyAttachmentToClipboardJob job = new CopyAttachmentToClipboardJob(attachment);
job.schedule();
job.join();
Clipboard clipboard = new Clipboard(PlatformUI.getWorkbench().getDisplay());
assertEquals(expected, clipboard.getContents(TextTransfer.getInstance()));
}
public void testDownloadAttachmentJob() throws Exception {
File file = File.createTempFile("mylyn", null);
file.deleteOnExit();
String expected = "attachment\ncontent";
attachmentHandler.setAttachmentData(expected.getBytes());
DownloadAttachmentJob job = new DownloadAttachmentJob(attachment, file);
job.schedule();
job.join();
assertEquals(Status.OK_STATUS, job.getResult());
RandomAccessFile raf = new RandomAccessFile(file, "r");
byte[] data = new byte[expected.getBytes().length];
try {
raf.readFully(data);
} finally {
raf.close();
}
assertEquals(expected, new String(data));
}
public void testDownloadAttachmentJobExceptionThrown() throws Exception {
File file = File.createTempFile("mylyn", null);
file.delete();
attachmentHandler.setException(new CoreException(Status.CANCEL_STATUS));
DownloadAttachmentJob job = new DownloadAttachmentJob(attachment, file);
job.schedule();
job.join();
assertEquals(Status.OK_STATUS, job.getResult());
assertFalse(file.exists());
}
}
|