package ibc

import (
	"strings"
	"testing"

	"github.com/stretchr/testify/assert"
	"github.com/tendermint/basecoin/testutils"
	"github.com/tendermint/basecoin/types"
	cmn "github.com/tendermint/go-common"
	"github.com/tendermint/go-wire"
	tm "github.com/tendermint/tendermint/types"
)

func genGenesisDoc(chainID string, numVals int) (*tm.GenesisDoc, []*tm.Validator) {
	var vals []*tm.Validator
	genDoc := &tm.GenesisDoc{
		ChainID:    chainID,
		Validators: nil,
	}

	for i := 0; i < numVals; i++ {
		name := cmn.Fmt("%v_val_%v", chainID, i)
		valPrivAcc := testutils.PrivAccountFromSecret(name)
		val := tm.NewValidator(valPrivAcc.Account.PubKey, 1)
		genDoc.Validators = append(genDoc.Validators, tm.GenesisValidator{
			PubKey: val.PubKey,
			Amount: 1,
			Name:   name,
		})
		vals = append(vals, val)
	}

	return genDoc, vals
}

func TestIBCPlugin(t *testing.T) {

	store := types.NewKVCache(nil)
	store.SetLogging() // Log all activity

	ibcPlugin := New()
	ctx := types.CallContext{
		CallerAddress: nil,
		CallerAccount: nil,
		Coins:         types.Coins{},
	}

	chainID_1 := "test_chain"
	genDoc_1, vals_1 := genGenesisDoc(chainID_1, 4)
	genDocJSON_1 := wire.JSONBytesPretty(genDoc_1)

	// Register a malformed chain
	res := ibcPlugin.RunTx(store, ctx, wire.BinaryBytes(struct{ IBCTx }{IBCRegisterChainTx{
		BlockchainGenesis{
			ChainID: "test_chain",
			Genesis: "<THIS IS NOT JSON>",
		},
	}}))
	assert.Equal(t, res.Code, IBCCodeEncodingError)
	t.Log(">>", strings.Join(store.GetLogLines(), "\n"))
	store.ClearLogLines()

	// Successfully register a chain
	res = ibcPlugin.RunTx(store, ctx, wire.BinaryBytes(struct{ IBCTx }{IBCRegisterChainTx{
		BlockchainGenesis{
			ChainID: "test_chain",
			Genesis: string(genDocJSON_1),
		},
	}}))
	assert.True(t, res.IsOK(), res)
	t.Log(">>", strings.Join(store.GetLogLines(), "\n"))
	store.ClearLogLines()

	// Duplicate request fails
	res = ibcPlugin.RunTx(store, ctx, wire.BinaryBytes(struct{ IBCTx }{IBCRegisterChainTx{
		BlockchainGenesis{
			ChainID: "test_chain",
			Genesis: string(genDocJSON_1),
		},
	}}))
	assert.Equal(t, res.Code, IBCCodeChainAlreadyExists, res)
	t.Log(">>", strings.Join(store.GetLogLines(), "\n"))
	store.ClearLogLines()

	t.Log(">>", vals_1)
}