Remove duplication in repository specs
This commit is contained in:
Eric Meyer
parent
7131eafe99
commit
1bd56ec499
@ -1,141 +0,0 @@
|
|||||||
package repositories_test
|
|
||||||
|
|
||||||
import (
|
|
||||||
"github.com/8thlight/vulcanizedb/pkg/core"
|
|
||||||
"github.com/8thlight/vulcanizedb/pkg/repositories"
|
|
||||||
_ "github.com/lib/pq"
|
|
||||||
. "github.com/onsi/ginkgo"
|
|
||||||
. "github.com/onsi/gomega"
|
|
||||||
)
|
|
||||||
|
|
||||||
var _ = Describe("The In Memory Repository", func() {
|
|
||||||
|
|
||||||
Describe("Saving blocks", func() {
|
|
||||||
It("starts with no blocks", func() {
|
|
||||||
count := buildRepository().BlockCount()
|
|
||||||
Expect(count).Should(Equal(0))
|
|
||||||
})
|
|
||||||
|
|
||||||
It("increments the block count", func() {
|
|
||||||
block := core.Block{Number: 123}
|
|
||||||
repository := buildRepository()
|
|
||||||
|
|
||||||
repository.CreateBlock(block)
|
|
||||||
|
|
||||||
Expect(repository.BlockCount()).To(Equal(1))
|
|
||||||
})
|
|
||||||
|
|
||||||
It("saves the attributes of the block", func() {
|
|
||||||
blockNumber := int64(123)
|
|
||||||
gasLimit := int64(1000000)
|
|
||||||
gasUsed := int64(10)
|
|
||||||
blockHash := "x123"
|
|
||||||
blockParentHash := "x456"
|
|
||||||
blockNonce := "0x881db2ca900682e9a9"
|
|
||||||
blockTime := int64(1508981640)
|
|
||||||
uncleHash := "x789"
|
|
||||||
blockSize := int64(1000)
|
|
||||||
difficulty := int64(10)
|
|
||||||
block := core.Block{
|
|
||||||
Difficulty: difficulty,
|
|
||||||
GasLimit: gasLimit,
|
|
||||||
GasUsed: gasUsed,
|
|
||||||
Hash: blockHash,
|
|
||||||
Nonce: blockNonce,
|
|
||||||
Number: blockNumber,
|
|
||||||
ParentHash: blockParentHash,
|
|
||||||
Size: blockSize,
|
|
||||||
Time: blockTime,
|
|
||||||
UncleHash: uncleHash,
|
|
||||||
}
|
|
||||||
|
|
||||||
repository := buildRepository()
|
|
||||||
repository.CreateBlock(block)
|
|
||||||
|
|
||||||
savedBlock := repository.FindBlockByNumber(blockNumber)
|
|
||||||
Expect(savedBlock.Difficulty).To(Equal(difficulty))
|
|
||||||
Expect(savedBlock.GasLimit).To(Equal(gasLimit))
|
|
||||||
Expect(savedBlock.GasUsed).To(Equal(gasUsed))
|
|
||||||
Expect(savedBlock.Hash).To(Equal(blockHash))
|
|
||||||
Expect(savedBlock.Nonce).To(Equal(blockNonce))
|
|
||||||
Expect(savedBlock.Number).To(Equal(blockNumber))
|
|
||||||
Expect(savedBlock.ParentHash).To(Equal(blockParentHash))
|
|
||||||
Expect(savedBlock.Size).To(Equal(blockSize))
|
|
||||||
Expect(savedBlock.Time).To(Equal(blockTime))
|
|
||||||
Expect(savedBlock.UncleHash).To(Equal(uncleHash))
|
|
||||||
})
|
|
||||||
|
|
||||||
It("does not find a block when searching for a number that does not exist", func() {
|
|
||||||
repository := buildRepository()
|
|
||||||
|
|
||||||
savedBlock := repository.FindBlockByNumber(111)
|
|
||||||
|
|
||||||
Expect(savedBlock).To(BeNil())
|
|
||||||
})
|
|
||||||
|
|
||||||
It("saves one transaction associated to the block", func() {
|
|
||||||
block := core.Block{
|
|
||||||
Number: 123,
|
|
||||||
Transactions: []core.Transaction{{}},
|
|
||||||
}
|
|
||||||
repository := buildRepository()
|
|
||||||
|
|
||||||
repository.CreateBlock(block)
|
|
||||||
|
|
||||||
savedBlock := repository.FindBlockByNumber(123)
|
|
||||||
Expect(len(savedBlock.Transactions)).To(Equal(1))
|
|
||||||
})
|
|
||||||
|
|
||||||
It("saves two transactions associated to the block", func() {
|
|
||||||
block := core.Block{
|
|
||||||
Number: 123,
|
|
||||||
Transactions: []core.Transaction{{}, {}},
|
|
||||||
}
|
|
||||||
repository := buildRepository()
|
|
||||||
|
|
||||||
repository.CreateBlock(block)
|
|
||||||
|
|
||||||
savedBlock := repository.FindBlockByNumber(123)
|
|
||||||
Expect(len(savedBlock.Transactions)).To(Equal(2))
|
|
||||||
})
|
|
||||||
|
|
||||||
It("saves the attributes associated to a transaction", func() {
|
|
||||||
gasLimit := int64(5000)
|
|
||||||
gasPrice := int64(3)
|
|
||||||
nonce := uint64(10000)
|
|
||||||
to := "1234567890"
|
|
||||||
value := int64(10)
|
|
||||||
|
|
||||||
transaction := core.Transaction{
|
|
||||||
Hash: "x1234",
|
|
||||||
GasPrice: gasPrice,
|
|
||||||
GasLimit: gasLimit,
|
|
||||||
Nonce: nonce,
|
|
||||||
To: to,
|
|
||||||
Value: value,
|
|
||||||
}
|
|
||||||
block := core.Block{
|
|
||||||
Number: 123,
|
|
||||||
Transactions: []core.Transaction{transaction},
|
|
||||||
}
|
|
||||||
repository := buildRepository()
|
|
||||||
|
|
||||||
repository.CreateBlock(block)
|
|
||||||
|
|
||||||
savedBlock := repository.FindBlockByNumber(123)
|
|
||||||
Expect(len(savedBlock.Transactions)).To(Equal(1))
|
|
||||||
savedTransaction := savedBlock.Transactions[0]
|
|
||||||
Expect(savedTransaction.Hash).To(Equal(transaction.Hash))
|
|
||||||
Expect(savedTransaction.To).To(Equal(to))
|
|
||||||
Expect(savedTransaction.Nonce).To(Equal(nonce))
|
|
||||||
Expect(savedTransaction.GasLimit).To(Equal(gasLimit))
|
|
||||||
Expect(savedTransaction.GasPrice).To(Equal(gasPrice))
|
|
||||||
Expect(savedTransaction.Value).To(Equal(value))
|
|
||||||
})
|
|
||||||
})
|
|
||||||
|
|
||||||
})
|
|
||||||
|
|
||||||
func buildRepository() *repositories.InMemory {
|
|
||||||
return repositories.NewInMemory()
|
|
||||||
}
|
|
||||||
@ -1,159 +0,0 @@
|
|||||||
package repositories_test
|
|
||||||
|
|
||||||
import (
|
|
||||||
"github.com/8thlight/vulcanizedb/pkg/config"
|
|
||||||
"github.com/8thlight/vulcanizedb/pkg/core"
|
|
||||||
"github.com/8thlight/vulcanizedb/pkg/repositories"
|
|
||||||
"github.com/jmoiron/sqlx"
|
|
||||||
_ "github.com/lib/pq"
|
|
||||||
. "github.com/onsi/ginkgo"
|
|
||||||
. "github.com/onsi/gomega"
|
|
||||||
)
|
|
||||||
|
|
||||||
var _ = Describe("The Postgres Repository", func() {
|
|
||||||
|
|
||||||
var db *sqlx.DB
|
|
||||||
var err error
|
|
||||||
|
|
||||||
BeforeEach(func() {
|
|
||||||
pgConfig := config.DbConnectionString(config.NewConfig("private").Database)
|
|
||||||
db, err = sqlx.Connect("postgres", pgConfig)
|
|
||||||
db.MustExec("DELETE FROM transactions")
|
|
||||||
db.MustExec("DELETE FROM blocks")
|
|
||||||
})
|
|
||||||
|
|
||||||
AfterEach(func() {
|
|
||||||
db.Close()
|
|
||||||
})
|
|
||||||
|
|
||||||
It("connects to the database", func() {
|
|
||||||
Expect(err).Should(BeNil())
|
|
||||||
Expect(db).ShouldNot(BeNil())
|
|
||||||
})
|
|
||||||
|
|
||||||
Describe("Saving blocks", func() {
|
|
||||||
It("starts with no blocks", func() {
|
|
||||||
count := repositories.NewPostgres(db).BlockCount()
|
|
||||||
Expect(count).Should(Equal(0))
|
|
||||||
})
|
|
||||||
|
|
||||||
It("increments the block count", func() {
|
|
||||||
block := core.Block{Number: 123}
|
|
||||||
repository := repositories.NewPostgres(db)
|
|
||||||
|
|
||||||
repository.CreateBlock(block)
|
|
||||||
|
|
||||||
Expect(repository.BlockCount()).To(Equal(1))
|
|
||||||
})
|
|
||||||
|
|
||||||
It("saves the attributes of the block", func() {
|
|
||||||
blockNumber := int64(123)
|
|
||||||
gasLimit := int64(1000000)
|
|
||||||
gasUsed := int64(10)
|
|
||||||
blockHash := "x123"
|
|
||||||
blockParentHash := "x456"
|
|
||||||
blockNonce := "0x881db2ca900682e9a9"
|
|
||||||
blockTime := int64(1508981640)
|
|
||||||
uncleHash := "x789"
|
|
||||||
blockSize := int64(1000)
|
|
||||||
difficulty := int64(10)
|
|
||||||
block := core.Block{
|
|
||||||
Difficulty: difficulty,
|
|
||||||
GasLimit: gasLimit,
|
|
||||||
GasUsed: gasUsed,
|
|
||||||
Hash: blockHash,
|
|
||||||
Nonce: blockNonce,
|
|
||||||
Number: blockNumber,
|
|
||||||
ParentHash: blockParentHash,
|
|
||||||
Size: blockSize,
|
|
||||||
Time: blockTime,
|
|
||||||
UncleHash: uncleHash,
|
|
||||||
}
|
|
||||||
|
|
||||||
repository := repositories.NewPostgres(db)
|
|
||||||
|
|
||||||
repository.CreateBlock(block)
|
|
||||||
|
|
||||||
savedBlock := repository.FindBlockByNumber(blockNumber)
|
|
||||||
Expect(savedBlock.Difficulty).To(Equal(difficulty))
|
|
||||||
Expect(savedBlock.GasLimit).To(Equal(gasLimit))
|
|
||||||
Expect(savedBlock.GasUsed).To(Equal(gasUsed))
|
|
||||||
Expect(savedBlock.Hash).To(Equal(blockHash))
|
|
||||||
Expect(savedBlock.Nonce).To(Equal(blockNonce))
|
|
||||||
Expect(savedBlock.Number).To(Equal(blockNumber))
|
|
||||||
Expect(savedBlock.ParentHash).To(Equal(blockParentHash))
|
|
||||||
Expect(savedBlock.Size).To(Equal(blockSize))
|
|
||||||
Expect(savedBlock.Time).To(Equal(blockTime))
|
|
||||||
Expect(savedBlock.UncleHash).To(Equal(uncleHash))
|
|
||||||
})
|
|
||||||
|
|
||||||
It("does not find a block when searching for a number that does not exist", func() {
|
|
||||||
repository := repositories.NewPostgres(db)
|
|
||||||
|
|
||||||
savedBlock := repository.FindBlockByNumber(111)
|
|
||||||
|
|
||||||
Expect(savedBlock).To(BeNil())
|
|
||||||
})
|
|
||||||
|
|
||||||
It("saves one transaction associated to the block", func() {
|
|
||||||
block := core.Block{
|
|
||||||
Number: 123,
|
|
||||||
Transactions: []core.Transaction{{}},
|
|
||||||
}
|
|
||||||
repository := repositories.NewPostgres(db)
|
|
||||||
|
|
||||||
repository.CreateBlock(block)
|
|
||||||
|
|
||||||
savedBlock := repository.FindBlockByNumber(123)
|
|
||||||
Expect(len(savedBlock.Transactions)).To(Equal(1))
|
|
||||||
})
|
|
||||||
|
|
||||||
It("saves two transactions associated to the block", func() {
|
|
||||||
block := core.Block{
|
|
||||||
Number: 123,
|
|
||||||
Transactions: []core.Transaction{{}, {}},
|
|
||||||
}
|
|
||||||
repository := repositories.NewPostgres(db)
|
|
||||||
|
|
||||||
repository.CreateBlock(block)
|
|
||||||
|
|
||||||
savedBlock := repository.FindBlockByNumber(123)
|
|
||||||
Expect(len(savedBlock.Transactions)).To(Equal(2))
|
|
||||||
})
|
|
||||||
|
|
||||||
It("saves the attributes associated to a transaction", func() {
|
|
||||||
gasLimit := int64(5000)
|
|
||||||
gasPrice := int64(3)
|
|
||||||
nonce := uint64(10000)
|
|
||||||
to := "1234567890"
|
|
||||||
value := int64(10)
|
|
||||||
|
|
||||||
transaction := core.Transaction{
|
|
||||||
Hash: "x1234",
|
|
||||||
GasPrice: gasPrice,
|
|
||||||
GasLimit: gasLimit,
|
|
||||||
Nonce: nonce,
|
|
||||||
To: to,
|
|
||||||
Value: value,
|
|
||||||
}
|
|
||||||
block := core.Block{
|
|
||||||
Number: 123,
|
|
||||||
Transactions: []core.Transaction{transaction},
|
|
||||||
}
|
|
||||||
repository := repositories.NewPostgres(db)
|
|
||||||
|
|
||||||
repository.CreateBlock(block)
|
|
||||||
|
|
||||||
savedBlock := repository.FindBlockByNumber(123)
|
|
||||||
Expect(len(savedBlock.Transactions)).To(Equal(1))
|
|
||||||
savedTransaction := savedBlock.Transactions[0]
|
|
||||||
Expect(savedTransaction.Hash).To(Equal(transaction.Hash))
|
|
||||||
Expect(savedTransaction.To).To(Equal(to))
|
|
||||||
Expect(savedTransaction.Nonce).To(Equal(nonce))
|
|
||||||
Expect(savedTransaction.GasLimit).To(Equal(gasLimit))
|
|
||||||
Expect(savedTransaction.GasPrice).To(Equal(gasPrice))
|
|
||||||
Expect(savedTransaction.Value).To(Equal(value))
|
|
||||||
})
|
|
||||||
})
|
|
||||||
|
|
||||||
})
|
|
||||||
@ -4,4 +4,6 @@ import "github.com/8thlight/vulcanizedb/pkg/core"
|
|||||||
|
|
||||||
type Repository interface {
|
type Repository interface {
|
||||||
CreateBlock(block core.Block)
|
CreateBlock(block core.Block)
|
||||||
|
BlockCount() int
|
||||||
|
FindBlockByNumber(blockNumber int64) *core.Block
|
||||||
}
|
}
|
||||||
|
|||||||
162
pkg/repositories/repository_test.go
Normal file
162
pkg/repositories/repository_test.go
Normal file
@ -0,0 +1,162 @@
|
|||||||
|
package repositories_test
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/8thlight/vulcanizedb/pkg/config"
|
||||||
|
"github.com/8thlight/vulcanizedb/pkg/core"
|
||||||
|
"github.com/8thlight/vulcanizedb/pkg/repositories"
|
||||||
|
"github.com/jmoiron/sqlx"
|
||||||
|
_ "github.com/lib/pq"
|
||||||
|
. "github.com/onsi/ginkgo"
|
||||||
|
. "github.com/onsi/gomega"
|
||||||
|
)
|
||||||
|
|
||||||
|
var _ = Describe("Repositories", func() {
|
||||||
|
|
||||||
|
AssertRepositoryBehavior := func(buildRepository func() repositories.Repository) {
|
||||||
|
var repository repositories.Repository
|
||||||
|
|
||||||
|
BeforeEach(func() {
|
||||||
|
repository = buildRepository()
|
||||||
|
})
|
||||||
|
|
||||||
|
Describe("Saving blocks", func() {
|
||||||
|
It("starts with no blocks", func() {
|
||||||
|
count := repository.BlockCount()
|
||||||
|
Expect(count).Should(Equal(0))
|
||||||
|
})
|
||||||
|
|
||||||
|
It("increments the block count", func() {
|
||||||
|
block := core.Block{Number: 123}
|
||||||
|
|
||||||
|
repository.CreateBlock(block)
|
||||||
|
|
||||||
|
Expect(repository.BlockCount()).To(Equal(1))
|
||||||
|
})
|
||||||
|
|
||||||
|
It("saves the attributes of the block", func() {
|
||||||
|
blockNumber := int64(123)
|
||||||
|
gasLimit := int64(1000000)
|
||||||
|
gasUsed := int64(10)
|
||||||
|
blockHash := "x123"
|
||||||
|
blockParentHash := "x456"
|
||||||
|
blockNonce := "0x881db2ca900682e9a9"
|
||||||
|
blockTime := int64(1508981640)
|
||||||
|
uncleHash := "x789"
|
||||||
|
blockSize := int64(1000)
|
||||||
|
difficulty := int64(10)
|
||||||
|
block := core.Block{
|
||||||
|
Difficulty: difficulty,
|
||||||
|
GasLimit: gasLimit,
|
||||||
|
GasUsed: gasUsed,
|
||||||
|
Hash: blockHash,
|
||||||
|
Nonce: blockNonce,
|
||||||
|
Number: blockNumber,
|
||||||
|
ParentHash: blockParentHash,
|
||||||
|
Size: blockSize,
|
||||||
|
Time: blockTime,
|
||||||
|
UncleHash: uncleHash,
|
||||||
|
}
|
||||||
|
|
||||||
|
repository.CreateBlock(block)
|
||||||
|
|
||||||
|
savedBlock := repository.FindBlockByNumber(blockNumber)
|
||||||
|
Expect(savedBlock.Difficulty).To(Equal(difficulty))
|
||||||
|
Expect(savedBlock.GasLimit).To(Equal(gasLimit))
|
||||||
|
Expect(savedBlock.GasUsed).To(Equal(gasUsed))
|
||||||
|
Expect(savedBlock.Hash).To(Equal(blockHash))
|
||||||
|
Expect(savedBlock.Nonce).To(Equal(blockNonce))
|
||||||
|
Expect(savedBlock.Number).To(Equal(blockNumber))
|
||||||
|
Expect(savedBlock.ParentHash).To(Equal(blockParentHash))
|
||||||
|
Expect(savedBlock.Size).To(Equal(blockSize))
|
||||||
|
Expect(savedBlock.Time).To(Equal(blockTime))
|
||||||
|
Expect(savedBlock.UncleHash).To(Equal(uncleHash))
|
||||||
|
})
|
||||||
|
|
||||||
|
It("does not find a block when searching for a number that does not exist", func() {
|
||||||
|
savedBlock := repository.FindBlockByNumber(111)
|
||||||
|
|
||||||
|
Expect(savedBlock).To(BeNil())
|
||||||
|
})
|
||||||
|
|
||||||
|
It("saves one transaction associated to the block", func() {
|
||||||
|
block := core.Block{
|
||||||
|
Number: 123,
|
||||||
|
Transactions: []core.Transaction{{}},
|
||||||
|
}
|
||||||
|
|
||||||
|
repository.CreateBlock(block)
|
||||||
|
|
||||||
|
savedBlock := repository.FindBlockByNumber(123)
|
||||||
|
Expect(len(savedBlock.Transactions)).To(Equal(1))
|
||||||
|
})
|
||||||
|
|
||||||
|
It("saves two transactions associated to the block", func() {
|
||||||
|
block := core.Block{
|
||||||
|
Number: 123,
|
||||||
|
Transactions: []core.Transaction{{}, {}},
|
||||||
|
}
|
||||||
|
|
||||||
|
repository.CreateBlock(block)
|
||||||
|
|
||||||
|
savedBlock := repository.FindBlockByNumber(123)
|
||||||
|
Expect(len(savedBlock.Transactions)).To(Equal(2))
|
||||||
|
})
|
||||||
|
|
||||||
|
It("saves the attributes associated to a transaction", func() {
|
||||||
|
gasLimit := int64(5000)
|
||||||
|
gasPrice := int64(3)
|
||||||
|
nonce := uint64(10000)
|
||||||
|
to := "1234567890"
|
||||||
|
value := int64(10)
|
||||||
|
transaction := core.Transaction{
|
||||||
|
Hash: "x1234",
|
||||||
|
GasPrice: gasPrice,
|
||||||
|
GasLimit: gasLimit,
|
||||||
|
Nonce: nonce,
|
||||||
|
To: to,
|
||||||
|
Value: value,
|
||||||
|
}
|
||||||
|
block := core.Block{
|
||||||
|
Number: 123,
|
||||||
|
Transactions: []core.Transaction{transaction},
|
||||||
|
}
|
||||||
|
|
||||||
|
repository.CreateBlock(block)
|
||||||
|
|
||||||
|
savedBlock := repository.FindBlockByNumber(123)
|
||||||
|
Expect(len(savedBlock.Transactions)).To(Equal(1))
|
||||||
|
savedTransaction := savedBlock.Transactions[0]
|
||||||
|
Expect(savedTransaction.Hash).To(Equal(transaction.Hash))
|
||||||
|
Expect(savedTransaction.To).To(Equal(to))
|
||||||
|
Expect(savedTransaction.Nonce).To(Equal(nonce))
|
||||||
|
Expect(savedTransaction.GasLimit).To(Equal(gasLimit))
|
||||||
|
Expect(savedTransaction.GasPrice).To(Equal(gasPrice))
|
||||||
|
Expect(savedTransaction.Value).To(Equal(value))
|
||||||
|
})
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
Describe("In memory repository", func() {
|
||||||
|
AssertRepositoryBehavior(func() repositories.Repository {
|
||||||
|
return repositories.NewInMemory()
|
||||||
|
})
|
||||||
|
})
|
||||||
|
|
||||||
|
Describe("Postgres repository", func() {
|
||||||
|
It("connects to the database", func() {
|
||||||
|
pgConfig := config.DbConnectionString(config.NewConfig("private").Database)
|
||||||
|
db, err := sqlx.Connect("postgres", pgConfig)
|
||||||
|
Expect(err).Should(BeNil())
|
||||||
|
Expect(db).ShouldNot(BeNil())
|
||||||
|
})
|
||||||
|
|
||||||
|
AssertRepositoryBehavior(func() repositories.Repository {
|
||||||
|
pgConfig := config.DbConnectionString(config.NewConfig("private").Database)
|
||||||
|
db, _ := sqlx.Connect("postgres", pgConfig)
|
||||||
|
db.MustExec("DELETE FROM transactions")
|
||||||
|
db.MustExec("DELETE FROM blocks")
|
||||||
|
return repositories.NewPostgres(db)
|
||||||
|
})
|
||||||
|
})
|
||||||
|
|
||||||
|
})
|
||||||
Loading…
Reference in New Issue
Block a user