shame: forgot x/auth/context_test.go

x/auth/context_test.go

@@ -0,0 +1,39 @@
+package auth
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+
+	abci "github.com/tendermint/abci/types"
+
+	sdk "github.com/cosmos/cosmos-sdk/types"
+)
+
+func TestContextWithSigners(t *testing.T) {
+	ms, _ := setupMultiStore()
+	ctx := sdk.NewContext(ms, abci.Header{ChainID: "mychainid"}, false, nil)
+
+	_, _, addr1 := keyPubAddr()
+	_, _, addr2 := keyPubAddr()
+	acc1 := NewBaseAccountWithAddress(addr1)
+	acc1.SetSequence(7132)
+	acc2 := NewBaseAccountWithAddress(addr2)
+	acc2.SetSequence(8821)
+
+	// new ctx has no signers
+	signers := GetSigners(ctx)
+	assert.Equal(t, 0, len(signers))
+
+	ctx2 := WithSigners(ctx, []sdk.Account{&acc1, &acc2})
+
+	// original context is unchanged
+	signers = GetSigners(ctx)
+	assert.Equal(t, 0, len(signers))
+
+	// new context has signers
+	signers = GetSigners(ctx2)
+	assert.Equal(t, 2, len(signers))
+	assert.Equal(t, acc1, *(signers[0].(*BaseAccount)))
+	assert.Equal(t, acc2, *(signers[1].(*BaseAccount)))
+}