fix: update stake authorization to take validator array (#167)

docs/transactions/authz/README.md

@@ -62,7 +62,7 @@ export const createTxMsgGenericRevoke: (
 ```ts
 export interface MsgStakeAuthorizationParams {
   granteeAddress: string
-  validatorAddress: string
+  validatorAddress: string | string[]
   denom: string
   maxTokens: string | undefined
   expiration: number

packages/proto/src/messages/authz/stake.spec.ts

@@ -0,0 +1,36 @@
+import { createStakeAuthorization } from './stake'
+import { StakeAuthorization } from '../../proto/cosmos/staking/authz'
+import { denom, val, val2 } from '../../proto/tests/utils'
+import { JSONOptions } from '../../proto/tests/common'
+
+const validateStakeAuthorization = (validators: string | string[]) => {
+  const authorization = 1
+  const amount = '100000'
+
+  const msg = createStakeAuthorization(validators, denom, amount, authorization)
+
+  const expValidators = Array.isArray(validators) ? validators : [validators]
+
+  expect(msg.message.toJson(JSONOptions)).toStrictEqual({
+    allow_list: {
+      address: expValidators,
+    },
+    authorization_type: authorization,
+    max_tokens: {
+      denom,
+      amount,
+    },
+  })
+  expect(msg.path).toStrictEqual(StakeAuthorization.typeName)
+}
+
+describe('test authz staking messages', () => {
+  it('encodes single validator addresses', () => {
+    validateStakeAuthorization(val)
+  })
+
+  it('encodes multiple validator addresses', () => {
+    const validators = [val, val2]
+    validateStakeAuthorization(validators)
+  })
+})

packages/proto/src/messages/authz/stake.ts

@@ -7,15 +7,18 @@ import {
 } from '../../proto/cosmos/staking/authz.js'
 
 export function createStakeAuthorization(
-  allowAddress: string,
+  allowAddresses: string | string[],
   denom: string,
   maxTokens: string | undefined,
   authorizationType: AuthorizationType,
 ) {
+  const addresses = Array.isArray(allowAddresses)
+    ? allowAddresses
+    : [allowAddresses]
   const msg = new StakeAuthorization({
     validators: {
       value: new StakeAuthorization_Validators({
-        address: [allowAddress],
+        address: addresses,
       }),
       case: 'allowList',
     },

packages/transactions/src/messages/authz/stakingGrant.spec.ts

@@ -9,47 +9,58 @@ import TestUtils from '../../tests/utils'
 
 const { context, denom } = TestUtils
 const granteeAddress = TestUtils.addr1
-const validatorAddress = TestUtils.addrVal1
 const maxTokens = TestUtils.amount1
 const expiration = 10000
 
-const params: MsgStakeAuthorizationParams = {
+const createParams = (validatorAddress: string | string[]) => ({
   granteeAddress,
   validatorAddress,
   denom,
   maxTokens,
   expiration,
+})
+
+const validatePayload = (params: MsgStakeAuthorizationParams) => {
+  const typedData = {
+    types: {},
+    message: {},
+  }
+
+  const stakeAuthType = Proto.Cosmos.Staking.Authz.AuthorizationType.DELEGATE
+  const auth = createStakeAuthorization(
+    params.validatorAddress,
+    params.denom,
+    params.maxTokens,
+    stakeAuthType,
+  )
+
+  const messageCosmos = createMsgGrant(
+    context.sender.accountAddress,
+    params.granteeAddress,
+    auth,
+    params.expiration,
+  )
+
+  const payload = createTxMsgStakeAuthorization(context, params)
+  const expectedPayload = createTransactionPayload(
+    context,
+    typedData,
+    messageCosmos,
+  )
+
+  expect(payload).toStrictEqual(expectedPayload)
 }
 
 describe('test tx payload', () => {
-  it('produces tx payloads as expected', () => {
-    const typedData = {
-      types: {},
-      message: {},
-    }
-
-    const stakeAuthType = Proto.Cosmos.Staking.Authz.AuthorizationType.DELEGATE
-    const auth = createStakeAuthorization(
-      params.validatorAddress,
-      params.denom,
-      params.maxTokens,
-      stakeAuthType,
-    )
-
-    const messageCosmos = createMsgGrant(
-      context.sender.accountAddress,
-      params.granteeAddress,
-      auth,
-      params.expiration,
-    )
-
-    const payload = createTxMsgStakeAuthorization(context, params)
-    const expectedPayload = createTransactionPayload(
-      context,
-      typedData,
-      messageCosmos,
-    )
-
-    expect(payload).toStrictEqual(expectedPayload)
+  it('correctly produces tx payloads with single validator', () => {
+    const validator = TestUtils.addrVal1
+    const params = createParams(validator)
+    validatePayload(params)
+  })
+
+  it('correctly produces tx payloads with multiple validators', () => {
+    const validators = [TestUtils.addrVal1, TestUtils.addrVal2]
+    const params = createParams(validators)
+    validatePayload(params)
   })
 })

packages/transactions/src/messages/authz/stakingGrant.ts

@@ -3,7 +3,7 @@ import { createTransactionPayload, TxContext } from '../base.js'
 
 export interface MsgStakeAuthorizationParams {
   granteeAddress: string
-  validatorAddress: string
+  validatorAddress: string | string[]
   denom: string
   maxTokens: string | undefined
   expiration: number