Merge pull request #201 from Bytom/v0.1

[bytom/vapor.git] / test / mock / mempool.go

diff --git a/test/mock/mempool.go b/test/mock/mempool.go
new file mode 100644 (file)
index 0000000..767fb7b
--- /dev/null
+++ b/test/mock/mempool.go
@@ -0,0 +1,24 @@
+package mock
+
+import (
+       "github.com/vapor/protocol"
+       "github.com/vapor/protocol/bc/types"
+)
+
+type Mempool struct {
+       txs []*protocol.TxDesc
+}
+
+func newMempool() *Mempool {
+       return &Mempool{
+               txs: []*protocol.TxDesc{},
+       }
+}
+
+func (m *Mempool) AddTx(tx *types.Tx) {
+       m.txs = append(m.txs, &protocol.TxDesc{Tx: tx})
+}
+
+func (m *Mempool) GetTransactions() []*protocol.TxDesc {
+       return m.txs
+}