mirror of
https://github.com/filecoin-project/lotus.git
synced 2025-05-18 15:48:54 +08:00
265710874b
* fix(api): make stick sessions actually work and make them non-racy We apparently have a way to specify that all "related" requests should go to the same node. However: 1. It didn't work at all. All future requests would go to the first successful node from the first request. Because that's how stack variables work. 2. It was racy if the context was re-used concurrently. But only the first time, see point 1. * test(api): test the API merge proxy 1. Test whether or not it works. 2. Test stickiness. * fix(api): update OnSingleNode documentation
72 lines
1.7 KiB
Go
72 lines
1.7 KiB
Go
package itests
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/filecoin-project/go-state-types/big"
|
|
|
|
"github.com/filecoin-project/lotus/chain/types"
|
|
cliutil "github.com/filecoin-project/lotus/cli/util"
|
|
"github.com/filecoin-project/lotus/itests/kit"
|
|
)
|
|
|
|
func TestAPIMergeProxy(t *testing.T) {
|
|
ctx := context.Background()
|
|
|
|
// The default is too high for many nodes.
|
|
initialBalance := types.MustParseFIL("100000FIL")
|
|
|
|
nopts := []kit.NodeOpt{
|
|
kit.ThroughRPC(),
|
|
kit.WithAllSubsystems(),
|
|
kit.OwnerBalance(big.Int(initialBalance)),
|
|
}
|
|
ens := kit.NewEnsemble(t, kit.MockProofs())
|
|
nodes := make([]*kit.TestFullNode, 10)
|
|
for i := range nodes {
|
|
var nd kit.TestFullNode
|
|
ens.FullNode(&nd, nopts...)
|
|
nodes[i] = &nd
|
|
}
|
|
merged := kit.MergeFullNodes(nodes)
|
|
|
|
var miner kit.TestMiner
|
|
ens.Miner(&miner, merged, nopts...)
|
|
|
|
ens.Start()
|
|
|
|
nd1ID, err := nodes[0].ID(ctx)
|
|
require.NoError(t, err)
|
|
nd2ID, err := nodes[1].ID(ctx)
|
|
require.NoError(t, err)
|
|
|
|
// Expect to start on node 1, and switch to node 2 on failure.
|
|
mergedID, err := merged.ID(ctx)
|
|
require.NoError(t, err)
|
|
require.Equal(t, nd1ID, mergedID)
|
|
require.NoError(t, nodes[0].Stop(ctx))
|
|
mergedID, err = merged.ID(ctx)
|
|
require.NoError(t, err)
|
|
require.Equal(t, nd2ID, mergedID)
|
|
|
|
// Now see if sticky sessions work
|
|
stickyCtx := cliutil.OnSingleNode(ctx)
|
|
for i, nd := range nodes[1:] {
|
|
// kill off the previous node.
|
|
require.NoError(t, nodes[i].Stop(ctx))
|
|
|
|
got, err := merged.ID(stickyCtx)
|
|
require.NoError(t, err)
|
|
expected, err := nd.ID(ctx)
|
|
require.NoError(t, err)
|
|
require.Equal(t, expected, got)
|
|
}
|
|
|
|
// This should fail because we'll run out of retries because it's _not_ sticky!
|
|
_, err = merged.ID(ctx)
|
|
require.Error(t, err)
|
|
}
|