feat(restoretest): enable --unpin-agent-cpu, --stream-to-all-replicas…

… and --table-parallel flags
scylladb · Aug 28, 2024 · c14c3dd · c14c3dd
1 parent 81febdc
commit c14c3dd
Showing 1 changed file with 16 additions and 8 deletions.
diff --git a/pkg/service/restore/service_restore_integration_test.go b/pkg/service/restore/service_restore_integration_test.go
@@ -546,10 +546,13 @@ func TestRestoreTablesSmokeIntegration(t *testing.T) {
 				Path:     testBucket,
 			},
 		},
-		Keyspace:      []string{testKeyspace},
-		BatchSize:     testBatchSize,
-		Parallel:      testParallel,
-		RestoreTables: true,
+		Keyspace:            []string{testKeyspace},
+		BatchSize:           testBatchSize,
+		Parallel:            testParallel,
+		RestoreTables:       true,
+		StreamToAllReplicas: true,
+		UnpinAgentCPU:       true,
+		TableParallel:       2,
 	}
 
 	smokeRestore(t, target, testKeyspace, testLoadCnt, testLoadSize, testUser, "{'class': 'NetworkTopologyStrategy', 'dc1': 2}")
@@ -718,6 +721,8 @@ func TestRestoreTablesResumeIntegration(t *testing.T) {
 		BatchSize:     testBatchSize,
 		Parallel:      testParallel,
 		RestoreTables: true,
+		UnpinAgentCPU: true,
+		TableParallel: 2,
 		Continue:      true,
 	}
 
@@ -1327,10 +1332,13 @@ func TestRestoreFullIntegration(t *testing.T) {
 	}
 
 	tablesTarget := Target{
-		Location:      locs,
-		BatchSize:     testBatchSize,
-		Parallel:      testParallel,
-		RestoreTables: true,
+		Location:            locs,
+		BatchSize:           testBatchSize,
+		Parallel:            testParallel,
+		RestoreTables:       true,
+		StreamToAllReplicas: true,
+		UnpinAgentCPU:       true,
+		TableParallel:       2,
 	}
 
 	restoreAllTables(t, schemaTarget, tablesTarget, testKeyspace, testLoadCnt, testLoadSize, testUser)