8ab643dc10
Co-authored-by: marbar3778 <marbar3778@yahoo.com> Co-authored-by: Marko <marko@baricevic.me> Co-authored-by: Matt Kocubinski <mkocubinski@gmail.com> Co-authored-by: Likhita Polavarapu <78951027+likhita-809@users.noreply.github.com> Co-authored-by: unknown unknown <unknown@unknown> Co-authored-by: Julien Robert <julien@rbrt.fr>
137 lines
3.1 KiB
Go
137 lines
3.1 KiB
Go
package serverv2_test
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
"time"
|
|
|
|
gogoproto "github.com/cosmos/gogoproto/proto"
|
|
"github.com/spf13/viper"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
coreapp "cosmossdk.io/core/app"
|
|
"cosmossdk.io/core/transaction"
|
|
"cosmossdk.io/log"
|
|
serverv2 "cosmossdk.io/server/v2"
|
|
grpc "cosmossdk.io/server/v2/api/grpc"
|
|
)
|
|
|
|
type mockInterfaceRegistry struct{}
|
|
|
|
func (*mockInterfaceRegistry) Resolve(typeUrl string) (gogoproto.Message, error) {
|
|
panic("not implemented")
|
|
}
|
|
|
|
func (*mockInterfaceRegistry) ListImplementations(ifaceTypeURL string) []string {
|
|
panic("not implemented")
|
|
}
|
|
func (*mockInterfaceRegistry) ListAllInterfaces() []string { panic("not implemented") }
|
|
|
|
type mockApp[T transaction.Tx] struct {
|
|
serverv2.AppI[T]
|
|
}
|
|
|
|
func (*mockApp[T]) InterfaceRegistry() coreapp.InterfaceRegistry {
|
|
return &mockInterfaceRegistry{}
|
|
}
|
|
|
|
// TODO split this test into multiple tests
|
|
// test read config
|
|
// test write config
|
|
// test server configs
|
|
// test start empty
|
|
// test start config exists
|
|
// test stop
|
|
func TestServer(t *testing.T) {
|
|
currentDir, err := os.Getwd()
|
|
if err != nil {
|
|
t.Log(err)
|
|
t.Fail()
|
|
}
|
|
configPath := filepath.Join(currentDir, "testdata")
|
|
|
|
v, err := serverv2.ReadConfig(configPath)
|
|
if err != nil {
|
|
v = viper.New()
|
|
}
|
|
|
|
logger := log.NewLogger(os.Stdout)
|
|
grpcServer, err := grpc.New().Init(&mockApp[transaction.Tx]{}, v, logger)
|
|
if err != nil {
|
|
t.Log(err)
|
|
t.Fail()
|
|
}
|
|
|
|
mockServer := &mockServer{name: "mock-server-1", ch: make(chan string, 100)}
|
|
|
|
server := serverv2.NewServer(
|
|
logger,
|
|
grpcServer,
|
|
mockServer,
|
|
)
|
|
|
|
serverCfgs := server.Configs()
|
|
if serverCfgs[grpcServer.Name()].(*grpc.Config).Address != grpc.DefaultConfig().Address {
|
|
t.Logf("config is not equal: %v", serverCfgs[grpcServer.Name()])
|
|
t.Fail()
|
|
}
|
|
if serverCfgs[mockServer.Name()].(*mockServerConfig).MockFieldOne != MockServerDefaultConfig().MockFieldOne {
|
|
t.Logf("config is not equal: %v", serverCfgs[mockServer.Name()])
|
|
t.Fail()
|
|
}
|
|
|
|
// write config
|
|
if err := server.WriteConfig(configPath); err != nil {
|
|
t.Log(err)
|
|
t.Fail()
|
|
}
|
|
|
|
v, err = serverv2.ReadConfig(configPath)
|
|
if err != nil {
|
|
t.Log(err) // config should be created by WriteConfig
|
|
t.FailNow()
|
|
}
|
|
if v.GetString(grpcServer.Name()+".address") != grpc.DefaultConfig().Address {
|
|
t.Logf("config is not equal: %v", v)
|
|
t.Fail()
|
|
}
|
|
|
|
// start empty
|
|
ctx := context.Background()
|
|
ctx = context.WithValue(ctx, serverv2.ServerContextKey, serverv2.Config{StartBlock: true})
|
|
ctx, cancelFn := context.WithCancel(ctx)
|
|
go func() {
|
|
// wait 5sec and cancel context
|
|
<-time.After(5 * time.Second)
|
|
cancelFn()
|
|
|
|
if err := server.Stop(ctx); err != nil {
|
|
t.Logf("failed to stop servers: %s", err)
|
|
t.Fail()
|
|
}
|
|
}()
|
|
|
|
if err := server.Start(ctx); err != nil {
|
|
t.Log(err)
|
|
t.Fail()
|
|
}
|
|
}
|
|
|
|
func TestReadConfig(t *testing.T) {
|
|
currentDir, err := os.Getwd()
|
|
if err != nil {
|
|
t.Log(err)
|
|
t.Fail()
|
|
}
|
|
configPath := filepath.Join(currentDir, "testdata")
|
|
|
|
v, err := serverv2.ReadConfig(configPath)
|
|
require.NoError(t, err)
|
|
|
|
grpcConfig := grpc.DefaultConfig()
|
|
err = v.Sub("grpc-server").Unmarshal(&grpcConfig)
|
|
require.NoError(t, err)
|
|
}
|