main_test.go

package main

import (
	"flag"
	"os"
	"testing"

	"github.com/metadium/go-delegator/crypto"
	"github.com/metadium/go-delegator/json"

	"github.com/aws/aws-lambda-go/events"
)

func testHelp() {
	os.Args = os.Args[:1]
}

func testEnv() {
	os.Setenv(crypto.IsAwsLambda, "")
	os.Setenv(crypto.Path, "crypto/test/testkey")
	os.Setenv(crypto.Passphrase, "")
}

func testArg() {
	os.Setenv(crypto.IsAwsLambda, "")
	os.Args[1] = "crypto/test/testkey"
	os.Args[2] = ""
}

// FIXME: os.Setenv is not reflected to main()
func TestMain(m *testing.M) {
	//testHelp()
	//testEnv()
	testArg()

	flag.Parse()
	os.Exit(m.Run())
}

func TestLambdaHandler(t *testing.T) {
	req := json.RPCRequest{
		Jsonrpc: "2.0",
		Method:  "eth_getBalance",
		ID:      1,
	}
	req.Params = append(req.Params, "0xeeaf5f87cb85433a0db0fc31863b21d1c8279f7d")
	req.Params = append(req.Params, "latest")
	req.Params = append(req.Params, "ether")
	resp, err := lambdaHandler(nil, events.APIGatewayProxyRequest{
		Body: req.String(),
	})
	if resp.Body == "" || err != nil {
		t.Errorf("Failed to start main")
	}
}