DRY up storage diff repo tests

This commit is contained in:
Edvard
2019-02-20 14:37:55 +01:00
parent 5746de7c97
commit 56ca45509f
5 changed files with 173 additions and 345 deletions
@@ -19,6 +19,7 @@ package vow_test
import (
. "github.com/onsi/ginkgo"
. "github.com/onsi/gomega"
. "github.com/vulcanize/vulcanizedb/pkg/transformers/storage_diffs/maker/test_helpers"
"github.com/vulcanize/vulcanizedb/pkg/datastore/postgres"
"github.com/vulcanize/vulcanizedb/pkg/transformers/storage_diffs/maker/vow"
@@ -49,16 +50,11 @@ var _ = Describe("Vow storage repository test", func() {
err = repo.Create(blockNumber, blockHash, vow.VatMetadata, expectedVat)
Expect(err).NotTo(HaveOccurred())
type VowVat struct {
BlockMetadata
Vat string
}
var result VowVat
err = db.Get(&result, `SELECT block_number, block_hash, vat from maker.vow_vat`)
var result VariableRes
err = db.Get(&result, `SELECT block_number, block_hash, vat AS value from maker.vow_vat`)
Expect(err).NotTo(HaveOccurred())
Expect(result.BlockNumber).To(Equal(blockNumber))
Expect(result.BlockHash).To(Equal(blockHash))
Expect(result.Vat).To(Equal(expectedVat))
AssertVariable(result, blockNumber, blockHash, expectedVat)
})
It("persists a vow cow", func() {
@@ -67,16 +63,11 @@ var _ = Describe("Vow storage repository test", func() {
err = repo.Create(blockNumber, blockHash, vow.CowMetadata, expectedCow)
Expect(err).NotTo(HaveOccurred())
type VowCow struct {
BlockMetadata
Cow string
}
var result VowCow
err = db.Get(&result, `SELECT block_number, block_hash, cow from maker.vow_cow`)
var result VariableRes
err = db.Get(&result, `SELECT block_number, block_hash, cow AS value from maker.vow_cow`)
Expect(err).NotTo(HaveOccurred())
Expect(result.BlockNumber).To(Equal(blockNumber))
Expect(result.BlockHash).To(Equal(blockHash))
Expect(result.Cow).To(Equal(expectedCow))
AssertVariable(result, blockNumber, blockHash, expectedCow)
})
It("persists a vow row", func() {
@@ -85,16 +76,11 @@ var _ = Describe("Vow storage repository test", func() {
err = repo.Create(blockNumber, blockHash, vow.RowMetadata, expectedRow)
Expect(err).NotTo(HaveOccurred())
type VowRow struct {
BlockMetadata
Row string
}
var result VowRow
err = db.Get(&result, `SELECT block_number, block_hash, row from maker.vow_row`)
var result VariableRes
err = db.Get(&result, `SELECT block_number, block_hash, row AS value from maker.vow_row`)
Expect(err).NotTo(HaveOccurred())
Expect(result.BlockNumber).To(Equal(blockNumber))
Expect(result.BlockHash).To(Equal(blockHash))
Expect(result.Row).To(Equal(expectedRow))
AssertVariable(result, blockNumber, blockHash, expectedRow)
})
It("persists a vow Sin", func() {
@@ -103,16 +89,11 @@ var _ = Describe("Vow storage repository test", func() {
err = repo.Create(blockNumber, blockHash, vow.SinMetadata, expectedSow)
Expect(err).NotTo(HaveOccurred())
type VowSin struct {
BlockMetadata
Sin string
}
var result VowSin
err = db.Get(&result, `SELECT block_number, block_hash, sin from maker.vow_sin`)
var result VariableRes
err = db.Get(&result, `SELECT block_number, block_hash, sin AS value from maker.vow_sin`)
Expect(err).NotTo(HaveOccurred())
Expect(result.BlockNumber).To(Equal(blockNumber))
Expect(result.BlockHash).To(Equal(blockHash))
Expect(result.Sin).To(Equal(expectedSow))
AssertVariable(result, blockNumber, blockHash, expectedSow)
})
It("persists a vow woe", func() {
@@ -121,16 +102,11 @@ var _ = Describe("Vow storage repository test", func() {
err = repo.Create(blockNumber, blockHash, vow.WoeMetadata, expectedWoe)
Expect(err).NotTo(HaveOccurred())
type VowWoe struct {
BlockMetadata
Woe string
}
var result VowWoe
err = db.Get(&result, `SELECT block_number, block_hash, woe from maker.vow_woe`)
var result VariableRes
err = db.Get(&result, `SELECT block_number, block_hash, woe AS value from maker.vow_woe`)
Expect(err).NotTo(HaveOccurred())
Expect(result.BlockNumber).To(Equal(blockNumber))
Expect(result.BlockHash).To(Equal(blockHash))
Expect(result.Woe).To(Equal(expectedWoe))
AssertVariable(result, blockNumber, blockHash, expectedWoe)
})
It("persists a vow Ash", func() {
@@ -139,16 +115,11 @@ var _ = Describe("Vow storage repository test", func() {
err = repo.Create(blockNumber, blockHash, vow.AshMetadata, expectedAsh)
Expect(err).NotTo(HaveOccurred())
type VowAsh struct {
BlockMetadata
Ash string
}
var result VowAsh
err = db.Get(&result, `SELECT block_number, block_hash, ash from maker.vow_ash`)
var result VariableRes
err = db.Get(&result, `SELECT block_number, block_hash, ash AS value from maker.vow_ash`)
Expect(err).NotTo(HaveOccurred())
Expect(result.BlockNumber).To(Equal(blockNumber))
Expect(result.BlockHash).To(Equal(blockHash))
Expect(result.Ash).To(Equal(expectedAsh))
AssertVariable(result, blockNumber, blockHash, expectedAsh)
})
It("persists a vow Wait", func() {
@@ -157,16 +128,11 @@ var _ = Describe("Vow storage repository test", func() {
err = repo.Create(blockNumber, blockHash, vow.WaitMetadata, expectedWait)
Expect(err).NotTo(HaveOccurred())
type VowWait struct {
BlockMetadata
Wait string
}
var result VowWait
err = db.Get(&result, `SELECT block_number, block_hash, wait from maker.vow_wait`)
var result VariableRes
err = db.Get(&result, `SELECT block_number, block_hash, wait AS value from maker.vow_wait`)
Expect(err).NotTo(HaveOccurred())
Expect(result.BlockNumber).To(Equal(blockNumber))
Expect(result.BlockHash).To(Equal(blockHash))
Expect(result.Wait).To(Equal(expectedWait))
AssertVariable(result, blockNumber, blockHash, expectedWait)
})
It("persists a vow Bump", func() {
@@ -175,16 +141,11 @@ var _ = Describe("Vow storage repository test", func() {
err = repo.Create(blockNumber, blockHash, vow.BumpMetadata, expectedBump)
Expect(err).NotTo(HaveOccurred())
type VowBump struct {
BlockMetadata
Bump string
}
var result VowBump
err = db.Get(&result, `SELECT block_number, block_hash, bump from maker.vow_bump`)
var result VariableRes
err = db.Get(&result, `SELECT block_number, block_hash, bump AS value from maker.vow_bump`)
Expect(err).NotTo(HaveOccurred())
Expect(result.BlockNumber).To(Equal(blockNumber))
Expect(result.BlockHash).To(Equal(blockHash))
Expect(result.Bump).To(Equal(expectedBump))
AssertVariable(result, blockNumber, blockHash, expectedBump)
})
It("persists a vow Sump", func() {
@@ -193,16 +154,11 @@ var _ = Describe("Vow storage repository test", func() {
err = repo.Create(blockNumber, blockHash, vow.SumpMetadata, expectedSump)
Expect(err).NotTo(HaveOccurred())
type VowSump struct {
BlockMetadata
Sump string
}
var result VowSump
err = db.Get(&result, `SELECT block_number, block_hash, sump from maker.vow_sump`)
var result VariableRes
err = db.Get(&result, `SELECT block_number, block_hash, sump AS value from maker.vow_sump`)
Expect(err).NotTo(HaveOccurred())
Expect(result.BlockNumber).To(Equal(blockNumber))
Expect(result.BlockHash).To(Equal(blockHash))
Expect(result.Sump).To(Equal(expectedSump))
AssertVariable(result, blockNumber, blockHash, expectedSump)
})
It("persists a vow Hump", func() {
@@ -211,20 +167,10 @@ var _ = Describe("Vow storage repository test", func() {
err = repo.Create(blockNumber, blockHash, vow.HumpMetadata, expectedHump)
Expect(err).NotTo(HaveOccurred())
type VowHump struct {
BlockMetadata
Hump string
}
var result VowHump
err = db.Get(&result, `SELECT block_number, block_hash, hump from maker.vow_hump`)
var result VariableRes
err = db.Get(&result, `SELECT block_number, block_hash, hump AS value from maker.vow_hump`)
Expect(err).NotTo(HaveOccurred())
Expect(result.BlockNumber).To(Equal(blockNumber))
Expect(result.BlockHash).To(Equal(blockHash))
Expect(result.Hump).To(Equal(expectedHump))
AssertVariable(result, blockNumber, blockHash, expectedHump)
})
})
type BlockMetadata struct {
BlockNumber int `db:"block_number"`
BlockHash string `db:"block_hash"`
}