decred.org/dcrdex@v1.0.3/client/webserver/webserver_test.go

//go:build !live

package webserver

import (
	"bytes"
	"context"
	"encoding/hex"
	"encoding/json"
	"errors"
	"fmt"
	"io"
	"math/rand"
	"net"
	"net/http"
	"os"
	"strings"
	"testing"
	"time"

	"decred.org/dcrdex/client/asset"
	"decred.org/dcrdex/client/core"
	"decred.org/dcrdex/client/db"
	"decred.org/dcrdex/client/mnemonic"
	"decred.org/dcrdex/client/orderbook"
	"decred.org/dcrdex/dex"
	"decred.org/dcrdex/dex/encode"
	"decred.org/dcrdex/dex/order"
	"github.com/go-chi/chi/v5"
)

var (
	tErr    = fmt.Errorf("expected dummy error")
	tLogger dex.Logger
	tCtx    context.Context
)

type tCoin struct {
	id       []byte
	confs    uint32
	confsErr error
}

func (c *tCoin) ID() dex.Bytes {
	return c.id
}

func (c *tCoin) String() string {
	return hex.EncodeToString(c.id)
}

func (c *tCoin) TxID() string {
	return hex.EncodeToString(c.id)
}

func (c *tCoin) Value() uint64 {
	return 0
}

func (c *tCoin) Confirmations(context.Context) (uint32, error) {
	return c.confs, c.confsErr
}

type TCore struct {
	balanceErr       error
	syncFeed         core.BookFeed
	syncErr          error
	postBondErr      error
	loginErr         error
	logoutErr        error
	initErr          error
	isInited         bool
	getDEXConfigErr  error
	createWalletErr  error
	openWalletErr    error
	closeWalletErr   error
	rescanWalletErr  error
	sendErr          error
	notHas           bool
	notRunning       bool
	notOpen          bool
	rateSourceErr    error
	estFee           uint64
	estFeeErr        error
	validAddr        bool
	walletDisabled   bool
	walletStatusErr  error
	deletedRecords   int
	deleteRecordsErr error
	tradeErr         error
	notes            []*db.Notification
	notesErr         error
}

func (c *TCore) Network() dex.Network                         { return dex.Mainnet }
func (c *TCore) Exchanges() map[string]*core.Exchange         { return nil }
func (c *TCore) Exchange(host string) (*core.Exchange, error) { return nil, nil }
func (c *TCore) GetDEXConfig(dexAddr string, certI any) (*core.Exchange, error) {
	return nil, c.getDEXConfigErr // TODO along with test for apiUser / Exchanges() / User()
}
func (c *TCore) AddDEX(appPW []byte, dexAddr string, certI any) error {
	return nil
}
func (c *TCore) DiscoverAccount(dexAddr string, pw []byte, certI any) (*core.Exchange, bool, error) {
	return nil, false, nil
}
func (c *TCore) PostBond(r *core.PostBondForm) (*core.PostBondResult, error) {
	return nil, c.postBondErr
}
func (c *TCore) RedeemPrepaidBond(appPW []byte, code []byte, host string, certI any) (tier uint64, err error) {
	return 1, nil
}
func (c *TCore) UpdateBondOptions(form *core.BondOptionsForm) error {
	return c.postBondErr
}
func (c *TCore) BondsFeeBuffer(assetID uint32) (uint64, error) {
	return 222, nil
}
func (c *TCore) ToggleRateSourceStatus(src string, disable bool) error {
	return c.rateSourceErr
}
func (c *TCore) FiatRateSources() map[string]bool {
	return nil
}

func (c *TCore) InitializeClient(pw []byte, seed *string) (string, error) {
	var mnemonicSeed string
	if seed == nil {
		_, mnemonicSeed = mnemonic.New()
	}
	return mnemonicSeed, c.initErr
}
func (c *TCore) Login(pw []byte) error { return c.loginErr }
func (c *TCore) IsInitialized() bool   { return c.isInited }
func (c *TCore) SyncBook(dex string, base, quote uint32) (*orderbook.OrderBook, core.BookFeed, error) {
	return nil, c.syncFeed, c.syncErr
}
func (c *TCore) Book(dex string, base, quote uint32) (*core.OrderBook, error) {
	return &core.OrderBook{}, nil
}
func (c *TCore) AssetBalance(assetID uint32) (*core.WalletBalance, error) { return nil, c.balanceErr }
func (c *TCore) WalletState(assetID uint32) *core.WalletState {
	if c.notHas {
		return nil
	}
	return &core.WalletState{
		Symbol:  unbip(assetID),
		AssetID: assetID,
		Open:    !c.notOpen,
		Running: !c.notRunning,
	}
}
func (c *TCore) CreateWallet(appPW, walletPW []byte, form *core.WalletForm) error {
	return c.createWalletErr
}
func (c *TCore) RescanWallet(assetID uint32, force bool) error    { return c.rescanWalletErr }
func (c *TCore) OpenWallet(assetID uint32, pw []byte) error       { return c.openWalletErr }
func (c *TCore) CloseWallet(assetID uint32) error                 { return c.closeWalletErr }
func (c *TCore) ConnectWallet(assetID uint32) error               { return nil }
func (c *TCore) Wallets() []*core.WalletState                     { return nil }
func (c *TCore) WalletSettings(uint32) (map[string]string, error) { return nil, nil }
func (c *TCore) ReconfigureWallet(aPW, nPW []byte, form *core.WalletForm) error {
	return nil
}
func (c *TCore) ToggleWalletStatus(assetID uint32, disable bool) error {
	if c.walletStatusErr != nil {
		return c.walletStatusErr
	}
	c.walletDisabled = disable
	return c.walletStatusErr
}
func (c *TCore) ChangeAppPass(appPW, newAppPW []byte) error                         { return nil }
func (c *TCore) ResetAppPass(newAppPW []byte, seed string) error                    { return nil }
func (c *TCore) SetWalletPassword(appPW []byte, assetID uint32, newPW []byte) error { return nil }
func (c *TCore) NewDepositAddress(assetID uint32) (string, error)                   { return "", nil }
func (c *TCore) AutoWalletConfig(assetID uint32, walletType string) (map[string]string, error) {
	return nil, nil
}
func (c *TCore) User() *core.User { return nil }
func (c *TCore) SupportedAssets() map[uint32]*core.SupportedAsset {
	return make(map[uint32]*core.SupportedAsset)
}
func (c *TCore) Send(pw []byte, assetID uint32, value uint64, address string, subtract bool) (asset.Coin, error) {
	return &tCoin{id: []byte{0xde, 0xc7, 0xed}}, c.sendErr
}
func (c *TCore) ValidateAddress(address string, assetID uint32) (bool, error) {
	return c.validAddr, nil
}
func (c *TCore) EstimateSendTxFee(addr string, assetID uint32, value uint64, subtract, maxWithdraw bool) (fee uint64, isValidAddress bool, err error) {
	return c.estFee, true, c.estFeeErr
}
func (c *TCore) Trade(pw []byte, form *core.TradeForm) (*core.Order, error) {
	if c.tradeErr != nil {
		return nil, c.tradeErr
	}
	return trade(form), nil
}
func (c *TCore) TradeAsync(pw []byte, form *core.TradeForm) (*core.InFlightOrder, error) {
	if c.tradeErr != nil {
		return nil, c.tradeErr
	}
	return &core.InFlightOrder{
		Order:       trade(form),
		TemporaryID: uint64(rand.Int63()),
	}, nil
}
func trade(form *core.TradeForm) *core.Order {
	oType := order.LimitOrderType
	if !form.IsLimit {
		oType = order.MarketOrderType
	}
	return &core.Order{
		Type:  oType,
		Stamp: uint64(time.Now().UnixMilli()),
		Rate:  form.Rate,
		Qty:   form.Qty,
		Sell:  form.Sell,
	}
}
func (c *TCore) Cancel(oid dex.Bytes) error { return nil }

func (c *TCore) NotificationFeed() *core.NoteFeed {
	return &core.NoteFeed{
		C: make(chan core.Notification, 1),
	}
}

func (c *TCore) AckNotes(ids []dex.Bytes) {}

func (c *TCore) Logout() error { return c.logoutErr }

func (c *TCore) Orders(*core.OrderFilter) ([]*core.Order, error) { return nil, nil }
func (c *TCore) Order(oid dex.Bytes) (*core.Order, error)        { return nil, nil }
func (c *TCore) MaxBuy(host string, base, quote uint32, rate uint64) (*core.MaxOrderEstimate, error) {
	return nil, nil
}
func (c *TCore) MaxSell(host string, base, quote uint32) (*core.MaxOrderEstimate, error) {
	return nil, nil
}
func (c *TCore) PreOrder(*core.TradeForm) (*core.OrderEstimate, error) {
	return nil, nil
}
func (c *TCore) AccountExport(pw []byte, host string) (*core.Account, []*db.Bond, error) {
	return nil, nil, nil
}
func (c *TCore) AccountImport(pw []byte, account *core.Account, bonds []*db.Bond) error {
	return nil
}
func (c *TCore) ToggleAccountStatus(pw []byte, host string, disable bool) error { return nil }

func (c *TCore) ExportSeed(pw []byte) (string, error) {
	return "seed words here", nil
}
func (c *TCore) WalletLogFilePath(uint32) (string, error) {
	return "", nil
}
func (c *TCore) AccelerateOrder(pw []byte, oidB dex.Bytes, newFeeRate uint64) (string, error) {
	return "", nil
}
func (c *TCore) AccelerationEstimate(oidB dex.Bytes, newFeeRate uint64) (uint64, error) {
	return 0, nil
}
func (c *TCore) PreAccelerateOrder(oidB dex.Bytes) (*core.PreAccelerate, error) {
	return nil, nil
}
func (c *TCore) RecoverWallet(uint32, []byte, bool) error {
	return nil
}
func (c *TCore) UpdateCert(string, []byte) error {
	return nil
}
func (c *TCore) UpdateDEXHost(string, string, []byte, any) (*core.Exchange, error) {
	return nil, nil
}
func (c *TCore) WalletRestorationInfo(pw []byte, assetID uint32) ([]*asset.WalletRestoration, error) {
	return nil, nil
}
func (c *TCore) DeleteArchivedRecordsWithBackup(endDateTime *time.Time, saveMatchesToFile, saveOrdersToFile bool) (string, int, error) {
	return "/path/to/records", c.deletedRecords, c.deleteRecordsErr
}
func (c *TCore) WalletPeers(assetID uint32) ([]*asset.WalletPeer, error) {
	return nil, nil
}
func (c *TCore) AddWalletPeer(assetID uint32, address string) error {
	return nil
}
func (c *TCore) RemoveWalletPeer(assetID uint32, address string) error {
	return nil
}
func (c *TCore) Notifications(n int) (notes, pokes []*db.Notification, _ error) {
	return c.notes, []*db.Notification{}, c.notesErr
}
func (c *TCore) ApproveToken(appPW []byte, assetID uint32, dexAddr string, onConfirm func()) (string, error) {
	return "", nil
}
func (c *TCore) UnapproveToken(appPW []byte, assetID uint32, version uint32) (string, error) {
	return "", nil
}
func (c *TCore) ApproveTokenFee(assetID uint32, version uint32, approval bool) (uint64, error) {
	return 0, nil
}
func (c *TCore) StakeStatus(assetID uint32) (*asset.TicketStakingStatus, error) {
	return nil, nil
}

func (c *TCore) SetVSP(assetID uint32, addr string) error {
	return nil
}

func (c *TCore) PurchaseTickets(assetID uint32, appPW []byte, n int) error {
	return nil
}

func (c *TCore) SetVotingPreferences(assetID uint32, choices, tSpendPolicy, treasuryPolicy map[string]string) error {
	return nil
}

func (c *TCore) ListVSPs(assetID uint32) ([]*asset.VotingServiceProvider, error) {
	return nil, nil
}

func (c *TCore) TicketPage(assetID uint32, scanStart int32, n, skipN int) ([]*asset.Ticket, error) {
	return nil, nil
}

func (c *TCore) TxHistory(assetID uint32, n int, refID *string, past bool) ([]*asset.WalletTransaction, error) {
	return nil, nil
}

func (c *TCore) FundsMixingStats(assetID uint32) (*asset.FundsMixingStats, error) {
	return nil, nil
}

func (c *TCore) ConfigureFundsMixer(appPW []byte, assetID uint32, enabled bool) error {
	return nil
}

func (c *TCore) RedeemGeocode(appPW, code []byte, msg string) (dex.Bytes, uint64, error) {
	return nil, 0, nil
}

func (*TCore) SetLanguage(string) error { return nil }
func (*TCore) Language() string         { return "en-US" }

func (*TCore) TakeAction(assetID uint32, actionID string, actionB json.RawMessage) error { return nil }

func (*TCore) ExtensionModeConfig() *core.ExtensionModeConfig {
	return nil
}

type TWriter struct {
	b []byte
}

func (*TWriter) Header() http.Header {
	return http.Header{}
}

func (w *TWriter) Write(b []byte) (int, error) {
	w.b = b
	return len(b), nil
}

func (w *TWriter) WriteHeader(int) {}

type TReader struct {
	msg []byte
	err error
}

func (r *TReader) Read(p []byte) (n int, err error) {
	if r.err != nil {
		return 0, r.err
	}
	if len(r.msg) == 0 {
		return 0, io.EOF
	}
	copy(p, r.msg)
	if len(p) < len(r.msg) {
		r.msg = r.msg[:len(p)]
		return len(p), nil
	}
	l := len(r.msg)
	r.msg = nil
	return l, io.EOF
}

func (r *TReader) Close() error { return nil }

func newTServer(t *testing.T, start bool) (*WebServer, *TCore, func()) {
	t.Helper()
	c := &TCore{}
	var shutdown func()
	ctx, killCtx := context.WithCancel(tCtx)
	s, err := New(&Config{
		Core:   c,
		Addr:   "127.0.0.1:0",
		Logger: tLogger,
	})
	if err != nil {
		t.Fatalf("error creating server: %v", err)
	}

	if start {
		cm := dex.NewConnectionMaster(s)
		err := cm.Connect(ctx)
		if err != nil {
			t.Fatalf("Error starting WebServer: %v", err)
		}
		shutdown = func() {
			killCtx()
			cm.Disconnect()
		}
	} else {
		shutdown = killCtx
	}
	return s, c, shutdown
}

func ensureResponse(t *testing.T, f func(w http.ResponseWriter, r *http.Request), want string, reader *TReader, writer *TWriter, body any, cookies map[string]string) {
	t.Helper()
	var err error
	reader.msg, err = json.Marshal(body)
	if err != nil {
		t.Fatalf("error marshalling request body: %v", err)
	}
	req, err := http.NewRequest("GET", "/", reader)
	if err != nil {
		t.Fatalf("error creating request: %v", err)
	}
	for name, value := range cookies {
		cookie := http.Cookie{
			Name:  name,
			Value: value,
		}
		req.AddCookie(&cookie)
	}
	f(writer, req)
	if len(writer.b) == 0 {
		t.Fatalf("no response")
	}
	// Drop the line feed.
	errMsg := string(writer.b[:len(writer.b)-1])
	if errMsg != want {
		t.Fatalf("wrong response. expected %s, got %s", want, errMsg)
	}
	writer.b = nil
}

func TestMain(m *testing.M) {
	tLogger = dex.StdOutLogger("TEST", dex.LevelTrace)
	var shutdown func()
	tCtx, shutdown = context.WithCancel(context.Background())
	doIt := func() int {
		// Not counted as coverage, must test Archiver constructor explicitly.
		defer shutdown()
		return m.Run()
	}
	os.Exit(doIt())
}

func TestNew_siteError(t *testing.T) {
	cwd, err := os.Getwd()
	if err != nil {
		t.Fatalf("cannot get current directory: %v", err)
	}

	// Change to a directory with no "site" or "../../webserver/site" folder.
	dir := t.TempDir()
	defer os.Chdir(cwd) // leave the temp dir before trying to delete it

	if err = os.Chdir(dir); err != nil {
		t.Fatalf("Cannot cd to %q", dir)
	}

	c := &TCore{}
	_, err = New(&Config{
		Core:    c,
		Addr:    "127.0.0.1:0",
		Logger:  tLogger,
		NoEmbed: true, // this tests locating on-disk files, not the embedded ones
	})
	if err == nil || !strings.HasPrefix(err.Error(), "no HTML template files found") {
		t.Errorf("Should have failed to start with no site folder.")
	}
}

func TestConnectStart(t *testing.T) {
	_, _, shutdown := newTServer(t, true)
	defer shutdown()
}

func TestConnectBindError(t *testing.T) {
	s0, _, shutdown := newTServer(t, true)
	defer shutdown()

	tAddr := s0.addr
	s, err := New(&Config{
		Core:   &TCore{},
		Addr:   tAddr,
		Logger: tLogger,
	})
	if err != nil {
		t.Fatalf("error creating server: %v", err)
	}

	cm := dex.NewConnectionMaster(s)
	err = cm.Connect(tCtx)
	if err == nil {
		shutdown() // shutdown both servers with shared context
		cm.Disconnect()
		t.Fatalf("should have failed to bind")
	}
}

func TestAPILogin(t *testing.T) {
	writer := new(TWriter)
	var body any
	reader := new(TReader)
	s, tCore, shutdown := newTServer(t, false)
	defer shutdown()

	ensure := func(want string) {
		t.Helper()
		ensureResponse(t, s.apiLogin, want, reader, writer, body, nil)
	}

	goodBody := &loginForm{
		Pass: encode.PassBytes("def"),
	}
	body = goodBody
	ensure(`{"ok":true,"notes":null,"pokes":[]}`)

	tCore.notes = []*db.Notification{{
		TopicID: core.TopicAccountUnlockError,
	}}
	ensure(`{"ok":true,"notes":[{"type":"","topic":"AccountUnlockError","subject":"","details":"","severity":0,"stamp":0,"acked":false,"id":""}],"pokes":[]}`)

	tCore.notes = nil
	tCore.notesErr = errors.New("")
	ensure(`{"ok":true,"notes":null,"pokes":[]}`)

	// Login error
	tCore.loginErr = tErr
	ensure(fmt.Sprintf(`{"ok":false,"msg":"%s"}`, tErr))
	tCore.loginErr = nil
}

func TestSend(t *testing.T) {
	writer := new(TWriter)
	var body any
	reader := new(TReader)
	s, tCore, shutdown := newTServer(t, false)
	defer shutdown()

	isOK := func() bool {
		reader.msg, _ = json.Marshal(body)
		req, err := http.NewRequest("GET", "/", reader)
		if err != nil {
			t.Fatalf("error creating request: %v", err)
		}
		s.apiSend(writer, req)
		if len(writer.b) == 0 {
			t.Fatalf("no response")
		}
		resp := &standardResponse{}
		err = json.Unmarshal(writer.b, resp)
		if err != nil {
			t.Fatalf("json unmarshal error: %v", err)
		}
		return resp.OK
	}

	body = &sendForm{
		Pass: encode.PassBytes("dummyAppPass"),
	}

	// initial success
	if !isOK() {
		t.Fatalf("not ok: %s", string(writer.b))
	}

	// no wallet
	tCore.notHas = true
	if isOK() {
		t.Fatalf("no error for missing wallet")
	}
	tCore.notHas = false

	// Send/Withdraw error
	tCore.sendErr = tErr
	if isOK() {
		t.Fatalf("no error for Send/Withdraw error")
	}
	tCore.sendErr = nil

	// re-success
	if !isOK() {
		t.Fatalf("not ok afterwards: %s", string(writer.b))
	}
}

func TestAPIInit(t *testing.T) {
	writer := new(TWriter)
	var body any
	reader := new(TReader)
	s, tCore, shutdown := newTServer(t, false)
	defer shutdown()

	ensure := func(f func(http.ResponseWriter, *http.Request), want string) {
		t.Helper()
		ensureResponse(t, f, want, reader, writer, body, nil)
	}

	body = struct{}{}

	// Success but uninitialized
	ensure(s.apiIsInitialized, `{"ok":true,"initialized":false}`)

	// Now initialized
	tCore.isInited = true
	ensure(s.apiIsInitialized, `{"ok":true,"initialized":true}`)

	// Initialization error
	tCore.initErr = tErr
	ensure(s.apiInit, fmt.Sprintf(`{"ok":false,"msg":"%s"}`, tErr))
	tCore.initErr = nil
}

// TODO: TestAPIGetDEXInfo

func TestAPINewWallet(t *testing.T) {
	writer := new(TWriter)
	var body any
	reader := new(TReader)
	s, tCore, shutdown := newTServer(t, false)
	defer shutdown()

	ensure := func(want string) {
		ensureResponse(t, s.apiNewWallet, want, reader, writer, body, nil)
	}

	body = &newWalletForm{
		Pass:  encode.PassBytes("abc"),
		AppPW: encode.PassBytes("dummyAppPass"),
	}
	tCore.notHas = true
	ensure(`{"ok":true}`)

	tCore.notHas = false
	ensure(`{"ok":false,"msg":"already have a wallet for btc"}`)
	tCore.notHas = true

	tCore.createWalletErr = tErr
	ensure(fmt.Sprintf(`{"ok":false,"msg":"%s"}`, tErr))
	tCore.createWalletErr = nil

	tCore.notHas = false
}

func TestAPILogout(t *testing.T) {
	writer := new(TWriter)
	reader := new(TReader)
	s, tCore, shutdown := newTServer(t, false)
	defer shutdown()

	ensure := func(want string) {
		ensureResponse(t, s.apiLogout, want, reader, writer, nil, nil)
	}
	ensure(`{"ok":true}`)

	// Logout error
	tCore.logoutErr = tErr
	ensure(fmt.Sprintf(`{"ok":false,"msg":"%s"}`, tErr))
	tCore.logoutErr = nil
}

func TestApiGetBalance(t *testing.T) {
	writer := new(TWriter)
	reader := new(TReader)
	s, tCore, shutdown := newTServer(t, false)
	defer shutdown()

	ensure := func(want string) {
		ensureResponse(t, s.apiGetBalance, want, reader, writer, struct{}{}, nil)
	}
	ensure(`{"ok":true,"balance":null}`)

	// Logout error
	tCore.balanceErr = tErr
	ensure(fmt.Sprintf(`{"ok":false,"msg":"%s"}`, tErr))
	tCore.balanceErr = nil
}

type tHTTPHandler struct {
	req *http.Request
}

func (h *tHTTPHandler) ServeHTTP(_ http.ResponseWriter, req *http.Request) {
	h.req = req
}

func TestOrderIDCtx(t *testing.T) {
	hexOID := hex.EncodeToString(encode.RandomBytes(32))
	req := (&http.Request{}).WithContext(context.WithValue(context.Background(), chi.RouteCtxKey, &chi.Context{
		URLParams: chi.RouteParams{
			Keys:   []string{"oid"},
			Values: []string{hexOID},
		},
	}))

	tNextHandler := &tHTTPHandler{}
	handlerFunc := orderIDCtx(tNextHandler)
	handlerFunc.ServeHTTP(nil, req)

	reqCtx := tNextHandler.req.Context()
	untypedOID := reqCtx.Value(ctxOID)
	if untypedOID == nil {
		t.Fatalf("oid not embedded in request context")
	}
	oidStr, ok := untypedOID.(string)
	if !ok {
		t.Fatalf("string type assertion failed")
	}

	if oidStr != hexOID {
		t.Fatalf("wrong value embedded in request context. wanted %s, got %s", hexOID, oidStr)
	}
}

func TestGetOrderIDCtx(t *testing.T) {
	oid := encode.RandomBytes(32)
	hexOID := hex.EncodeToString(oid)

	r := (&http.Request{}).WithContext(context.WithValue(context.Background(), ctxOID, hexOID))

	bytesOut, err := getOrderIDCtx(r)
	if err != nil {
		t.Fatalf("getOrderIDCtx error: %v", err)
	}
	if len(bytesOut) == 0 {
		t.Fatalf("empty oid")
	}
	if !bytes.Equal(oid, bytesOut) {
		t.Fatalf("wrong bytes. wanted %x, got %s", oid, bytesOut)
	}

	// Test some negative paths
	for name, v := range map[string]any{
		"nil":          nil,
		"int":          5,
		"wrong length": "abc",
		"not hex":      "zyxwzyxwzyxwzyxwzyxwzyxwzyxwzyxwzyxwzyxwzyxwzyxwzyxwzyxwzyxwzyxw",
	} {
		r := (&http.Request{}).WithContext(context.WithValue(context.Background(), ctxOID, v))
		_, err := getOrderIDCtx(r)
		if err == nil {
			t.Fatalf("no error for %v", name)
		}
	}
}

func TestPasswordCache(t *testing.T) {
	s, tCore, shutdown := newTServer(t, false)
	defer shutdown()

	password := encode.PassBytes("def")
	authToken1 := s.authorize()
	authToken2 := s.authorize()

	key1, err := s.cacheAppPassword(password, authToken1)
	if err != nil {
		t.Fatalf("error caching password: %v", err)
	}

	key2, err := s.cacheAppPassword(password, authToken2)
	if err != nil {
		t.Fatalf("error caching password: %v", err)
	}

	retrievedPW, err := s.getCachedPassword(key1, authToken1)
	if err != nil {
		t.Fatalf("error getting password: %v", err)
	}
	if !bytes.Equal(password, retrievedPW) {
		t.Fatalf("retrieved PW not same: %v - %v", password, retrievedPW)
	}

	retrievedPW, err = s.getCachedPassword(key2, authToken2)
	if err != nil {
		t.Fatalf("error getting password: %v", err)
	}
	if !bytes.Equal(password, retrievedPW) {
		t.Fatalf("retrieved PW not same: %v - %v", password, retrievedPW)
	}

	// test new wallet request first without the cookies populated, then with
	writer := new(TWriter)
	reader := new(TReader)
	body := &newWalletForm{
		Pass: encode.PassBytes(""),
	}
	want := `{"ok":false,"msg":"app pass cannot be empty"}`
	tCore.notHas = true
	ensureResponse(t, s.apiNewWallet, want, reader, writer, body, nil)

	want = `{"ok":true}`
	ensureResponse(t, s.apiNewWallet, want, reader, writer, body, map[string]string{
		authCK:  authToken1,
		pwKeyCK: hex.EncodeToString(key1),
	})

	s.apiLogout(writer, nil)

	if len(s.cachedPasswords) != 0 {
		t.Fatal("logout should clear all cached passwords")
	}
}

func TestAPI_ToggleRatesource(t *testing.T) {
	s, tCore, shutdown := newTServer(t, false)
	defer shutdown()

	writer := new(TWriter)
	reader := new(TReader)

	type rateSourceForm struct {
		Disable bool   `json:"disable"`
		Source  string `json:"source"`
	}

	// Test enabling fiat rate sources.
	enableTests := []struct {
		name, source, want string
		wantErr            error
	}{{
		name:    "Invalid rate source",
		source:  "binance",
		wantErr: errors.New("cannot enable unknown fiat rate source"),
		want:    `{"ok":false,"msg":"cannot enable unknown fiat rate source"}`,
	}, {
		name:   "ok valid source",
		source: "dcrdata",
		want:   `{"ok":true}`,
	}, {
		name:   "ok already initialized",
		source: "dcrdata",
		want:   `{"ok":true}`,
	}}

	for _, test := range enableTests {
		body := &rateSourceForm{
			Disable: false,
			Source:  test.source,
		}
		tCore.rateSourceErr = test.wantErr
		ensureResponse(t, s.apiToggleRateSource, test.want, reader, writer, body, nil)
	}

	// Test disabling fiat rate sources.
	disableTests := []struct {
		name, source, want string
		wantErr            error
	}{{
		name:    "Invalid rate source",
		source:  "binance",
		wantErr: errors.New("cannot disable unknown fiat rate source"),
		want:    `{"ok":false,"msg":"cannot disable unknown fiat rate source"}`,
	}, {
		name:   "ok valid source",
		source: "Messari",
		want:   `{"ok":true}`,
	}, {
		name:   "ok already disabled/not initialized",
		source: "Messari",
		want:   `{"ok":true}`,
	}}

	for _, test := range disableTests {
		body := &rateSourceForm{
			Disable: true,
			Source:  test.source,
		}
		tCore.rateSourceErr = test.wantErr
		ensureResponse(t, s.apiToggleRateSource, test.want, reader, writer, body, nil)
	}
}

func TestAPIValidateAddress(t *testing.T) {
	s, tCore, shutdown := newTServer(t, false)
	defer shutdown()

	writer := new(TWriter)
	reader := new(TReader)
	testID := uint32(42)

	body := &struct {
		Addr    string  `json:"addr"`
		AssetID *uint32 `json:"assetID"`
	}{
		Addr:    "addr",
		AssetID: &testID,
	}

	want := `{"ok":true}`
	tCore.validAddr = true
	ensureResponse(t, s.apiValidateAddress, want, reader, writer, body, nil)

	want = `{"ok":false}`
	tCore.validAddr = false
	ensureResponse(t, s.apiValidateAddress, want, reader, writer, body, nil)
}

func TestAPIEstimateSendTxFee(t *testing.T) {
	s, tCore, shutdown := newTServer(t, false)
	defer shutdown()

	writer := new(TWriter)
	reader := new(TReader)
	testID := uint32(42)

	body := &sendTxFeeForm{
		Addr:     "addr",
		Value:    1e8,
		Subtract: false,
		AssetID:  &testID,
	}

	want := `{"ok":true,"txfee":10000,"validaddress":true}`
	tCore.estFee = 10000
	ensureResponse(t, s.apiEstimateSendTxFee, want, reader, writer, body, nil)

	want = fmt.Sprintf(`{"ok":false,"msg":"%s"}`, tErr)
	tCore.estFeeErr = tErr
	ensureResponse(t, s.apiEstimateSendTxFee, want, reader, writer, body, nil)
}

func TestAPIToggleWalletStatus(t *testing.T) {
	s, tCore, shutdown := newTServer(t, false)
	defer shutdown()
	writer := new(TWriter)
	reader := new(TReader)

	var body *walletStatusForm
	ensure := func(want string) {
		ensureResponse(t, s.apiToggleWalletStatus, want, reader, writer, body, nil)
	}

	body = &walletStatusForm{
		Disable: true,
		AssetID: 12,
	}

	ensure(`{"ok":true}`)
	if !tCore.walletDisabled {
		t.Fatal("Expected wallet to be disabled")
	}

	tCore.walletStatusErr = errors.New("wallet not found")
	ensure(`{"ok":false,"msg":"wallet not found"}`)

	tCore.walletDisabled = false
	body.Disable = false
	tCore.walletStatusErr = nil
	ensure(`{"ok":true}`)
	if tCore.walletDisabled {
		t.Fatal("Expected wallet to be enabled")
	}
}

func TestAPIDeleteArchivedRecords(t *testing.T) {
	s, tCore, shutdown := newTServer(t, false)
	defer shutdown()
	writer := new(TWriter)
	reader := new(TReader)

	var body *deleteRecordsForm
	ensure := func(want string) {
		ensureResponse(t, s.apiDeleteArchivedRecords, want, reader, writer, body, nil)
	}

	body = &deleteRecordsForm{
		OlderThanMs: time.Now().UnixMilli(),
	}

	tCore.deletedRecords = 23
	ensure(`{"ok":true,"archivedRecordsDeleted":23,"archivedRecordsPath":"/path/to/records"}`)

	tCore.deleteRecordsErr = tErr
	ensure(`{"ok":false,"msg":"expected dummy error"}`)
}

func TestAPITrade(t *testing.T) {
	testTrade(t, false)
}

func TestAPITradeAsync(t *testing.T) {
	testTrade(t, true)
}

func testTrade(t *testing.T, async bool) {
	s, tCore, shutdown := newTServer(t, false)
	defer shutdown()
	writer := new(TWriter)
	reader := new(TReader)

	body := &tradeForm{
		Pass:  []byte("random"),
		Order: new(core.TradeForm),
	}

	ensure := func(want string) {
		if async {
			ensureResponse(t, s.apiTradeAsync, want, reader, writer, body, nil)
		} else {
			ensureResponse(t, s.apiTrade, want, reader, writer, body, nil)
		}
	}

	tCore.tradeErr = tErr
	ensure(`{"ok":false,"msg":"expected dummy error"}`)
}

func Test_prepareAddr(t *testing.T) {
	tests := []struct {
		name       string
		addr       net.Addr
		allowInCSP bool
		want       string
	}{{
		name: "OK: IPv4",
		addr: &net.TCPAddr{
			IP:   net.IPv4(127, 0, 0, 1),
			Port: 7232,
		},
		want:       "127.0.0.1:7232",
		allowInCSP: true,
	}, {
		name: "IPv6 loopback",
		addr: &net.TCPAddr{
			IP:   net.IPv6loopback,
			Port: 7232,
		},
		want: "[::1]:7232",
	}, {
		name: "OK: IPv6 unspecified",
		addr: &net.TCPAddr{
			IP:   net.IPv6unspecified,
			Port: 7232,
		},
		want:       "127.0.0.1:7232",
		allowInCSP: true,
	}, {
		name: "OK: zero IPv4",
		addr: &net.TCPAddr{
			IP:   net.IPv4zero,
			Port: 7232,
		},
		want:       "127.0.0.1:7232",
		allowInCSP: true,
	}, {
		name: "others",
		addr: &net.UDPAddr{
			IP:   []byte{},
			Port: 7232,
		},
		want: ":7232",
	}}

	for _, test := range tests {
		gotAddr, allowInCSP := prepareAddr(test.addr)
		if gotAddr != test.want {
			t.Fatalf("%s: address: got %s, want %s", test.name, gotAddr, test.want)
		}
		if allowInCSP != test.allowInCSP {
			t.Fatalf("%s: allow in CSP: got %v, want %v", test.name, allowInCSP, test.allowInCSP)
		}
	}
}