Skip to content

Commit

Permalink
RPC tests for Health and Status functions
Browse files Browse the repository at this point in the history
  • Loading branch information
ramilexe committed May 1, 2024
1 parent 7b805b4 commit f8b917b
Show file tree
Hide file tree
Showing 2 changed files with 58 additions and 29 deletions.
58 changes: 58 additions & 0 deletions vm/rpc_test.go
Original file line number Diff line number Diff line change
@@ -1 +1,59 @@
package vm

import (
"context"
"net/http"
"testing"
"time"

ctypes "github.com/cometbft/cometbft/rpc/core/types"
"github.com/cometbft/cometbft/rpc/jsonrpc/client"
"github.com/stretchr/testify/require"

"github.com/consideritdone/landslidevm/jsonrpc"
)

func setupRPC(t *testing.T) (*http.Server, *LandslideVM, *client.Client) {
vm := newFreshKvApp(t)
vmLnd := vm.(*LandslideVM)
mux := http.NewServeMux()
jsonrpc.RegisterRPCFuncs(mux, NewRPC(vmLnd).Routes(), vmLnd.logger)

address := "127.0.0.1:44444"
server := &http.Server{Addr: address, Handler: mux}
go func() {
server.ListenAndServe()
//panic(err)
//require.NoError(t, err)
}()

// wait for servers to start
time.Sleep(time.Second * 2)

client, err := client.New("tcp://" + address)
require.NoError(t, err)

return server, vmLnd, client
}

func TestHealth(t *testing.T) {
server, _, client := setupRPC(t)
defer server.Close()

result := new(ctypes.ResultHealth)
_, err := client.Call(context.Background(), "health", map[string]interface{}{}, result)
require.NoError(t, err)

t.Logf("Health result %+v", result)
}

func TestStatus(t *testing.T) {
server, _, client := setupRPC(t)
defer server.Close()

result := new(ctypes.ResultStatus)
_, err := client.Call(context.Background(), "status", map[string]interface{}{}, result)
require.NoError(t, err)

t.Logf("Status result %+v", result)
}
29 changes: 0 additions & 29 deletions vm/vm.go
Original file line number Diff line number Diff line change
Expand Up @@ -410,35 +410,6 @@ func (vm *LandslideVM) CreateHandlers(context.Context, *emptypb.Empty) (*vmpb.Cr
}, nil
}

/*
func (vm *VMServer) CreateHandlers(ctx context.Context, _ *emptypb.Empty) (*vmpb.CreateHandlersResponse, error) {
handlers, err := vm.vm.CreateHandlers(ctx)
if err != nil {
return nil, err
}
resp := &vmpb.CreateHandlersResponse{}
for prefix, handler := range handlers {
serverListener, err := grpcutils.NewListener()
if err != nil {
return nil, err
}
server := grpcutils.NewServer()
vm.serverCloser.Add(server)
httppb.RegisterHTTPServer(server, ghttp.NewServer(handler))
// Start HTTP service
go grpcutils.Serve(serverListener, server)
resp.Handlers = append(resp.Handlers, &vmpb.Handler{
Prefix: prefix,
ServerAddr: serverListener.Addr().String(),
})
}
return resp, nil
}
*/

func (vm *LandslideVM) Connected(context.Context, *vmpb.ConnectedRequest) (*emptypb.Empty, error) {
vm.vmconnected.Set(true)
return &emptypb.Empty{}, nil
Expand Down

0 comments on commit f8b917b

Please sign in to comment.