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 MockTendRepository struct {
 	markHeaderCheckedPassedHeaderId int64
 	missingHeaders                  []core.Header
 	missingHeadersError             error
+	SetDbCalled                     bool
 }

 func (repository *MockTendRepository) Create(headerId int64, tend []interface{}) error {
@@ -70,4 +71,6 @@ func (repository *MockTendRepository) AssertMarkHeaderCheckedCalledWith(headerId
 	Expect(repository.markHeaderCheckedPassedHeaderId).To(Equal(headerId))
 }

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