Privatise DB repository field and add transformer test

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 MockVatInitRepository struct {
 	PassedEndingBlockNumber         int64
 	PassedHeaderID                  int64
 	PassedModels                    []interface{}
+	SetDbCalled                     bool
 }

 func (repository *MockVatInitRepository) Create(headerID int64, models []interface{}) error {
@@ -70,4 +71,6 @@ func (repository *MockVatInitRepository) AssertMarkHeaderCheckedCalledWith(i int
 	Expect(repository.markHeaderCheckedPassedHeaderID).To(Equal(i))
 }

-func (repository *MockVatInitRepository) SetDB(db *postgres.DB) {}
+func (repository *MockVatInitRepository) SetDB(db *postgres.DB) {
+	repository.SetDbCalled = true
+}