forked from cerc-io/ipld-eth-server
Privatise DB repository field and add transformer test
This commit is contained in:
@@ -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
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user