103103import static org .mockito .Matchers .eq ;
104104import static org .mockito .Mockito .mock ;
105105import static org .mockito .Mockito .reset ;
106+ import static org .mockito .Mockito .spy ;
106107import static org .mockito .Mockito .times ;
107108import static org .mockito .Mockito .verify ;
108109
@@ -1190,7 +1191,9 @@ public void testDownloadById() throws B2Exception {
11901191 final B2DownloadByIdRequest request = B2DownloadByIdRequest
11911192 .builder (fileId (1 ))
11921193 .build ();
1193- webifier .downloadById (ACCOUNT_AUTH , request , noopContentHandler );
1194+ final B2StorageClientWebifierImpl webifierSpy = spy (webifier );
1195+ webifierSpy .downloadById (ACCOUNT_AUTH , request , noopContentHandler );
1196+ verify (webifierSpy ).getDownloadByIdUrl (ACCOUNT_AUTH , request );
11941197
11951198 webApiClient .check ("getContent.\n " +
11961199 "url:\n " +
@@ -1201,7 +1204,7 @@ public void testDownloadById() throws B2Exception {
12011204 " X-Bz-Test-Mode: force_cap_exceeded\n "
12021205 );
12031206
1204- assertEquals (expectedUrl , webifier .getDownloadByIdUrl (ACCOUNT_AUTH , request ));
1207+ assertEquals (expectedUrl , webifierSpy .getDownloadByIdUrl (ACCOUNT_AUTH , request ));
12051208
12061209 checkRequestCategory (OTHER , w -> w .downloadById (ACCOUNT_AUTH , request , noopContentHandler ));
12071210 }
@@ -1303,7 +1306,9 @@ public void testDownloadByName() throws B2Exception {
13031306 final B2DownloadByNameRequest request = B2DownloadByNameRequest
13041307 .builder (bucketName (1 ), fileName (1 ))
13051308 .build ();
1306- webifier .downloadByName (ACCOUNT_AUTH , request , noopContentHandler );
1309+ final B2StorageClientWebifierImpl webifierSpy = spy (webifier );
1310+ webifierSpy .downloadByName (ACCOUNT_AUTH , request , noopContentHandler );
1311+ verify (webifierSpy ).getDownloadByNameUrl (ACCOUNT_AUTH , request );
13071312
13081313 webApiClient .check ("getContent.\n " +
13091314 "url:\n " +
@@ -1313,7 +1318,7 @@ public void testDownloadByName() throws B2Exception {
13131318 " User-Agent: SecretAgentMan/3.19.28\n " +
13141319 " X-Bz-Test-Mode: force_cap_exceeded\n "
13151320 );
1316- assertEquals (expectedUrl , webifier .getDownloadByNameUrl (ACCOUNT_AUTH , request ));
1321+ assertEquals (expectedUrl , webifierSpy .getDownloadByNameUrl (ACCOUNT_AUTH , request ));
13171322
13181323 checkRequestCategory (OTHER , w -> w .downloadByName (ACCOUNT_AUTH , request , noopContentHandler ));
13191324 }
0 commit comments