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 package org.eclipse.jgit.lfs.server.fs;
44
45 import static org.apache.http.HttpStatus.SC_NOT_FOUND;
46 import static org.apache.http.HttpStatus.SC_UNPROCESSABLE_ENTITY;
47 import static org.junit.Assert.assertEquals;
48 import static org.junit.Assert.fail;
49
50 import java.io.IOException;
51 import java.nio.file.Path;
52 import java.nio.file.Paths;
53 import java.text.MessageFormat;
54
55 import org.apache.http.client.ClientProtocolException;
56 import org.eclipse.jgit.lfs.lib.AnyLongObjectId;
57 import org.eclipse.jgit.lfs.test.LongObjectIdTestUtils;
58 import org.eclipse.jgit.util.FileUtils;
59 import org.junit.Test;
60
61 public class DownloadTest extends LfsServerTest {
62
63 @Test
64 public void testDownload() throws Exception {
65 String TEXT = "test";
66 AnyLongObjectId id = putContent(TEXT);
67 Path f = Paths.get(getTempDirectory().toString(), "download");
68 long len = getContent(id, f);
69 assertEquals(TEXT.length(), len);
70 FileUtils.delete(f.toFile(), FileUtils.RETRY);
71 }
72
73 @Test
74 public void testDownloadInvalidPathInfo()
75 throws ClientProtocolException, IOException {
76 String TEXT = "test";
77 String id = putContent(TEXT).name().substring(0, 60);
78 Path f = Paths.get(getTempDirectory().toString(), "download");
79 try {
80 getContent(id, f);
81 fail("expected RuntimeException");
82 } catch (RuntimeException e) {
83 String error = String.format(
84 "Invalid pathInfo '/%s' does not match '/{SHA-256}'", id);
85 assertEquals(formatErrorMessage(SC_UNPROCESSABLE_ENTITY, error),
86 e.getMessage());
87 }
88 }
89
90 @Test
91 public void testDownloadInvalidId()
92 throws ClientProtocolException, IOException {
93 String TEXT = "test";
94 String id = putContent(TEXT).name().replace('f', 'z');
95 Path f = Paths.get(getTempDirectory().toString(), "download");
96 try {
97 getContent(id, f);
98 fail("expected RuntimeException");
99 } catch (RuntimeException e) {
100 String error = String.format("Invalid id: : %s", id);
101 assertEquals(formatErrorMessage(SC_UNPROCESSABLE_ENTITY, error),
102 e.getMessage());
103 }
104 }
105
106 @Test
107 public void testDownloadNotFound()
108 throws ClientProtocolException, IOException {
109 String TEXT = "test";
110 AnyLongObjectId id = LongObjectIdTestUtils.hash(TEXT);
111 Path f = Paths.get(getTempDirectory().toString(), "download");
112 try {
113 getContent(id, f);
114 fail("expected RuntimeException");
115 } catch (RuntimeException e) {
116 String error = String.format("Object '%s' not found", id.getName());
117 assertEquals(formatErrorMessage(SC_NOT_FOUND, error),
118 e.getMessage());
119 }
120 }
121
122 @SuppressWarnings("boxing")
123 @Test
124 public void testLargeFileDownload() throws Exception {
125 Path f = Paths.get(getTempDirectory().toString(), "largeRandomFile");
126 long expectedLen = createPseudoRandomContentFile(f, 5 * MiB);
127 AnyLongObjectId id = putContent(f);
128 Path f2 = Paths.get(getTempDirectory().toString(), "download");
129 long start = System.nanoTime();
130 long len = getContent(id, f2);
131 System.out.println(
132 MessageFormat.format("dowloaded 10 MiB random data in {0}ms",
133 (System.nanoTime() - start) / 1e6));
134 assertEquals(expectedLen, len);
135 FileUtils.delete(f.toFile(), FileUtils.RETRY);
136
137 }
138
139 @SuppressWarnings("boxing")
140 private String formatErrorMessage(int status, String message) {
141 return String.format("Status: %d {\n \"message\": \"%s\"\n}", status,
142 message);
143 }
144 }