client/lcd: wip use in-proc tm and lcd for tests
This commit is contained in:
parent
7f3a6e0c04
commit
d807d32f8a
|
@ -5,6 +5,7 @@ import (
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
"fmt"
|
"fmt"
|
||||||
"io/ioutil"
|
"io/ioutil"
|
||||||
|
"net"
|
||||||
"net/http"
|
"net/http"
|
||||||
"os"
|
"os"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
|
@ -12,18 +13,18 @@ import (
|
||||||
"testing"
|
"testing"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/stretchr/testify/assert"
|
"github.com/chain/core/config"
|
||||||
"github.com/stretchr/testify/require"
|
client "github.com/cosmos/cosmos-sdk/client"
|
||||||
|
|
||||||
cryptoKeys "github.com/tendermint/go-crypto/keys"
|
|
||||||
"github.com/tendermint/tendermint/p2p"
|
|
||||||
ctypes "github.com/tendermint/tendermint/rpc/core/types"
|
|
||||||
tmtypes "github.com/tendermint/tendermint/types"
|
|
||||||
|
|
||||||
"github.com/cosmos/cosmos-sdk/client"
|
|
||||||
keys "github.com/cosmos/cosmos-sdk/client/keys"
|
keys "github.com/cosmos/cosmos-sdk/client/keys"
|
||||||
"github.com/cosmos/cosmos-sdk/tests"
|
"github.com/cosmos/cosmos-sdk/tests"
|
||||||
"github.com/cosmos/cosmos-sdk/x/auth"
|
"github.com/cosmos/cosmos-sdk/wire"
|
||||||
|
auth "github.com/cosmos/cosmos-sdk/x/auth/rest"
|
||||||
|
"github.com/stretchr/testify/assert"
|
||||||
|
"github.com/stretchr/testify/require"
|
||||||
|
p2p "github.com/tendermint/go-p2p"
|
||||||
|
"github.com/tendermint/mintdb/types"
|
||||||
|
"github.com/tendermint/tendermint/proxy"
|
||||||
|
"github.com/tendermint/tmlibs/log"
|
||||||
)
|
)
|
||||||
|
|
||||||
func TestKeys(t *testing.T) {
|
func TestKeys(t *testing.T) {
|
||||||
|
@ -299,9 +300,53 @@ func setupEnvironment(t *testing.T) (kill func(), port string, seed string) {
|
||||||
cmdNode.Process.Wait()
|
cmdNode.Process.Wait()
|
||||||
os.Remove(dir)
|
os.Remove(dir)
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// strt TM and the LCD in process, listening on their respective sockets
|
||||||
|
func startTMAndLCD(t *testing.T) (kill func(), port string, seed string) {
|
||||||
|
|
||||||
|
// make the keybase and its key ...
|
||||||
|
|
||||||
|
startTM(cfg, genDoc, app)
|
||||||
|
startLCD(cdc, listenAddr, logger)
|
||||||
|
|
||||||
|
kill = func() {
|
||||||
|
// TODO: cleanup
|
||||||
|
// TODO: it would be great if TM could run without
|
||||||
|
// persiting anything in the first place
|
||||||
|
}
|
||||||
return kill, port, seed
|
return kill, port, seed
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Create & start in-process tendermint node with memdb
|
||||||
|
// and in-process abci application.
|
||||||
|
// TODO: need to clean up the WAL dir or enable it to be not persistent
|
||||||
|
func startTM(cfg *config.Config, genDoc types.GenesisDoc, app abci.Application) (*Node, error) {
|
||||||
|
genDocProvider := func() (*types.GenesisDoc, error) { return genDoc, nil }
|
||||||
|
dbProvider := func() (*dbm.DB, error) { return dbm.NewMemDB(), nil }
|
||||||
|
n, err := node.NewNode(cfg,
|
||||||
|
privVal,
|
||||||
|
proxy.NewLocalClientCreator(app),
|
||||||
|
genDocProvider,
|
||||||
|
dbProvider,
|
||||||
|
logger.With("module", "node"))
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
err = n.Start()
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return n, err
|
||||||
|
}
|
||||||
|
|
||||||
|
// start the LCD. note this blocks!
|
||||||
|
func startLCD(cdc *wire.Codec, listenAddr string, logger log.Logger) (net.Listener, error) {
|
||||||
|
handler := createHandler(cdc)
|
||||||
|
return StartHTTPServer(listenAddr, handler, logger)
|
||||||
|
}
|
||||||
|
|
||||||
func request(t *testing.T, port, method, path string, payload []byte) (*http.Response, string) {
|
func request(t *testing.T, port, method, path string, payload []byte) (*http.Response, string) {
|
||||||
var res *http.Response
|
var res *http.Response
|
||||||
var err error
|
var err error
|
||||||
|
|
|
@ -0,0 +1,24 @@
|
||||||
|
package client_test
|
||||||
|
|
||||||
|
import (
|
||||||
|
"os"
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
nm "github.com/tendermint/tendermint/node"
|
||||||
|
)
|
||||||
|
|
||||||
|
var node *nm.Node
|
||||||
|
|
||||||
|
// See https://golang.org/pkg/testing/#hdr-Main
|
||||||
|
// for more details
|
||||||
|
func TestMain(m *testing.M) {
|
||||||
|
// start a basecoind node and LCD server in the background to test against
|
||||||
|
|
||||||
|
// run all the tests against a single server instance
|
||||||
|
code := m.Run()
|
||||||
|
|
||||||
|
// tear down
|
||||||
|
|
||||||
|
//
|
||||||
|
os.Exit(code)
|
||||||
|
}
|
|
@ -2,10 +2,12 @@ package lcd
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"net/http"
|
"net/http"
|
||||||
|
"os"
|
||||||
|
|
||||||
"github.com/gorilla/mux"
|
"github.com/gorilla/mux"
|
||||||
"github.com/spf13/cobra"
|
"github.com/spf13/cobra"
|
||||||
"github.com/spf13/viper"
|
"github.com/spf13/viper"
|
||||||
|
"github.com/tendermint/tmlibs/log"
|
||||||
|
|
||||||
client "github.com/cosmos/cosmos-sdk/client"
|
client "github.com/cosmos/cosmos-sdk/client"
|
||||||
keys "github.com/cosmos/cosmos-sdk/client/keys"
|
keys "github.com/cosmos/cosmos-sdk/client/keys"
|
||||||
|
@ -18,7 +20,7 @@ import (
|
||||||
)
|
)
|
||||||
|
|
||||||
const (
|
const (
|
||||||
flagBind = "bind"
|
flagListenAddr = "laddr"
|
||||||
flagCORS = "cors"
|
flagCORS = "cors"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -29,25 +31,35 @@ func ServeCommand(cdc *wire.Codec) *cobra.Command {
|
||||||
cmd := &cobra.Command{
|
cmd := &cobra.Command{
|
||||||
Use: "rest-server",
|
Use: "rest-server",
|
||||||
Short: "Start LCD (light-client daemon), a local REST server",
|
Short: "Start LCD (light-client daemon), a local REST server",
|
||||||
RunE: startRESTServer(cdc),
|
RunE: startRESTServerFn(cdc),
|
||||||
}
|
}
|
||||||
// TODO: handle unix sockets also?
|
cmd.Flags().StringP(flagListenAddr, "a", "tcp://localhost:1317", "Address for server to listen on")
|
||||||
cmd.Flags().StringP(flagBind, "b", "localhost:1317", "Interface and port that server binds to")
|
|
||||||
cmd.Flags().String(flagCORS, "", "Set to domains that can make CORS requests (* for all)")
|
cmd.Flags().String(flagCORS, "", "Set to domains that can make CORS requests (* for all)")
|
||||||
cmd.Flags().StringP(client.FlagChainID, "c", "", "ID of chain we connect to")
|
cmd.Flags().StringP(client.FlagChainID, "c", "", "ID of chain we connect to")
|
||||||
cmd.Flags().StringP(client.FlagNode, "n", "tcp://localhost:46657", "Node to connect to")
|
cmd.Flags().StringP(client.FlagNode, "n", "tcp://localhost:46657", "Node to connect to")
|
||||||
return cmd
|
return cmd
|
||||||
}
|
}
|
||||||
|
|
||||||
func startRESTServer(cdc *wire.Codec) func(cmd *cobra.Command, args []string) error {
|
func startRESTServerFn(cdc *wire.Codec) func(cmd *cobra.Command, args []string) error {
|
||||||
return func(cmd *cobra.Command, args []string) error {
|
return func(cmd *cobra.Command, args []string) error {
|
||||||
bind := viper.GetString(flagBind)
|
listenAddr := viper.GetString(flagListenAddr)
|
||||||
r := initRouter(cdc)
|
handler := createHandler(cdc)
|
||||||
return http.ListenAndServe(bind, r)
|
logger := log.NewTMLogger(log.NewSyncWriter(os.Stdout)).
|
||||||
|
With("module", "rest-server")
|
||||||
|
listener, err := StartHTTPServer(listenAddr, handler, logger)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
// Wait forever and cleanup
|
||||||
|
cmn.TrapSignal(func() {
|
||||||
|
err := listener.Close()
|
||||||
|
logger.Error("Error closing listener", "err", err)
|
||||||
|
})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func initRouter(cdc *wire.Codec) http.Handler {
|
func createHandler(cdc *wire.Codec) http.Handler {
|
||||||
r := mux.NewRouter()
|
r := mux.NewRouter()
|
||||||
r.HandleFunc("/version", version.VersionRequestHandler).Methods("GET")
|
r.HandleFunc("/version", version.VersionRequestHandler).Methods("GET")
|
||||||
|
|
||||||
|
|
|
@ -42,6 +42,10 @@ func main() {
|
||||||
// get the codec
|
// get the codec
|
||||||
cdc := app.MakeCodec()
|
cdc := app.MakeCodec()
|
||||||
|
|
||||||
|
// TODO: setup keybase, viper object, etc. to be passed into
|
||||||
|
// the below functions and eliminate global vars, like we do
|
||||||
|
// with the cdc
|
||||||
|
|
||||||
// add standard rpc, and tx commands
|
// add standard rpc, and tx commands
|
||||||
rpc.AddCommands(basecliCmd)
|
rpc.AddCommands(basecliCmd)
|
||||||
basecliCmd.AddCommand(client.LineBreak)
|
basecliCmd.AddCommand(client.LineBreak)
|
||||||
|
|
|
@ -1,11 +1,14 @@
|
||||||
package rest
|
package rest
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"github.com/cosmos/cosmos-sdk/wire"
|
|
||||||
"github.com/gorilla/mux"
|
"github.com/gorilla/mux"
|
||||||
|
|
||||||
|
keys "github.com/tendermint/go-crypto/keys"
|
||||||
|
|
||||||
|
"github.com/cosmos/cosmos-sdk/wire"
|
||||||
)
|
)
|
||||||
|
|
||||||
// RegisterRoutes - Central function to define routes that get registered by the main application
|
// RegisterRoutes - Central function to define routes that get registered by the main application
|
||||||
func RegisterRoutes(r *mux.Router, cdc *wire.Codec) {
|
func RegisterRoutes(r *mux.Router, cdc *wire.Codec, kb keys.Keybase) {
|
||||||
r.HandleFunc("/accounts/{address}/send", SendRequestHandler(cdc)).Methods("POST")
|
r.HandleFunc("/accounts/{address}/send", SendRequestHandler(cdc, kb)).Methods("POST")
|
||||||
}
|
}
|
||||||
|
|
|
@ -7,6 +7,7 @@ import (
|
||||||
"net/http"
|
"net/http"
|
||||||
|
|
||||||
"github.com/gorilla/mux"
|
"github.com/gorilla/mux"
|
||||||
|
"github.com/tendermint/go-crypto/keys"
|
||||||
|
|
||||||
"github.com/cosmos/cosmos-sdk/client/builder"
|
"github.com/cosmos/cosmos-sdk/client/builder"
|
||||||
"github.com/cosmos/cosmos-sdk/client/keys"
|
"github.com/cosmos/cosmos-sdk/client/keys"
|
||||||
|
@ -26,7 +27,7 @@ type sendBody struct {
|
||||||
}
|
}
|
||||||
|
|
||||||
// SendRequestHandler - http request handler to send coins to a address
|
// SendRequestHandler - http request handler to send coins to a address
|
||||||
func SendRequestHandler(cdc *wire.Codec) func(http.ResponseWriter, *http.Request) {
|
func SendRequestHandler(cdc *wire.Codec, kb keys.Keybase) func(http.ResponseWriter, *http.Request) {
|
||||||
c := commands.Commander{cdc}
|
c := commands.Commander{cdc}
|
||||||
return func(w http.ResponseWriter, r *http.Request) {
|
return func(w http.ResponseWriter, r *http.Request) {
|
||||||
// collect data
|
// collect data
|
||||||
|
@ -47,13 +48,6 @@ func SendRequestHandler(cdc *wire.Codec) func(http.ResponseWriter, *http.Request
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
kb, err := keys.GetKeyBase()
|
|
||||||
if err != nil {
|
|
||||||
w.WriteHeader(http.StatusInternalServerError)
|
|
||||||
w.Write([]byte(err.Error()))
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
info, err := kb.Get(m.LocalAccountName)
|
info, err := kb.Get(m.LocalAccountName)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
w.WriteHeader(http.StatusUnauthorized)
|
w.WriteHeader(http.StatusUnauthorized)
|
||||||
|
|
Loading…
Reference in New Issue