package rpcclient

import (
	"testing"

	"github.com/stretchr/testify/require"
)

// TestParseBitcoindVersion checks that the correct version from bitcoind's
// `getnetworkinfo` RPC call is parsed.
func TestParseBitcoindVersion(t *testing.T) {
	t.Parallel()

	testCases := []struct {
		name          string
		rpcVersion    string
		parsedVersion BitcoindVersion
	}{
		{
			name:          "parse version 0.19 and below",
			rpcVersion:    "/Satoshi:0.18.0/",
			parsedVersion: BitcoindPre19,
		},
		{
			name:          "parse version 0.19",
			rpcVersion:    "/Satoshi:0.19.0/",
			parsedVersion: BitcoindPre22,
		},
		{
			name:          "parse version 0.19 - 22.0",
			rpcVersion:    "/Satoshi:0.20.1/",
			parsedVersion: BitcoindPre22,
		},
		{
			name:          "parse version 22.0",
			rpcVersion:    "/Satoshi:22.0.0/",
			parsedVersion: BitcoindPre24,
		},
		{
			name:          "parse version 22.0 - 24.0",
			rpcVersion:    "/Satoshi:23.1.0/",
			parsedVersion: BitcoindPre24,
		},
		{
			name:          "parse version 24.0",
			rpcVersion:    "/Satoshi:24.0.0/",
			parsedVersion: BitcoindPre25,
		},
		{
			name:          "parse version 25.0",
			rpcVersion:    "/Satoshi:25.0.0/",
			parsedVersion: BitcoindPost25,
		},
		{
			name:          "parse version 25.0 and above",
			rpcVersion:    "/Satoshi:26.0.0/",
			parsedVersion: BitcoindPost25,
		},
	}

	for _, tc := range testCases {
		tc := tc

		t.Run(tc.name, func(t *testing.T) {
			version := parseBitcoindVersion(tc.rpcVersion)
			require.Equal(t, tc.parsedVersion, version)
		})
	}
}

// TestParseBtcdVersion checks that the correct version from btcd's `getinfo`
// RPC call is parsed.
func TestParseBtcdVersion(t *testing.T) {
	t.Parallel()

	testCases := []struct {
		name          string
		rpcVersion    int32
		parsedVersion BtcdVersion
	}{
		{
			name:          "parse version 0.24 and below",
			rpcVersion:    230000,
			parsedVersion: BtcdPre2401,
		},
		{
			name:          "parse version 0.24.1",
			rpcVersion:    240100,
			parsedVersion: BtcdPost2401,
		},
		{
			name:          "parse version 0.24.1 and above",
			rpcVersion:    250000,
			parsedVersion: BtcdPost2401,
		},
	}

	for _, tc := range testCases {
		tc := tc

		t.Run(tc.name, func(t *testing.T) {
			version := parseBtcdVersion(tc.rpcVersion)
			require.Equal(t, tc.parsedVersion, version)
		})
	}
}

// TestVersionSupports checks all the versions of bitcoind and btcd to ensure
// that the RPCs are supported correctly.
func TestVersionSupports(t *testing.T) {
	t.Parallel()

	require := require.New(t)

	// For bitcoind, unified softforks format is supported in 19.0 and
	// above.
	require.False(BitcoindPre19.SupportUnifiedSoftForks())
	require.True(BitcoindPre22.SupportUnifiedSoftForks())
	require.True(BitcoindPre24.SupportUnifiedSoftForks())
	require.True(BitcoindPre25.SupportUnifiedSoftForks())
	require.True(BitcoindPost25.SupportUnifiedSoftForks())

	// For bitcoind, `testmempoolaccept` is supported in 22.0 and above.
	require.False(BitcoindPre19.SupportTestMempoolAccept())
	require.False(BitcoindPre22.SupportTestMempoolAccept())
	require.True(BitcoindPre24.SupportTestMempoolAccept())
	require.True(BitcoindPre25.SupportTestMempoolAccept())
	require.True(BitcoindPost25.SupportTestMempoolAccept())

	// For bitcoind, `gettxspendingprevout` is supported in 24.0 and above.
	require.False(BitcoindPre19.SupportGetTxSpendingPrevOut())
	require.False(BitcoindPre22.SupportGetTxSpendingPrevOut())
	require.False(BitcoindPre24.SupportGetTxSpendingPrevOut())
	require.True(BitcoindPre25.SupportGetTxSpendingPrevOut())
	require.True(BitcoindPost25.SupportGetTxSpendingPrevOut())

	// For btcd, unified softforks format is supported in all versions.
	require.True(BtcdPre2401.SupportUnifiedSoftForks())
	require.True(BtcdPost2401.SupportUnifiedSoftForks())

	// For btcd, `testmempoolaccept` is supported in 24.1 and above.
	require.False(BtcdPre2401.SupportTestMempoolAccept())
	require.True(BtcdPost2401.SupportTestMempoolAccept())

	// For btcd, `gettxspendingprevout` is supported in 24.1 and above.
	require.False(BtcdPre2401.SupportGetTxSpendingPrevOut())
	require.True(BtcdPost2401.SupportGetTxSpendingPrevOut())
}