Privatise DB repository field and add transformer test

This commit is contained in:
Edvard
2018-10-23 10:47:01 +02:00
parent cb74cbe990
commit 7ee3329baf
37 changed files with 182 additions and 138 deletions
@@ -31,6 +31,7 @@ type MockPitFileDebtCeilingRepository struct {
PassedEndingBlockNumber int64
PassedHeaderID int64
PassedModels []interface{}
SetDbCalled bool
}
func (repository *MockPitFileDebtCeilingRepository) MarkHeaderChecked(headerID int64) error {
@@ -70,4 +71,6 @@ func (repository *MockPitFileDebtCeilingRepository) AssertMarkHeaderCheckedCalle
Expect(repository.markHeaderCheckedPassedHeaderID).To(Equal(i))
}
func (repository *MockPitFileDebtCeilingRepository) SetDB(db *postgres.DB) {}
func (repository *MockPitFileDebtCeilingRepository) SetDB(db *postgres.DB) {
repository.SetDbCalled = true
}
@@ -31,6 +31,7 @@ type MockPitFileIlkRepository struct {
markHeaderCheckedPassedHeaderID int64
missingHeaders []core.Header
missingHeadersError error
SetDbCalled bool
}
func (repository *MockPitFileIlkRepository) SetCreateError(err error) {
@@ -70,4 +71,6 @@ func (repository *MockPitFileIlkRepository) AssertMarkHeaderCheckedCalledWith(i
Expect(repository.markHeaderCheckedPassedHeaderID).To(Equal(i))
}
func (repository *MockPitFileIlkRepository) SetDB(db *postgres.DB) {}
func (repository *MockPitFileIlkRepository) SetDB(db *postgres.DB) {
repository.SetDbCalled = true
}
@@ -31,6 +31,7 @@ type MockPitFileStabilityFeeRepository struct {
PassedEndingBlockNumber int64
PassedHeaderID int64
PassedModels []interface{}
SetDbCalled bool
}
func (repository *MockPitFileStabilityFeeRepository) Create(headerID int64, models []interface{}) error {
@@ -70,4 +71,6 @@ func (repository *MockPitFileStabilityFeeRepository) AssertMarkHeaderCheckedCall
Expect(repository.markHeaderCheckedPassedHeaderID).To(Equal(i))
}
func (repository *MockPitFileStabilityFeeRepository) SetDB(db *postgres.DB) {}
func (repository *MockPitFileStabilityFeeRepository) SetDB(db *postgres.DB) {
repository.SetDbCalled = true
}