package rpc

import (
	"context"
	"encoding/json"
	"fmt"
	"github.com/ethereum/go-ethereum/rpc"
	"github.com/stretchr/testify/require"
	"io/ioutil"
	"math/rand"
	"net/http"
	"strings"
	"testing"
)

type TestService struct{}

func (s *TestService) Foo(arg string) string {
	return arg
}

func TestStartHTTPEndpointStartStop(t *testing.T) {
	config := &Config{
		RPCAddr: "127.0.0.1",
		RPCPort: randomPort(),
	}
	ctx, cancel := context.WithCancel(context.Background())
	_, err := StartHTTPEndpoint(ctx, config, []rpc.API{
		{
			Namespace: "test",
			Version:   "1.0",
			Service:   &TestService{},
			Public:    true,
		},
	})
	require.Nil(t, err, "unexpected error")
	res, err := rpcCall(config.RPCPort, "test_foo", []string{"baz"})
	require.Nil(t, err, "unexpected error")
	resStr := res.(string)
	require.Equal(t, "baz", resStr)
	cancel()
	_, err = rpcCall(config.RPCPort, "test_foo", []string{"baz"})
	require.NotNil(t, err)
}

func rpcCall(port int, method string, params []string) (interface{}, error) {
	parsedParams, err := json.Marshal(params)
	if err != nil {
		return nil, err
	}
	fullBody := fmt.Sprintf(`{ "id": 1, "jsonrpc": "2.0", "method": "%s", "params": %s }`,
		method, string(parsedParams))
	res, err := http.Post(fmt.Sprintf("http://127.0.0.1:%d", port), "application/json", strings.NewReader(fullBody))
	if err != nil {
		return nil, err
	}
	data, err := ioutil.ReadAll(res.Body)
	if err != nil {
		return nil, err
	}

	var out map[string]interface{}
	err = json.Unmarshal(data, &out)
	if err != nil {
		return nil, err
	}

	result := out["result"].(interface{})
	return result, nil
}

func randomPort() int {
	return rand.Intn(65535-1025) + 1025
}