39 lines
1.1 KiB
Go
Executable File
39 lines
1.1 KiB
Go
Executable File
package rpcclient_test
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"git.marketally.com/tutus-one/tutus-chain/internal/testcli"
|
|
"git.marketally.com/tutus-one/tutus-chain/pkg/config"
|
|
"git.marketally.com/tutus-one/tutus-chain/pkg/tutusrpc"
|
|
"git.marketally.com/tutus-one/tutus-chain/pkg/rpcclient"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestInternalClientClose(t *testing.T) {
|
|
icl, err := rpcclient.NewInternal(context.TODO(), func(ctx context.Context, ch chan<- tutusrpc.Notification) func(*tutusrpc.Request) (*tutusrpc.Response, error) {
|
|
return nil
|
|
})
|
|
require.NoError(t, err)
|
|
icl.Close()
|
|
require.NoError(t, icl.GetError())
|
|
}
|
|
|
|
func TestInternalClientCancel(t *testing.T) {
|
|
ctx, cancel := context.WithCancel(context.TODO())
|
|
bc, rpcSrv, netSrv := testcli.NewTestChain(t, func(c *config.Config) {
|
|
c.ApplicationConfiguration.Consensus.UnlockWallet.Path = "../../cli/testdata/wallet1_solo.json"
|
|
}, true)
|
|
t.Cleanup(func() {
|
|
netSrv.Shutdown()
|
|
rpcSrv.Shutdown()
|
|
bc.Close()
|
|
})
|
|
icl, err := rpcclient.NewInternal(ctx, rpcSrv.RegisterLocal)
|
|
require.NoError(t, err)
|
|
cancel()
|
|
icl.Close()
|
|
require.NoError(t, icl.GetError())
|
|
}
|