1+ package io .imagekit .sdk .tasks ;
2+
3+ import com .google .gson .Gson ;
4+ import io .imagekit .sdk .models .FileCreateRequest ;
5+ import io .imagekit .sdk .models .FileUpdateRequest ;
6+ import okhttp3 .*;
7+ import okhttp3 .mockwebserver .Dispatcher ;
8+ import okhttp3 .mockwebserver .MockResponse ;
9+ import okhttp3 .mockwebserver .MockWebServer ;
10+ import okhttp3 .mockwebserver .RecordedRequest ;
11+ import org .junit .After ;
12+ import org .junit .Before ;
13+ import org .junit .Test ;
14+ import java .util .ArrayList ;
15+ import java .util .HashMap ;
16+ import java .util .List ;
17+ import java .util .Map ;
18+ import java .util .concurrent .TimeUnit ;
19+ import static org .junit .Assert .assertEquals ;
20+
21+
22+ public class MultipartBuilderTest {
23+ MockWebServer server ;
24+ Map <String , String > headers ;
25+ OkHttpClient okHttpClient ;
26+ MultipartBuilder SUT ;
27+
28+ @ Before
29+ public void setUp () throws Exception {
30+ SUT =new MultipartBuilder ();
31+
32+ String credential = Credentials .basic ("private_key_test" ,"" );
33+ headers =new HashMap <>();
34+ headers .put ("Accept-Encoding" ,"application/json" );
35+ headers .put ("Content-Type" ,"application/json" );
36+ headers .put ("Authorization" ,credential );
37+
38+ server = new MockWebServer ();
39+ MockResponse response =new MockResponse ()
40+ .setResponseCode (200 )
41+ .
setBody (
"{isSuccessful=true, message='Your request contains invalid file.', help='For support kindly contact us at [email protected] .', fileId='null', name='null', url='null', thumbnail='null', height=0, width=0, size=0, filePath='null', tags='null', isPrivateFile=false, customCoordinates='null', fileType='null'}" );
42+
43+ server .setDispatcher (new Dispatcher () {
44+ @ Override
45+ public MockResponse dispatch (RecordedRequest request ) throws InterruptedException {
46+ return response ;
47+ }
48+ });
49+ server .start ();
50+
51+ okHttpClient = new OkHttpClient .Builder ()
52+ .connectTimeout (2 , TimeUnit .SECONDS )
53+ .readTimeout (2 , TimeUnit .SECONDS )
54+ .writeTimeout (2 , TimeUnit .SECONDS )
55+ .build ();
56+
57+ }
58+
59+ @ After
60+ public void tearDown () throws Exception {
61+ server .shutdown ();
62+ }
63+
64+ @ Test
65+ public void build_test_with_FileCreateRequest () throws Exception {
66+
67+ FileCreateRequest fileCreateRequest =new FileCreateRequest ("f06830ca9f1e3e90" ,"demo.jpg" );
68+ fileCreateRequest .setPrivateFile (true );
69+
70+ MultipartBody body = SUT .build (fileCreateRequest );
71+
72+ HttpUrl baseUrl = server .url ("/api/v1/files/upload" );
73+ Request request =new Request .Builder ()
74+ .url (baseUrl )
75+ .post (body )
76+ .headers (Headers .of (headers ))
77+ .build ();
78+
79+ Response response = okHttpClient .newCall (request ).execute ();
80+
81+ String boundary =body .boundary ();
82+
83+ String expectedMultipartData ="--" +boundary +"\r \n " +
84+ "Content-Disposition: form-data; name=\" file\" \r \n " +
85+ "Content-Length: 16\r \n \r \n " +
86+ "f06830ca9f1e3e90\r \n " +
87+ "--" +boundary +"\r \n " +
88+ "Content-Disposition: form-data; name=\" fileName\" \r \n " +
89+ "Content-Length: 8\r \n \r \n " +
90+ "demo.jpg\r \n " +
91+ "--" +boundary +"\r \n " +
92+ "Content-Disposition: form-data; name=\" useUniqueFileName\" \r \n " +
93+ "Content-Length: 4\r \n \r \n " +
94+ "true\r \n " +
95+ "--" +boundary +"\r \n " +
96+ "Content-Disposition: form-data; name=\" isPrivateFile\" \r \n " +
97+ "Content-Length: 4\r \n \r \n " +
98+ "true\r \n " +
99+ "--" +boundary +"--\r \n " ;
100+
101+ // It's capture multipart request
102+ RecordedRequest recordedRequest =server .takeRequest ();
103+
104+ String data =recordedRequest .getBody ().readUtf8 ();
105+
106+ assertEquals (expectedMultipartData , data );
107+ }
108+
109+ @ Test
110+ public void build_test_with_FileUpdateRequest () throws Exception {
111+
112+ FileUpdateRequest fileUpdateRequest =new FileUpdateRequest ("598821f949c0a938d57563bd" );
113+ List <String > tags =new ArrayList <>();
114+ tags .add ("t-shirt" );
115+ tags .add ("round-neck" );
116+ tags .add ("sale2020" );
117+ fileUpdateRequest .setTags (tags );
118+
119+ RequestBody body =SUT .build (fileUpdateRequest );
120+
121+ HttpUrl baseUrl = server .url (String .format ("/v1/files/%s/details" ,fileUpdateRequest .getFileId ()));
122+ Request request =new Request .Builder ()
123+ .url (baseUrl )
124+ .post (body )
125+ .headers (Headers .of (headers ))
126+ .build ();
127+
128+ Response response = okHttpClient .newCall (request ).execute ();
129+
130+
131+ String expectedRequestObject ="{\" fileId\" :\" 598821f949c0a938d57563bd\" ,\" tags\" :[\" t-shirt\" ,\" round-neck\" ,\" sale2020\" ]}" ;
132+
133+ // It's capture multipart request
134+ RecordedRequest recordedRequest =server .takeRequest ();
135+
136+ String data =recordedRequest .getBody ().readUtf8 ();
137+
138+ assertEquals (expectedRequestObject , data );
139+ }
140+
141+ @ Test
142+ public void build_test_with_From_Json_String () throws Exception {
143+
144+ String json ="{\" fileIds\" :[\" 598821f949c0a938d57534bd\" ,\" 338821f949c0a938d57563bz\" ,\" 987821f949c0a938d57563rt\" ]}" ;
145+
146+ RequestBody body =SUT .build (json );
147+
148+ HttpUrl baseUrl = server .url ("/v1/files/batch/deleteByFileIds" );
149+ Request request =new Request .Builder ()
150+ .url (baseUrl )
151+ .post (body )
152+ .headers (Headers .of (headers ))
153+ .build ();
154+
155+ Response response = okHttpClient .newCall (request ).execute ();
156+
157+
158+ String expectedRequestObject ="{\" fileIds\" :[\" 598821f949c0a938d57534bd\" ,\" 338821f949c0a938d57563bz\" ,\" 987821f949c0a938d57563rt\" ]}" ;
159+
160+ // It's capture multipart request
161+ RecordedRequest recordedRequest =server .takeRequest ();
162+
163+ String data =recordedRequest .getBody ().readUtf8 ();
164+
165+ assertEquals (expectedRequestObject , data );
166+ }
167+ }
0 commit comments