cerc-io/ipld-eth-server
17
0
Fork 1
Code Issues 10 Pull Requests 1 Actions Packages Projects Releases 37 Wiki Activity

changes to repo and repo tests so that supply table is organized and accessed by token_address

Browse Source
This commit is contained in:
Ian Norden 2018-08-12 15:35:23 -05:00
parent bea712ba10
commit 99e549b3df
3 changed files with 20 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
examples/erc20_watcher/every_block/repository.go
View File

@ -84,11 +84,13 @@ func (tsp *ERC20TokenRepository) MissingSupplyBlocks(startingBlock, highestBlock
&blockNumbers, &blockNumbers,
`SELECT number FROM BLOCKS `SELECT number FROM BLOCKS
LEFT JOIN token_supply ON blocks.id = block_id LEFT JOIN token_supply ON blocks.id = block_id
AND token_address = $1
WHERE block_id ISNULL WHERE block_id ISNULL
AND eth_node_id = $1 AND eth_node_id = $2
AND number >= $2 AND number >= $3
AND number <= $3 AND number <= $4
LIMIT 20`, LIMIT 20`,
tokenAddress,
tsp.NodeID, tsp.NodeID,
startingBlock, startingBlock,
highestBlock, highestBlock,

20
examples/erc20_watcher/every_block/repository_test.go
View File

@ -27,7 +27,7 @@ import (
"math/rand" "math/rand"
) )
var _ = Describe("ERC20 Token Repository", func() { var _ = Describe("ERC20 Token Supply Repository", func() {
var db *postgres.DB var db *postgres.DB
var blockId int64 var blockId int64
var blockNumber int64 var blockNumber int64
@ -136,7 +136,7 @@ var _ = Describe("ERC20 Token Repository", func() {
Describe("MissingBlocks", func() { Describe("MissingBlocks", func() {
It("returns the block numbers for which an associated TokenSupply record hasn't been created", func() { It("returns the block numbers for which an associated TokenSupply record hasn't been created", func() {
createTokenSupplyFor(repository, blockNumber) createTokenSupplyFor(repository, blockNumber, testAddress)
newBlockNumber := blockNumber + 1 newBlockNumber := blockNumber + 1
test_helpers.CreateBlock(newBlockNumber, blockRepository) test_helpers.CreateBlock(newBlockNumber, blockRepository)
@ -156,7 +156,7 @@ var _ = Describe("ERC20 Token Repository", func() {
}) })
It("does not return numbers that already have an associated TokenSupply record", func() { It("does not return numbers that already have an associated TokenSupply record", func() {
createTokenSupplyFor(repository, blockNumber) createTokenSupplyFor(repository, blockNumber, testAddress)
blocks, err := repository.MissingSupplyBlocks(blockNumber, blockNumber, testAddress) blocks, err := repository.MissingSupplyBlocks(blockNumber, blockNumber, testAddress)
Expect(blocks).To(BeEmpty()) Expect(blocks).To(BeEmpty())
@ -165,7 +165,7 @@ var _ = Describe("ERC20 Token Repository", func() {
}) })
It("deletes the token supply record when the associated block is deleted", func() { It("deletes the token supply record when the associated block is deleted", func() {
err := repository.CreateSupply(every_block.TokenSupply{BlockNumber: blockNumber, Value: "0"}) err := repository.CreateSupply(every_block.TokenSupply{BlockNumber: blockNumber, TokenAddress: testAddress, Value: "0"})
Expect(err).NotTo(HaveOccurred()) Expect(err).NotTo(HaveOccurred())
var count int var count int
@ -182,16 +182,20 @@ var _ = Describe("ERC20 Token Repository", func() {
}) })
}) })
func supplyModel(blockNumber int64, tokenAddress string, supplyValue string) every_block.TokenSupply { func supplyModel(blockNumber int64, tokenAddress, supplyValue string) every_block.TokenSupply {
return every_block.TokenSupply{ return every_block.TokenSupply{
Value: supplyValue, Value: supplyValue,
TokenAddress: tokenAddress, TokenAddress: tokenAddress,
BlockNumber: int64(blockNumber), BlockNumber: blockNumber,
} }
} }
func createTokenSupplyFor(repository every_block.ERC20TokenRepository, blockNumber int64) { func createTokenSupplyFor(repository every_block.ERC20TokenRepository, blockNumber int64, tokenAddress string) {
err := repository.CreateSupply(every_block.TokenSupply{BlockNumber: blockNumber, Value: "0"}) err := repository.CreateSupply(every_block.TokenSupply{
BlockNumber: blockNumber,
TokenAddress: tokenAddress,
Value: "0",
})
Expect(err).NotTo(HaveOccurred()) Expect(err).NotTo(HaveOccurred())
} }

4
examples/test_helpers/database.go
View File

@ -24,8 +24,8 @@ import (
) )
type TokenSupplyDBRow struct { type TokenSupplyDBRow struct {
ID int64 ID int64 `db:"id"`
Supply int64 Supply int64 `db:"supply"`
BlockID int64 `db:"block_id"` BlockID int64 `db:"block_id"`
TokenAddress string `db:"token_address"` TokenAddress string `db:"token_address"`
} }