feat: grant purchased traffic on the dedicated sequencer #21
There are no files selected for viewing
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,153 @@ | ||
| // Copyright (c) 2024 Digital Asset (Switzerland) GmbH and/or its affiliates. All rights reserved. | ||
| // SPDX-License-Identifier: Apache-2.0 | ||
|
|
||
| package org.lfdecentralizedtrust.splice.integration.tests | ||
|
|
||
| import com.digitalasset.canton.SynchronizerAlias | ||
| import com.digitalasset.canton.data.CantonTimestamp | ||
| import com.digitalasset.canton.topology.{Member, PartyId, SynchronizerId} | ||
| import org.lfdecentralizedtrust.splice.codegen.java.splice | ||
| import org.lfdecentralizedtrust.splice.codegen.java.splice.decentralizedsynchronizer.RegisteredSynchronizer | ||
| import org.lfdecentralizedtrust.splice.codegen.java.splice.round.IssuingMiningRound | ||
| import org.lfdecentralizedtrust.splice.codegen.java.splice.types.Round | ||
| import org.lfdecentralizedtrust.splice.integration.EnvironmentDefinition | ||
| import org.lfdecentralizedtrust.splice.integration.tests.SpliceTests.{ | ||
| IntegrationTest, | ||
| SpliceTestConsoleEnvironment, | ||
| } | ||
| import org.lfdecentralizedtrust.splice.util.{ | ||
| DisclosedContracts, | ||
| SynchronizerFeesTestUtil, | ||
| WalletTestUtil, | ||
| } | ||
|
|
||
| import scala.jdk.CollectionConverters.* | ||
| import scala.jdk.OptionConverters.* | ||
|
|
||
| /** Buys traffic for a registered synchronizer via `AmuletRules_BuyMemberTraffic` with the | ||
| * registration disclosed, and checks the operator grants it on that synchronizer's sequencer. | ||
| */ | ||
| class SyncOperatorTrafficIntegrationTest | ||
| extends IntegrationTest | ||
| with SynchronizerFeesTestUtil | ||
| with WalletTestUtil { | ||
|
|
||
| private val firstPurchase = 1_000_000L | ||
| private val secondPurchase = 2_000_000L | ||
|
|
||
| override def environmentDefinition: SpliceEnvironmentDefinition = | ||
| EnvironmentDefinition | ||
| .fromResources( | ||
| Seq("simple-topology-1sv.conf", "sync-operator-topology.conf"), | ||
| this.getClass.getSimpleName, | ||
| ) | ||
| .withStandardSetup | ||
|
|
||
| "sync operator" should { | ||
|
|
||
| "grant traffic purchased for its synchronizer on its own sequencer" in { implicit env => | ||
| val operatorParty = syncOperatorBackend.appState.store.key.operatorParty | ||
| val dsoParty = sv1Backend.getDsoInfo().dsoParty | ||
| val dsoRules = sv1Backend.getDsoInfo().dsoRules | ||
| // The operator is pointed at the splitwell sequencer, so that is the synchronizer whose | ||
| // traffic it grants. Alice's participant is a member of it. | ||
| val synchronizerId = aliceValidatorBackend.participantClientWithAdminToken.synchronizers | ||
| .id_of(SynchronizerAlias.tryCreate("splitwell")) | ||
| .logical | ||
| val member = aliceValidatorBackend.participantClient.id | ||
|
|
||
| val registration = clue("the DSO registers the synchronizer to this operator") { | ||
| val result = sv1Backend.participantClientWithAdminToken.ledger_api_extensions.commands | ||
| .submitWithResult( | ||
| sv1Backend.config.ledgerApiUser, | ||
| actAs = Seq(dsoParty), | ||
| readAs = Seq(dsoParty), | ||
| update = dsoRules.contractId.exerciseDsoRules_RegisterSynchronizer( | ||
| synchronizerId.toProtoPrimitive, | ||
| operatorParty.toProtoPrimitive, | ||
| ), | ||
| ) | ||
| result.exerciseResult.registeredSynchronizerCid | ||
| } | ||
|
|
||
| val aliceParty = onboardWalletUser(aliceWalletClient, aliceValidatorBackend) | ||
| aliceWalletClient.tap(walletUsdToAmulet(200.0)) | ||
|
|
||
| clue("no traffic is granted before any purchase") { | ||
| extraTrafficLimit(member) shouldBe 0L | ||
| } | ||
|
|
||
| clue("a purchase is granted on the splitwell sequencer") { | ||
| buyTraffic(aliceParty, member, synchronizerId, registration, dsoParty, firstPurchase) | ||
| eventually() { | ||
| extraTrafficLimit(member) shouldBe firstPurchase | ||
| } | ||
| } | ||
|
|
||
| clue("a second purchase raises the limit by exactly its amount") { | ||
| buyTraffic(aliceParty, member, synchronizerId, registration, dsoParty, secondPurchase) | ||
| eventually() { | ||
| extraTrafficLimit(member) shouldBe (firstPurchase + secondPurchase) | ||
| } | ||
| } | ||
| } | ||
| } | ||
|
|
||
| private def buyTraffic( | ||
| buyer: PartyId, | ||
| member: Member, | ||
| synchronizerId: SynchronizerId, | ||
| registration: RegisteredSynchronizer.ContractId, | ||
| dsoParty: PartyId, | ||
| trafficAmount: Long, | ||
| )(implicit env: SpliceTestConsoleEnvironment): Unit = { | ||
| val transferContext = | ||
| sv1ScanBackend.getTransferContextWithInstances(CantonTimestamp.now()) | ||
| val amulets = aliceWalletClient.list().amulets.map(_.contract.contractId.contractId) | ||
|
|
||
| aliceValidatorBackend.participantClientWithAdminToken.ledger_api_extensions.commands | ||
| .submitWithResult( | ||
| aliceValidatorBackend.config.ledgerApiUser, | ||
| actAs = Seq(buyer), | ||
| readAs = Seq(buyer), | ||
| update = transferContext.amuletRules.contract.contractId | ||
| .exerciseAmuletRules_BuyMemberTraffic( | ||
| amulets | ||
| .map[splice.amuletrules.TransferInput](cid => | ||
| new splice.amuletrules.transferinput.InputAmulet( | ||
| new splice.amulet.Amulet.ContractId(cid) | ||
| ) | ||
| ) | ||
| .asJava, | ||
| new splice.amuletrules.TransferContext( | ||
| transferContext.latestOpenMiningRound.contract.contractId, | ||
| Map.empty[Round, IssuingMiningRound.ContractId].asJava, | ||
| Map.empty[String, splice.amulet.ValidatorRight.ContractId].asJava, | ||
| None.toJava, | ||
| ), | ||
| buyer.toProtoPrimitive, | ||
| member.toProtoPrimitive, | ||
| synchronizerId.toProtoPrimitive, | ||
| // a registered synchronizer is pinned to migration id 0 | ||
| 0L, | ||
| trafficAmount, | ||
| Some(dsoParty.toProtoPrimitive).toJava, | ||
| Some(registration).toJava, | ||
| ), | ||
| disclosedContracts = DisclosedContracts | ||
| .forTesting( | ||
| transferContext.amuletRules, | ||
| transferContext.latestOpenMiningRound, | ||
| ) | ||
| .toLedgerApiDisclosedContracts, | ||
| ) | ||
| } | ||
|
|
||
| private def extraTrafficLimit( | ||
| member: Member | ||
| )(implicit env: SpliceTestConsoleEnvironment): Long = | ||
| syncOperatorBackend.appState.sequencerAdminConnection | ||
| .lookupSequencerTrafficControlState(member) | ||
| .futureValue | ||
| .fold(0L)(_.extraTrafficLimit.value) | ||
| } | ||
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,50 @@ | ||
| // Copyright (c) 2024 Digital Asset (Switzerland) GmbH and/or its affiliates. All rights reserved. | ||
| // SPDX-License-Identifier: Apache-2.0 | ||
|
|
||
| package org.lfdecentralizedtrust.splice.syncoperator.automation | ||
|
|
||
| import com.digitalasset.canton.config.NonNegativeFiniteDuration | ||
| import com.digitalasset.canton.topology.Member | ||
| import com.digitalasset.canton.tracing.TraceContext | ||
| import io.opentelemetry.api.trace.Tracer | ||
| import org.apache.pekko.stream.Materializer | ||
| import org.lfdecentralizedtrust.splice.automation.{ | ||
| ReconcileSequencerLimitWithMemberTrafficTriggerBase, | ||
| TriggerContext, | ||
| } | ||
| import org.lfdecentralizedtrust.splice.environment.SequencerAdminConnection | ||
| import org.lfdecentralizedtrust.splice.syncoperator.store.SyncOperatorStore | ||
|
|
||
| import scala.concurrent.{ExecutionContext, Future} | ||
|
|
||
| /** Reconciles the traffic purchased for this operator's synchronizer with its sequencer */ | ||
| class ReconcileDedicatedSequencerTrafficTrigger( | ||
| override protected val context: TriggerContext, | ||
| store: SyncOperatorStore, | ||
| sequencerConnection: SequencerAdminConnection, | ||
| trafficBalanceReconciliationDelay: NonNegativeFiniteDuration, | ||
| )(implicit | ||
| ec: ExecutionContext, | ||
| mat: Materializer, | ||
| tracer: Tracer, | ||
| ) extends ReconcileSequencerLimitWithMemberTrafficTriggerBase( | ||
| store, | ||
| trafficBalanceReconciliationDelay, | ||
| ) { | ||
|
|
||
| override protected def sequencerAdminConnection()(implicit | ||
| tc: TraceContext | ||
| ): Future[SequencerAdminConnection] = | ||
| Future.successful(sequencerConnection) | ||
|
|
||
| override protected def getTotalPurchasedMemberTraffic(memberId: Member)(implicit | ||
| tc: TraceContext | ||
| ): Future[Long] = | ||
| store.getTotalPurchasedMemberTraffic(memberId) | ||
|
|
||
| override protected def trafficLimitOffset(memberId: Member)(implicit | ||
| tc: TraceContext | ||
| ): Future[Either[String, Long]] = | ||
| // No prior consumption to carry. | ||
| Future.successful(Right(0L)) | ||
| } |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
use actAndCheck instead of clue + eventually