github.com/arieschain/arieschain@v0.0.0-20191023063405-37c074544356/core/helper_test.go (about)

     1  package core
     2  
     3  import (
     4  	"container/list"
     5  	"fmt"
     6  
     7  	"github.com/quickchainproject/quickchain/core/types"
     8  	"github.com/quickchainproject/quickchain/qctdb"
     9  	"github.com/quickchainproject/quickchain/event"
    10  )
    11  
    12  // Implement our EthTest Manager
    13  type TestManager struct {
    14  	// stateManager *StateManager
    15  	eventMux *event.TypeMux
    16  
    17  	db         qctdb.Database
    18  	txPool     *TxPool
    19  	blockChain *BlockChain
    20  	Blocks     []*types.Block
    21  }
    22  
    23  func (tm *TestManager) IsListening() bool {
    24  	return false
    25  }
    26  
    27  func (tm *TestManager) IsMining() bool {
    28  	return false
    29  }
    30  
    31  func (tm *TestManager) PeerCount() int {
    32  	return 0
    33  }
    34  
    35  func (tm *TestManager) Peers() *list.List {
    36  	return list.New()
    37  }
    38  
    39  func (tm *TestManager) BlockChain() *BlockChain {
    40  	return tm.blockChain
    41  }
    42  
    43  func (tm *TestManager) TxPool() *TxPool {
    44  	return tm.txPool
    45  }
    46  
    47  // func (tm *TestManager) StateManager() *StateManager {
    48  // 	return tm.stateManager
    49  // }
    50  
    51  func (tm *TestManager) EventMux() *event.TypeMux {
    52  	return tm.eventMux
    53  }
    54  
    55  // func (tm *TestManager) KeyManager() *crypto.KeyManager {
    56  // 	return nil
    57  // }
    58  
    59  func (tm *TestManager) Db() qctdb.Database {
    60  	return tm.db
    61  }
    62  
    63  func NewTestManager() *TestManager {
    64  	db, err := qctdb.NewMemDatabase()
    65  	if err != nil {
    66  		fmt.Println("Could not create mem-db, failing")
    67  		return nil
    68  	}
    69  
    70  	testManager := &TestManager{}
    71  	testManager.eventMux = new(event.TypeMux)
    72  	testManager.db = db
    73  	// testManager.txPool = NewTxPool(testManager)
    74  	// testManager.blockChain = NewBlockChain(testManager)
    75  	// testManager.stateManager = NewStateManager(testManager)
    76  
    77  	return testManager
    78  }