fix compile

etspaceman · Nov 8, 2024 · b733517 · b733517
1 parent 686f4bb
commit b733517
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/integration-tests/src/test/scala/kinesis/mock/KCLTests.scala b/integration-tests/src/test/scala/kinesis/mock/KCLTests.scala
@@ -78,6 +78,7 @@ class KCLTests extends AwsFunctionalTests {
           )
           isStarted <- Resource.eval(Deferred[IO, Unit])
           defaultLeaseManagement = new LeaseManagementConfig(
+            appName,
             appName,
             dynamoClient,
             resources.kinesisClient,
@@ -107,11 +108,11 @@ class KCLTests extends AwsFunctionalTests {
               defaultLeaseManagement.cacheMissWarningModulus(),
               defaultLeaseManagement.initialLeaseTableReadCapacity().toLong,
               defaultLeaseManagement.initialLeaseTableWriteCapacity().toLong,
-              defaultLeaseManagement.hierarchicalShardSyncer(),
               defaultLeaseManagement.tableCreatorCallback(),
               defaultLeaseManagement.dynamoDbRequestTimeout(),
               defaultLeaseManagement.billingMode(),
               defaultLeaseManagement.leaseTableDeletionProtectionEnabled(),
+              defaultLeaseManagement.leaseTablePitrEnabled(),
               defaultLeaseManagement.tags(),
               new DynamoDBLeaseSerializer(),
               defaultLeaseManagement.customShardDetectorProvider(),
@@ -121,7 +122,9 @@ class KCLTests extends AwsFunctionalTests {
                 .completedLeaseCleanupIntervalMillis(500L)
                 .garbageLeaseCleanupIntervalMillis(500L)
                 .leaseCleanupIntervalMillis(10.seconds.toMillis)
-                .build()
+                .build(),
+              defaultLeaseManagement.workerUtilizationAwareAssignmentConfig(),
+              defaultLeaseManagement.gracefulLeaseHandoffConfig()
             )
           )
           scheduler <- Resource.eval(