Move some ParallelPull integration to helper funcs

Signed-off-by: David Son <davbson@amazon.com>

Author: sondavidb

integration/pull_test.go

@@ -885,11 +885,7 @@ func TestPullWithParallelism(t *testing.T) {
 		},
 		{
 			name: "set chunk size",
-			opts: []snapshotterConfigOpt{
-				func(cfg *config.Config) {
-					cfg.PullModes.Parallel.ConcurrentDownloadChunkSize = 1_000_000
-				},
-			},
+			opts: []snapshotterConfigOpt{withConcurrentDownloadChunkSize(1_000_000)},
 		},
 	}
 
@@ -902,11 +898,7 @@ func TestPullWithParallelism(t *testing.T) {
 		},
 		{
 			name: "parallel unpacking",
-			opts: []snapshotterConfigOpt{
-				func(cfg *config.Config) {
-					cfg.PullModes.Parallel.MaxConcurrentUnpacks = 3
-				},
-			},
+			opts: []snapshotterConfigOpt{withConcurrentUnpacks(3)},
 		},
 	}
 

integration/util_test.go

@@ -377,6 +377,18 @@ func withUnboundedPullUnpack() snapshotterConfigOpt {
 	}
 }
 
+func withConcurrentDownloadChunkSize(chunkSize int64) snapshotterConfigOpt {
+	return func(c *config.Config) {
+		c.PullModes.Parallel.ConcurrentDownloadChunkSize = chunkSize
+	}
+}
+
+func withConcurrentUnpacks(n int64) snapshotterConfigOpt {
+	return func(c *config.Config) {
+		c.PullModes.Parallel.MaxConcurrentUnpacks = n
+	}
+}
+
 func withDiscardUnpackedLayers() snapshotterConfigOpt {
 	return func(c *config.Config) {
 		c.PullModes.Parallel.DiscardUnpackedLayers = true