status-go/node/status_node_rpc_client_test.go

package node

import (
	"context"
	"database/sql"
	"fmt"
	"io/ioutil"
	"os"
	"strings"
	"testing"

	"github.com/stretchr/testify/require"
	"go.uber.org/zap"

	"github.com/status-im/status-go/appdatabase"
	"github.com/status-im/status-go/multiaccounts"
	"github.com/status-im/status-go/params"
	"github.com/status-im/status-go/protocol/tt"
	"github.com/status-im/status-go/t/helpers"
	"github.com/status-im/status-go/walletdatabase"
)

type TestServiceAPI struct{}

func (api *TestServiceAPI) SomeMethod(_ context.Context) (string, error) {
	return "some method result", nil
}

func setupTestDBs() (appDB *sql.DB, walletDB *sql.DB, closeFn func() error, err error) {
	appDB, err = helpers.SetupTestMemorySQLDB(appdatabase.DbInitializer{})
	if err != nil {
		return nil, nil, nil, fmt.Errorf("failed to setup app db: %w", err)
	}

	walletDB, err = helpers.SetupTestMemorySQLDB(walletdatabase.DbInitializer{})
	if err != nil {
		return nil, nil, nil, fmt.Errorf("failed to setup wallet db: %w", err)
	}
	return appDB, walletDB, func() error {
		appErr := appDB.Close()
		walletErr := walletDB.Close()
		if appErr != nil {
			return fmt.Errorf("failed to close app db: %w", appErr)
		}
		if walletErr != nil {
			return fmt.Errorf("failed to close wallet db: %w", walletErr)
		}
		return nil
	}, err
}

func setupTestMultiDB() (*multiaccounts.Database, func() error, error) {
	tmpfile, err := ioutil.TempFile("", "tests")
	if err != nil {
		return nil, nil, err
	}
	db, err := multiaccounts.InitializeDB(tmpfile.Name())
	if err != nil {
		return nil, nil, err
	}
	return db, func() error {
		err := db.Close()
		if err != nil {
			return err
		}
		return os.Remove(tmpfile.Name())
	}, nil
}

func createAndStartStatusNode(config *params.NodeConfig) (*StatusNode, error) {
	statusNode := New(nil, tt.MustCreateTestLogger())

	appDB, walletDB, stop, err := setupTestDBs()
	defer func() {
		err := stop()
		if err != nil {
			statusNode.logger.Error("stopping db", zap.Error(err))
		}
	}()
	if err != nil {
		return nil, err
	}
	statusNode.appDB = appDB
	statusNode.walletDB = walletDB

	ma, stop2, err := setupTestMultiDB()
	defer func() {
		err := stop2()
		if err != nil {
			statusNode.logger.Error("stopping multiaccount db", zap.Error(err))
		}
	}()
	if err != nil {
		return nil, err
	}
	statusNode.multiaccountsDB = ma

	err = statusNode.Start(config, nil)
	if err != nil {
		return nil, err
	}

	return statusNode, nil
}

func createStatusNode() (*StatusNode, func() error, func() error, error) {
	appDB, walletDB, stop1, err := setupTestDBs()
	if err != nil {
		return nil, nil, nil, err
	}
	statusNode := New(nil, tt.MustCreateTestLogger())
	statusNode.SetAppDB(appDB)
	statusNode.SetWalletDB(walletDB)

	ma, stop2, err := setupTestMultiDB()
	statusNode.SetMultiaccountsDB(ma)

	return statusNode, stop1, stop2, err
}

func TestNodeRPCClientCallOnlyPublicAPIs(t *testing.T) {
	var err error

	statusNode, err := createAndStartStatusNode(&params.NodeConfig{
		APIModules: "", // no whitelisted API modules; use only public APIs
		WakuConfig: params.WakuConfig{
			Enabled: true,
		},
	})
	require.NoError(t, err)
	defer func() {
		err := statusNode.Stop()
		require.NoError(t, err)
	}()

	client := statusNode.RPCClient()
	require.NotNil(t, client)

	// call public API with public RPC Client
	result, err := statusNode.CallRPC(`{"jsonrpc": "2.0", "id": 1, "method": "eth_uninstallFilter", "params": ["id"]}`)
	require.NoError(t, err)

	// the call is successful
	require.False(t, strings.Contains(result, "error"))

	result, err = statusNode.CallRPC(`{"jsonrpc": "2.0", "id": 1, "method": "waku_info"}`)
	require.NoError(t, err)

	// call private API with public RPC client
	require.Equal(t, ErrRPCMethodUnavailable, result)

}

func TestNodeRPCPrivateClientCallPrivateService(t *testing.T) {
	var err error

	statusNode, err := createAndStartStatusNode(&params.NodeConfig{
		WakuConfig: params.WakuConfig{
			Enabled: true,
		},
	})
	require.NoError(t, err)
	defer func() {
		err := statusNode.Stop()
		require.NoError(t, err)
	}()

	result, err := statusNode.CallPrivateRPC(`{"jsonrpc": "2.0", "id": 1, "method": "waku_info"}`)
	require.NoError(t, err)

	// the call is successful
	require.False(t, strings.Contains(result, "error"))

	_, err = statusNode.CallPrivateRPC(`{"jsonrpc": "2.0", "id": 1, "method": "settings_getSettings"}`)
	require.NoError(t, err)
}