From c14c3dd902d1be600b7a084189da81dead83a920 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Micha=C5=82=20Leszczy=C5=84ski?= <2000michal@wp.pl>
Date: Tue, 27 Aug 2024 20:22:05 +0200
Subject: [PATCH] feat(restoretest): enable --unpin-agent-cpu,
 --stream-to-all-replicas and --table-parallel flags

---
 .../service_restore_integration_test.go       | 24 ++++++++++++-------
 1 file changed, 16 insertions(+), 8 deletions(-)

diff --git a/pkg/service/restore/service_restore_integration_test.go b/pkg/service/restore/service_restore_integration_test.go
index 435f05d7f..08956af77 100644
--- 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)