test: setting a contract as the SIP-031 recipient

Author: obycode

contrib/core-contract-tests/contracts/sip-031-indirect.clar

@@ -7,6 +7,22 @@
   )
 )
 
+;; WARNING: This is for testing purposes only.
+;; This is not a safe way to call `update-recipient` from an external contract,
+;; as it does not perform the necessary authorization checks.
+(define-public (update-recipient-as-contract (new-recipient principal))
+  (as-contract (contract-call? 'ST1PQHQKV0RJXZFY1DGX8MNSNYVE3VGZJSRTPGZGM.sip-031
+    update-recipient new-recipient
+  ))
+)
+
+;; WARNING: This is for testing purposes only.
+;; This is not a safe way to call `claim` from an external contract,
+;; as it does not perform the necessary authorization checks.
+(define-public (claim-as-contract)
+  (as-contract (contract-call? 'ST1PQHQKV0RJXZFY1DGX8MNSNYVE3VGZJSRTPGZGM.sip-031 claim))
+)
+
 ;; Helper function to transfer STX within tests
 (define-public (transfer-stx
     (amount uint)

contrib/core-contract-tests/tests/clarigen-types.ts

@@ -4687,6 +4687,12 @@ export const contracts = {
   },
   sip031Indirect: {
     functions: {
+      claimAsContract: {
+        name: 'claim-as-contract',
+        access: 'public',
+        args: [],
+        outputs: { type: { response: { ok: 'uint128', error: 'uint128' } } },
+      } as TypedAbiFunction<[], Response<bigint, bigint>>,
       transferStx: {
         name: 'transfer-stx',
         access: 'public',
@@ -4711,6 +4717,15 @@ export const contracts = {
         [newRecipient: TypedAbiArg<string, 'newRecipient'>],
         Response<boolean, bigint>
       >,
+      updateRecipientAsContract: {
+        name: 'update-recipient-as-contract',
+        access: 'public',
+        args: [{ name: 'new-recipient', type: 'principal' }],
+        outputs: { type: { response: { ok: 'bool', error: 'uint128' } } },
+      } as TypedAbiFunction<
+        [newRecipient: TypedAbiArg<string, 'newRecipient'>],
+        Response<boolean, bigint>
+      >,
       getBalance: {
         name: 'get-balance',
         access: 'read_only',

contrib/core-contract-tests/tests/sip-031/sip-031.test.ts

@@ -671,6 +671,31 @@ test('recipient change during vesting period preserves vested amounts', () => {
   expect(secondClaim.value).toBe(expectedVested);
 });
 
+test('recipient can be updated from an indirect contract set as recipient', () => {
+  txOk(
+    contract.updateRecipient(indirectContract.identifier),
+    accounts.deployer.address,
+  );
+  txOk(
+    indirectContract.updateRecipientAsContract(accounts.wallet_1.address),
+    accounts.wallet_1.address,
+  );
+  expect(rov(contract.getRecipient())).toBe(accounts.wallet_1.address);
+});
+
+test('claim can be called from an indirect contract set as recipient', () => {
+  mintInitial();
+  txOk(
+    contract.updateRecipient(indirectContract.identifier),
+    accounts.deployer.address,
+  );
+  let receipt = txOk(
+    indirectContract.claimAsContract(),
+    accounts.wallet_1.address,
+  );
+  expect(receipt.value).toBe(constants.INITIAL_MINT_IMMEDIATE_AMOUNT);
+});
+
 // -----------------------------------------------------------------------------
 // Fuzz Testing - Randomized Tests
 // -----------------------------------------------------------------------------