Skip to content

Commit a776cc4

Browse files
test: fix tus related assembly test
1 parent ddaaa96 commit a776cc4

File tree

2 files changed

+27
-2
lines changed

2 files changed

+27
-2
lines changed

src/test/java/com/transloadit/sdk/AssemblyTest.java

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -62,13 +62,18 @@ public void save() throws Exception {
6262
assertEquals(savedAssembly.json().get("ok"), "ASSEMBLY_COMPLETED");
6363

6464
mockServerClient.reset();
65-
assembly.removeFile("file_name");
65+
}
66+
67+
@Test
68+
public void saveWithTus() throws Exception {
69+
MockTusAssembly assembly = new MockTusAssembly(transloadit);
70+
assembly.addFile(new File("LICENSE"), "file_name");
6671

6772
mockServerClient.when(HttpRequest.request()
6873
.withPath("/assemblies")
6974
.withMethod("POST")
7075
.withBody(regex("[\\w\\W]*tus_num_expected_upload_files\"\\r\\nContent-Length: 1" +
71-
"\\r\\n\\r\\n0[\\w\\W]*")))
76+
"\\r\\n\\r\\n1[\\w\\W]*")))
7277
.respond(HttpResponse.response().withBody(getJson("resumable_assembly.json")));
7378

7479
AssemblyResponse resumableAssembly = assembly.save(true);
Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,20 @@
1+
package com.transloadit.sdk;
2+
3+
import io.tus.java.client.ProtocolException;
4+
5+
import java.io.File;
6+
import java.io.IOException;
7+
8+
/**
9+
* This class serves as a Mock to {@link Assembly}, which can be used to test
10+
* Assemblies with resumability enabled.
11+
*/
12+
public class MockTusAssembly extends Assembly {
13+
public MockTusAssembly(Transloadit transloadit) {
14+
super(transloadit);
15+
}
16+
17+
protected void processTusFile(File file, String fieldName, String assemblyUrl) throws IOException, ProtocolException {
18+
// do nothing
19+
}
20+
}

0 commit comments

Comments
 (0)