feat: stake state types & ext tests

programs/stake/AuthorizeChecked_test.go

@@ -0,0 +1,26 @@
+package stake
+
+import (
+	"testing"
+
+	"github.com/gagliardetto/solana-go"
+	"github.com/stretchr/testify/require"
+)
+
+func TestRoundTrip_AuthorizeChecked(t *testing.T) {
+	inst := NewAuthorizeCheckedInstructionBuilder().
+		SetStakeAccount(pubkeyOf(1)).
+		SetClockSysvar(solana.SysVarClockPubkey).
+		SetCurrentAuthority(pubkeyOf(2)).
+		SetNewAuthority(pubkeyOf(3)).
+		SetStakeAuthorize(StakeAuthorizeStaker)
+
+	data, err := encodeInst(inst)
+	require.NoError(t, err)
+	require.Equal(t, u32LE(Instruction_AuthorizeChecked), data[:4])
+
+	decoded, err := DecodeInstruction(nil, data)
+	require.NoError(t, err)
+	ac := decoded.Impl.(*AuthorizeChecked)
+	require.Equal(t, StakeAuthorizeStaker, *ac.StakeAuthorize)
+}

programs/stake/Authorize_test.go

@@ -0,0 +1,28 @@
+package stake
+
+import (
+	"testing"
+
+	"github.com/gagliardetto/solana-go"
+	"github.com/stretchr/testify/require"
+)
+
+func TestRoundTrip_Authorize(t *testing.T) {
+	newAuth := pubkeyOf(5)
+	inst := NewAuthorizeInstructionBuilder().
+		SetStakeAccount(pubkeyOf(1)).
+		SetClockSysvar(solana.SysVarClockPubkey).
+		SetAuthority(pubkeyOf(2)).
+		SetNewAuthorized(newAuth).
+		SetStakeAuthorize(StakeAuthorizeWithdrawer)
+
+	data, err := encodeInst(inst)
+	require.NoError(t, err)
+	require.Equal(t, u32LE(Instruction_Authorize), data[:4])
+
+	decoded, err := DecodeInstruction(nil, data)
+	require.NoError(t, err)
+	auth := decoded.Impl.(*Authorize)
+	require.Equal(t, newAuth, *auth.NewAuthorized)
+	require.Equal(t, StakeAuthorizeWithdrawer, *auth.StakeAuthorize)
+}

programs/stake/DeactivateDelinquent_test.go

@@ -0,0 +1,15 @@
+package stake
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/require"
+)
+
+func TestRoundTrip_DeactivateDelinquent(t *testing.T) {
+	inst := NewDeactivateDelinquentInstruction(pubkeyOf(1), pubkeyOf(2), pubkeyOf(3))
+	data, err := encodeInst(inst)
+	require.NoError(t, err)
+	require.Equal(t, u32LE(Instruction_DeactivateDelinquent), data[:4])
+	require.Len(t, data, 4)
+}

programs/stake/Deactivate_test.go

@@ -0,0 +1,20 @@
+package stake
+
+import (
+	"testing"
+
+	"github.com/gagliardetto/solana-go"
+	"github.com/stretchr/testify/require"
+)
+
+func TestRoundTrip_Deactivate(t *testing.T) {
+	inst := NewDeactivateInstructionBuilder().
+		SetStakeAccount(pubkeyOf(1)).
+		SetClockSysvar(solana.SysVarClockPubkey).
+		SetStakeAuthority(pubkeyOf(2))
+
+	data, err := encodeInst(inst)
+	require.NoError(t, err)
+	require.Equal(t, u32LE(Instruction_Deactivate), data[:4])
+	require.Len(t, data, 4)
+}

programs/stake/DelegateStake_test.go

@@ -0,0 +1,27 @@
+package stake
+
+import (
+	"testing"
+
+	"github.com/gagliardetto/solana-go"
+	"github.com/stretchr/testify/require"
+)
+
+func TestRoundTrip_DelegateStake(t *testing.T) {
+	inst := NewDelegateStakeInstructionBuilder().
+		SetStakeAccount(pubkeyOf(1)).
+		SetVoteAccount(pubkeyOf(2)).
+		SetClockSysvar(solana.SysVarClockPubkey).
+		SetStakeHistorySysvar(solana.SysVarStakeHistoryPubkey).
+		SetConfigAccount(pubkeyOf(3)).
+		SetStakeAuthority(pubkeyOf(4))
+
+	data, err := encodeInst(inst)
+	require.NoError(t, err)
+	require.Equal(t, u32LE(Instruction_DelegateStake), data[:4])
+	require.Len(t, data, 4)
+
+	decoded, err := DecodeInstruction(nil, data)
+	require.NoError(t, err)
+	require.IsType(t, &DelegateStake{}, decoded.Impl)
+}

programs/stake/GetMinimumDelegation_test.go

@@ -0,0 +1,15 @@
+package stake
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/require"
+)
+
+func TestRoundTrip_GetMinimumDelegation(t *testing.T) {
+	inst := NewGetMinimumDelegationInstruction()
+	data, err := encodeInst(inst)
+	require.NoError(t, err)
+	require.Equal(t, u32LE(Instruction_GetMinimumDelegation), data[:4])
+	require.Len(t, data, 4)
+}

programs/stake/InitializeChecked_test.go

@@ -0,0 +1,21 @@
+package stake
+
+import (
+	"testing"
+
+	"github.com/gagliardetto/solana-go"
+	"github.com/stretchr/testify/require"
+)
+
+func TestRoundTrip_InitializeChecked(t *testing.T) {
+	inst := NewInitializeCheckedInstructionBuilder().
+		SetStakeAccount(pubkeyOf(1)).
+		SetRentSysvar(solana.SysVarRentPubkey).
+		SetStakeAuthority(pubkeyOf(2)).
+		SetWithdrawAuthority(pubkeyOf(3))
+
+	data, err := encodeInst(inst)
+	require.NoError(t, err)
+	require.Equal(t, u32LE(Instruction_InitializeChecked), data[:4])
+	require.Len(t, data, 4)
+}

programs/stake/Initialize_test.go

@@ -0,0 +1,39 @@
+package stake
+
+import (
+	"testing"
+
+	"github.com/gagliardetto/solana-go"
+	"github.com/stretchr/testify/require"
+)
+
+func TestRoundTrip_Initialize(t *testing.T) {
+	staker := pubkeyOf(1)
+	withdrawer := pubkeyOf(2)
+	custodian := pubkeyOf(3)
+
+	inst := NewInitializeInstructionBuilder().
+		SetStakeAccount(pubkeyOf(10)).
+		SetRentSysvarAccount(solana.SysVarRentPubkey).
+		SetStaker(staker).
+		SetWithdrawer(withdrawer).
+		SetLockupTimestamp(1000).
+		SetLockupEpoch(42).
+		SetCustodian(custodian)
+
+	data, err := encodeInst(inst)
+	require.NoError(t, err)
+
+	// First 4 bytes: instruction ID (u32 LE)
+	require.Equal(t, u32LE(Instruction_Initialize), data[:4])
+
+	// Decode back
+	decoded, err := DecodeInstruction(nil, data)
+	require.NoError(t, err)
+	init := decoded.Impl.(*Initialize)
+	require.Equal(t, staker, *init.Authorized.Staker)
+	require.Equal(t, withdrawer, *init.Authorized.Withdrawer)
+	require.Equal(t, int64(1000), *init.Lockup.UnixTimestamp)
+	require.Equal(t, uint64(42), *init.Lockup.Epoch)
+	require.Equal(t, custodian, *init.Lockup.Custodian)
+}

programs/stake/Merge_test.go

@@ -0,0 +1,22 @@
+package stake
+
+import (
+	"testing"
+
+	"github.com/gagliardetto/solana-go"
+	"github.com/stretchr/testify/require"
+)
+
+func TestRoundTrip_Merge(t *testing.T) {
+	inst := NewMergeInstructionBuilder().
+		SetDestinationStakeAccount(pubkeyOf(1)).
+		SetSourceStakeAccount(pubkeyOf(2)).
+		SetClockSysvar(solana.SysVarClockPubkey).
+		SetStakeHistorySysvar(solana.SysVarStakeHistoryPubkey).
+		SetStakeAuthority(pubkeyOf(3))
+
+	data, err := encodeInst(inst)
+	require.NoError(t, err)
+	require.Equal(t, u32LE(Instruction_Merge), data[:4])
+	require.Len(t, data, 4)
+}

programs/stake/MoveLamports_test.go

@@ -0,0 +1,19 @@
+package stake
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/require"
+)
+
+func TestRoundTrip_MoveLamports(t *testing.T) {
+	inst := NewMoveLamportsInstruction(3_000_000_000, pubkeyOf(1), pubkeyOf(2), pubkeyOf(3))
+	data, err := encodeInst(inst)
+	require.NoError(t, err)
+	require.Equal(t, u32LE(Instruction_MoveLamports), data[:4])
+
+	decoded, err := DecodeInstruction(nil, data)
+	require.NoError(t, err)
+	ml := decoded.Impl.(*MoveLamports)
+	require.Equal(t, uint64(3_000_000_000), *ml.Lamports)
+}

programs/stake/MoveStake_test.go

@@ -0,0 +1,19 @@
+package stake
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/require"
+)
+
+func TestRoundTrip_MoveStake(t *testing.T) {
+	inst := NewMoveStakeInstruction(2_000_000_000, pubkeyOf(1), pubkeyOf(2), pubkeyOf(3))
+	data, err := encodeInst(inst)
+	require.NoError(t, err)
+	require.Equal(t, u32LE(Instruction_MoveStake), data[:4])
+
+	decoded, err := DecodeInstruction(nil, data)
+	require.NoError(t, err)
+	ms := decoded.Impl.(*MoveStake)
+	require.Equal(t, uint64(2_000_000_000), *ms.Lamports)
+}

programs/stake/SetLockup_test.go

@@ -0,0 +1,30 @@
+package stake
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/require"
+)
+
+func TestRoundTrip_SetLockup(t *testing.T) {
+	ts := int64(999)
+	epoch := uint64(100)
+	custodian := pubkeyOf(7)
+	inst := NewSetLockupInstructionBuilder().
+		SetStakeAccount(pubkeyOf(1)).
+		SetAuthority(pubkeyOf(2)).
+		SetLockupTimestamp(ts).
+		SetLockupEpoch(epoch).
+		SetCustodian(custodian)
+
+	data, err := encodeInst(inst)
+	require.NoError(t, err)
+	require.Equal(t, u32LE(Instruction_SetLockup), data[:4])
+
+	decoded, err := DecodeInstruction(nil, data)
+	require.NoError(t, err)
+	sl := decoded.Impl.(*SetLockup)
+	require.Equal(t, ts, *sl.LockupArgs.UnixTimestamp)
+	require.Equal(t, epoch, *sl.LockupArgs.Epoch)
+	require.Equal(t, custodian, *sl.LockupArgs.Custodian)
+}

programs/stake/Split_test.go

@@ -0,0 +1,22 @@
+package stake
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/require"
+)
+
+func TestRoundTrip_Split(t *testing.T) {
+	inst := NewSplitInstruction(1_000_000_000, pubkeyOf(1), pubkeyOf(2), pubkeyOf(3))
+	data, err := encodeInst(inst)
+	require.NoError(t, err)
+	require.Equal(t, u32LE(Instruction_Split), data[:4])
+
+	expected := concat(u32LE(3), u64LE(1_000_000_000))
+	require.Equal(t, expected, data)
+
+	decoded, err := DecodeInstruction(nil, data)
+	require.NoError(t, err)
+	split := decoded.Impl.(*Split)
+	require.Equal(t, uint64(1_000_000_000), *split.Lamports)
+}

programs/stake/Withdraw_test.go

@@ -0,0 +1,27 @@
+package stake
+
+import (
+	"testing"
+
+	"github.com/gagliardetto/solana-go"
+	"github.com/stretchr/testify/require"
+)
+
+func TestRoundTrip_Withdraw(t *testing.T) {
+	inst := NewWithdrawInstructionBuilder().
+		SetStakeAccount(pubkeyOf(1)).
+		SetRecipientAccount(pubkeyOf(2)).
+		SetClockSysvar(solana.SysVarClockPubkey).
+		SetStakeHistorySysvar(solana.SysVarStakeHistoryPubkey).
+		SetWithdrawAuthority(pubkeyOf(3)).
+		SetLamports(500_000)
+
+	data, err := encodeInst(inst)
+	require.NoError(t, err)
+	require.Equal(t, u32LE(Instruction_Withdraw), data[:4])
+
+	decoded, err := DecodeInstruction(nil, data)
+	require.NoError(t, err)
+	w := decoded.Impl.(*Withdraw)
+	require.Equal(t, uint64(500_000), *w.Lamports)
+}

programs/stake/instructions_test.go

@@ -0,0 +1,28 @@
+package stake
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/require"
+)
+
+func TestInstructionIDValues(t *testing.T) {
+	require.Equal(t, uint32(0), Instruction_Initialize)
+	require.Equal(t, uint32(1), Instruction_Authorize)
+	require.Equal(t, uint32(2), Instruction_DelegateStake)
+	require.Equal(t, uint32(3), Instruction_Split)
+	require.Equal(t, uint32(4), Instruction_Withdraw)
+	require.Equal(t, uint32(5), Instruction_Deactivate)
+	require.Equal(t, uint32(6), Instruction_SetLockup)
+	require.Equal(t, uint32(7), Instruction_Merge)
+	require.Equal(t, uint32(8), Instruction_AuthorizeWithSeed)
+	require.Equal(t, uint32(9), Instruction_InitializeChecked)
+	require.Equal(t, uint32(10), Instruction_AuthorizeChecked)
+	require.Equal(t, uint32(11), Instruction_AuthorizeCheckedWithSeed)
+	require.Equal(t, uint32(12), Instruction_SetLockupChecked)
+	require.Equal(t, uint32(13), Instruction_GetMinimumDelegation)
+	require.Equal(t, uint32(14), Instruction_DeactivateDelinquent)
+	require.Equal(t, uint32(15), Instruction_Redelegate)
+	require.Equal(t, uint32(16), Instruction_MoveStake)
+	require.Equal(t, uint32(17), Instruction_MoveLamports)
+}