@@ -1047,6 +1047,7 @@ internal class MultipartControllerTest : BaseControllerTest() {
10471047 fun testUploadPart_Ok_EtagReturned () {
10481048 val bucketMeta = bucketMetadata()
10491049 whenever(bucketService.verifyBucketExists(TEST_BUCKET_NAME )).thenReturn(bucketMeta)
1050+ whenever(multipartService.verifyPartNumberLimits(" 1" )).thenReturn(1 )
10501051 val uploadId = UUID .randomUUID()
10511052
10521053 val temp = java.nio.file.Files .createTempFile(" junie" , " part" )
@@ -1080,6 +1081,7 @@ internal class MultipartControllerTest : BaseControllerTest() {
10801081 )
10811082 val bucketMeta = bucketMetadata(versioningConfiguration = versioningConfiguration)
10821083 whenever(bucketService.verifyBucketExists(TEST_BUCKET_NAME )).thenReturn(bucketMeta)
1084+ whenever(multipartService.verifyPartNumberLimits(" 1" )).thenReturn(1 )
10831085
10841086 val s3meta = s3ObjectMetadata(
10851087 key = " source/key.txt" ,
@@ -1161,10 +1163,11 @@ internal class MultipartControllerTest : BaseControllerTest() {
11611163 fun testUploadPartCopy_InvalidPartNumber_BadRequest () {
11621164 val bucketMeta = bucketMetadata()
11631165 whenever(bucketService.verifyBucketExists(TEST_BUCKET_NAME )).thenReturn(bucketMeta)
1166+ whenever(multipartService.verifyPartNumberLimits(" 1" )).thenReturn(1 )
11641167
11651168 doThrow(S3Exception .INVALID_PART_NUMBER )
11661169 .whenever(multipartService)
1167- .verifyPartNumberLimits(1 )
1170+ .verifyPartNumberLimits(" 1 " )
11681171
11691172 val headers = HttpHeaders ().apply {
11701173 add(" x-amz-copy-source" , " /source-bucket/source/key.txt" )
@@ -1190,6 +1193,7 @@ internal class MultipartControllerTest : BaseControllerTest() {
11901193 fun testUploadPartCopy_SourceObjectNotFound () {
11911194 val bucketMeta = bucketMetadata()
11921195 whenever(bucketService.verifyBucketExists(TEST_BUCKET_NAME )).thenReturn(bucketMeta)
1196+ whenever(multipartService.verifyPartNumberLimits(" 1" )).thenReturn(1 )
11931197
11941198 doThrow(S3Exception .NO_SUCH_KEY )
11951199 .whenever(objectService)
@@ -1219,6 +1223,7 @@ internal class MultipartControllerTest : BaseControllerTest() {
12191223 fun testUploadPartCopy_PreconditionFailed () {
12201224 val bucketMeta = bucketMetadata()
12211225 whenever(bucketService.verifyBucketExists(TEST_BUCKET_NAME )).thenReturn(bucketMeta)
1226+ whenever(multipartService.verifyPartNumberLimits(" 1" )).thenReturn(1 )
12221227
12231228 val s3meta = s3ObjectMetadata(" source/key.txt" , UUID .randomUUID().toString())
12241229 whenever(objectService.verifyObjectExists(eq(" source-bucket" ), eq(" source/key.txt" ), anyOrNull()))
@@ -1260,6 +1265,7 @@ internal class MultipartControllerTest : BaseControllerTest() {
12601265 fun testUploadPartCopy_NoVersionHeaderWhenNotVersioned () {
12611266 val bucketMeta = bucketMetadata()
12621267 whenever(bucketService.verifyBucketExists(TEST_BUCKET_NAME )).thenReturn(bucketMeta)
1268+ whenever(multipartService.verifyPartNumberLimits(" 1" )).thenReturn(1 )
12631269
12641270 val s3meta = s3ObjectMetadata(
12651271 key = " source/key.txt" ,
@@ -1300,6 +1306,7 @@ internal class MultipartControllerTest : BaseControllerTest() {
13001306 fun testUploadPartCopy_EncryptionHeadersEchoed () {
13011307 val bucketMeta = bucketMetadata()
13021308 whenever(bucketService.verifyBucketExists(TEST_BUCKET_NAME )).thenReturn(bucketMeta)
1309+ whenever(multipartService.verifyPartNumberLimits(" 1" )).thenReturn(1 )
13031310
13041311 val s3meta = s3ObjectMetadata(" source/key.txt" , UUID .randomUUID().toString())
13051312 whenever(objectService.verifyObjectExists(eq(" source-bucket" ), eq(" source/key.txt" ), anyOrNull()))
@@ -1348,6 +1355,7 @@ internal class MultipartControllerTest : BaseControllerTest() {
13481355 fun testUploadPart_WithHeaderChecksum_VerifiedAndReturned () {
13491356 val bucketMeta = bucketMetadata()
13501357 whenever(bucketService.verifyBucketExists(TEST_BUCKET_NAME )).thenReturn(bucketMeta)
1358+ whenever(multipartService.verifyPartNumberLimits(" 1" )).thenReturn(1 )
13511359 val uploadId = UUID .randomUUID()
13521360
13531361 val temp = java.nio.file.Files .createTempFile(" junie" , " part" )
@@ -1395,7 +1403,7 @@ internal class MultipartControllerTest : BaseControllerTest() {
13951403 // Simulate invalid part number
13961404 doThrow(S3Exception .INVALID_PART_NUMBER )
13971405 .whenever(multipartService)
1398- .verifyPartNumberLimits(1 )
1406+ .verifyPartNumberLimits(" 1 " )
13991407
14001408 val uri = UriComponentsBuilder
14011409 .fromUriString(" /${TEST_BUCKET_NAME } /my/key.txt" )
0 commit comments