289 lines
8.0 KiB
Go
289 lines
8.0 KiB
Go
// VulcanizeDB
|
|
// Copyright © 2019 Vulcanize
|
|
|
|
// This program is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU Affero General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
|
|
// This program is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU Affero General Public License for more details.
|
|
|
|
// You should have received a copy of the GNU Affero General Public License
|
|
// along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
package btc_test
|
|
|
|
import (
|
|
"math/big"
|
|
|
|
"github.com/ethereum/go-ethereum/crypto"
|
|
. "github.com/onsi/ginkgo"
|
|
. "github.com/onsi/gomega"
|
|
|
|
"github.com/vulcanize/vulcanizedb/pkg/postgres"
|
|
"github.com/vulcanize/vulcanizedb/pkg/super_node/btc"
|
|
"github.com/vulcanize/vulcanizedb/pkg/super_node/shared"
|
|
)
|
|
|
|
var (
|
|
// Block 0
|
|
// header variables
|
|
blockHash1 = crypto.Keccak256Hash([]byte{00, 02})
|
|
blocKNumber1 = big.NewInt(0)
|
|
headerCid1 = "mockHeader1CID"
|
|
parentHash = crypto.Keccak256Hash([]byte{00, 01})
|
|
totalDifficulty = "50000000000000000000"
|
|
reward = "5000000000000000000"
|
|
headerModel1 = btc.HeaderModel{
|
|
BlockHash: blockHash1.String(),
|
|
BlockNumber: blocKNumber1.String(),
|
|
ParentHash: parentHash.String(),
|
|
CID: headerCid1,
|
|
}
|
|
|
|
// tx variables
|
|
tx1CID = "mockTx1CID"
|
|
tx2CID = "mockTx2CID"
|
|
tx1Hash = crypto.Keccak256Hash([]byte{01, 01})
|
|
tx2Hash = crypto.Keccak256Hash([]byte{01, 02})
|
|
opHash = crypto.Keccak256Hash([]byte{02, 01})
|
|
txModels1 = []btc.TxModelWithInsAndOuts{
|
|
{
|
|
Index: 0,
|
|
CID: tx1CID,
|
|
TxHash: tx1Hash.String(),
|
|
SegWit: true,
|
|
TxInputs: []btc.TxInput{
|
|
{
|
|
Index: 0,
|
|
TxWitness: []string{"mockWitness"},
|
|
SignatureScript: []byte{01},
|
|
PreviousOutPointIndex: 0,
|
|
PreviousOutPointHash: opHash.String(),
|
|
},
|
|
},
|
|
TxOutputs: []btc.TxOutput{
|
|
{
|
|
Index: 0,
|
|
Value: 50000000,
|
|
PkScript: []byte{02},
|
|
ScriptClass: 0,
|
|
RequiredSigs: 1,
|
|
},
|
|
},
|
|
},
|
|
{
|
|
Index: 1,
|
|
CID: tx2CID,
|
|
TxHash: tx2Hash.String(),
|
|
SegWit: true,
|
|
},
|
|
}
|
|
mockCIDPayload1 = &btc.CIDPayload{
|
|
HeaderCID: headerModel1,
|
|
TransactionCIDs: txModels1,
|
|
}
|
|
|
|
// Block 1
|
|
// header variables
|
|
blockHash2 = crypto.Keccak256Hash([]byte{00, 03})
|
|
blocKNumber2 = big.NewInt(1)
|
|
headerCid2 = "mockHeaderCID2"
|
|
headerModel2 = btc.HeaderModel{
|
|
BlockNumber: blocKNumber2.String(),
|
|
BlockHash: blockHash2.String(),
|
|
ParentHash: blockHash1.String(),
|
|
CID: headerCid2,
|
|
}
|
|
|
|
// tx variables
|
|
tx3CID = "mockTx3CID"
|
|
tx3Hash = crypto.Keccak256Hash([]byte{01, 03})
|
|
txModels2 = []btc.TxModelWithInsAndOuts{
|
|
{
|
|
Index: 0,
|
|
CID: tx3CID,
|
|
TxHash: tx3Hash.String(),
|
|
SegWit: true,
|
|
},
|
|
}
|
|
mockCIDPayload2 = &btc.CIDPayload{
|
|
HeaderCID: headerModel2,
|
|
TransactionCIDs: txModels2,
|
|
}
|
|
rngs = [][2]uint64{{0, 1}}
|
|
cids = []string{
|
|
headerCid1,
|
|
headerCid2,
|
|
tx1CID,
|
|
tx2CID,
|
|
tx3CID,
|
|
}
|
|
mockData = []byte{'\x01'}
|
|
)
|
|
|
|
var _ = Describe("Cleaner", func() {
|
|
var (
|
|
db *postgres.DB
|
|
repo *btc.CIDIndexer
|
|
cleaner *btc.Cleaner
|
|
)
|
|
BeforeEach(func() {
|
|
var err error
|
|
db, err = shared.SetupDB()
|
|
Expect(err).ToNot(HaveOccurred())
|
|
repo = btc.NewCIDIndexer(db)
|
|
cleaner = btc.NewCleaner(db)
|
|
})
|
|
|
|
Describe("Clean", func() {
|
|
BeforeEach(func() {
|
|
err := repo.Index(mockCIDPayload1)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
err = repo.Index(mockCIDPayload2)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
|
|
for _, cid := range cids {
|
|
_, err = db.Exec(`INSERT INTO public.blocks (key, data) VALUES ($1, $2)`, cid, mockData)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
}
|
|
|
|
tx, err := db.Beginx()
|
|
Expect(err).ToNot(HaveOccurred())
|
|
var startingIPFSBlocksCount int
|
|
pgStr := `SELECT COUNT(*) FROM public.blocks`
|
|
err = tx.Get(&startingIPFSBlocksCount, pgStr)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
var startingTxCount int
|
|
pgStr = `SELECT COUNT(*) FROM btc.transaction_cids`
|
|
err = tx.Get(&startingTxCount, pgStr)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
var startingHeaderCount int
|
|
pgStr = `SELECT COUNT(*) FROM btc.header_cids`
|
|
err = tx.Get(&startingHeaderCount, pgStr)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
|
|
err = tx.Commit()
|
|
Expect(err).ToNot(HaveOccurred())
|
|
|
|
Expect(startingIPFSBlocksCount).To(Equal(5))
|
|
Expect(startingTxCount).To(Equal(3))
|
|
Expect(startingHeaderCount).To(Equal(2))
|
|
})
|
|
AfterEach(func() {
|
|
btc.TearDownDB(db)
|
|
})
|
|
It("Cleans everything", func() {
|
|
err := cleaner.Clean(rngs, shared.Full)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
|
|
tx, err := db.Beginx()
|
|
Expect(err).ToNot(HaveOccurred())
|
|
var txCount int
|
|
pgStr := `SELECT COUNT(*) FROM btc.transaction_cids`
|
|
err = tx.Get(&txCount, pgStr)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
var txInCount int
|
|
pgStr = `SELECT COUNT(*) FROM btc.tx_inputs`
|
|
err = tx.Get(&txInCount, pgStr)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
var txOutCount int
|
|
pgStr = `SELECT COUNT(*) FROM btc.tx_outputs`
|
|
err = tx.Get(&txOutCount, pgStr)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
var headerCount int
|
|
pgStr = `SELECT COUNT(*) FROM btc.header_cids`
|
|
err = tx.Get(&headerCount, pgStr)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
var blocksCount int
|
|
pgStr = `SELECT COUNT(*) FROM public.blocks`
|
|
err = tx.Get(&blocksCount, pgStr)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
|
|
err = tx.Commit()
|
|
Expect(err).ToNot(HaveOccurred())
|
|
|
|
Expect(blocksCount).To(Equal(0))
|
|
Expect(txCount).To(Equal(0))
|
|
Expect(txInCount).To(Equal(0))
|
|
Expect(txOutCount).To(Equal(0))
|
|
Expect(headerCount).To(Equal(0))
|
|
})
|
|
It("Cleans headers and all linked data", func() {
|
|
err := cleaner.Clean(rngs, shared.Headers)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
|
|
tx, err := db.Beginx()
|
|
Expect(err).ToNot(HaveOccurred())
|
|
var txCount int
|
|
pgStr := `SELECT COUNT(*) FROM btc.transaction_cids`
|
|
err = tx.Get(&txCount, pgStr)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
var txInCount int
|
|
pgStr = `SELECT COUNT(*) FROM btc.tx_inputs`
|
|
err = tx.Get(&txInCount, pgStr)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
var txOutCount int
|
|
pgStr = `SELECT COUNT(*) FROM btc.tx_outputs`
|
|
err = tx.Get(&txOutCount, pgStr)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
var headerCount int
|
|
pgStr = `SELECT COUNT(*) FROM btc.header_cids`
|
|
err = tx.Get(&headerCount, pgStr)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
var blocksCount int
|
|
pgStr = `SELECT COUNT(*) FROM public.blocks`
|
|
err = tx.Get(&blocksCount, pgStr)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
|
|
err = tx.Commit()
|
|
Expect(err).ToNot(HaveOccurred())
|
|
|
|
Expect(blocksCount).To(Equal(0))
|
|
Expect(txCount).To(Equal(0))
|
|
Expect(txInCount).To(Equal(0))
|
|
Expect(txOutCount).To(Equal(0))
|
|
Expect(headerCount).To(Equal(0))
|
|
})
|
|
It("Cleans transactions", func() {
|
|
err := cleaner.Clean(rngs, shared.Transactions)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
|
|
tx, err := db.Beginx()
|
|
Expect(err).ToNot(HaveOccurred())
|
|
var txCount int
|
|
pgStr := `SELECT COUNT(*) FROM btc.transaction_cids`
|
|
err = tx.Get(&txCount, pgStr)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
var txInCount int
|
|
pgStr = `SELECT COUNT(*) FROM btc.tx_inputs`
|
|
err = tx.Get(&txInCount, pgStr)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
var txOutCount int
|
|
pgStr = `SELECT COUNT(*) FROM btc.tx_outputs`
|
|
err = tx.Get(&txOutCount, pgStr)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
var headerCount int
|
|
pgStr = `SELECT COUNT(*) FROM btc.header_cids`
|
|
err = tx.Get(&headerCount, pgStr)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
var blocksCount int
|
|
pgStr = `SELECT COUNT(*) FROM public.blocks`
|
|
err = tx.Get(&blocksCount, pgStr)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
|
|
err = tx.Commit()
|
|
Expect(err).ToNot(HaveOccurred())
|
|
|
|
Expect(blocksCount).To(Equal(2))
|
|
Expect(txCount).To(Equal(0))
|
|
Expect(txInCount).To(Equal(0))
|
|
Expect(txOutCount).To(Equal(0))
|
|
Expect(headerCount).To(Equal(2))
|
|
})
|
|
})
|
|
})
|