diff --git a/core/transaction_pool_test.go b/core/transaction_pool_test.go index 296c6bd8a..1214ad903 100644 --- a/core/transaction_pool_test.go +++ b/core/transaction_pool_test.go @@ -18,15 +18,6 @@ func (self stateQuery) GetAccount(addr []byte) *state.StateObject { return state.NewStateObject(addr) } -// State query interface -type invalidStateQuery struct{} - -func (self invalidStateQuery) GetAccount(addr []byte) *state.StateObject { - o := state.NewStateObject(addr) - o.Nonce++ - return o -} - func transaction() *types.Transaction { return types.NewTransactionMessage(make([]byte, 20), ethutil.Big0, ethutil.Big0, ethutil.Big0, nil) } @@ -72,11 +63,19 @@ func TestRemoveSet(t *testing.T) { } func TestRemoveInvalid(t *testing.T) { - pool, _ := setup() + pool, key := setup() tx1 := transaction() pool.pool.Add(tx1) - pool.RemoveInvalid(invalidStateQuery{}) + pool.RemoveInvalid(stateQuery{}) if pool.Size() > 0 { t.Error("expected pool size to be 0") } + + tx1.SetNonce(1) + tx1.SignECDSA(key) + pool.pool.Add(tx1) + pool.RemoveInvalid(stateQuery{}) + if pool.Size() != 1 { + t.Error("expected pool size to be 1, is", pool.Size()) + } }