proto/p/testcontext/context.go

// Copyright (C) 2019-2025, Lux Industries, Inc. All rights reserved.
// See the file LICENSE for licensing terms.

// Package testcontext provides a test context for platformvm tests
package testcontext

import (
	"context"
	"sync"

	"github.com/luxfi/ids"
	log "github.com/luxfi/log"
	consensuscontext "github.com/luxfi/runtime"
	"github.com/luxfi/vm/chains/atomic"
)

// Context provides a test context that mimics the old context.Context
// for compatibility with existing tests
type Context struct {
	context.Context
	NetworkID      uint32 // 1=mainnet, 2=testnet
	ChainID        ids.ID
	NodeID         ids.NodeID
	PublicKey      interface{} // BLS public key
	XChainID       ids.ID
	CChainID       ids.ID
	DChainID       ids.ID
	UTXOAssetID    ids.ID // Primary asset ID (X-chain native)
	ValidatorState interface{}
	WarpSigner     interface{}
	Log            log.Logger
	Lock           *sync.RWMutex
	SharedMemory   atomic.SharedMemory
	BCLookup       ids.AliaserReader
	ChainDataDir   string
	Keystore       interface{}
	Signer         interface{}
}

// New creates a new test context
func New(ctx context.Context) *Context {
	return &Context{
		Context: ctx,
		Lock:    &sync.RWMutex{},
		Log:     log.NoLog{},
	}
}

// WithIDs sets the IDs from consensuscontext.IDs
func (c *Context) WithIDs(ids consensuscontext.IDs) *Context {
	c.NetworkID = ids.NetworkID
	c.ChainID = ids.ChainID
	c.NodeID = ids.NodeID
	c.UTXOAssetID = ids.UTXOAssetID
	return c
}