@@ -130,9 +130,11 @@ func TestCommit(t *testing.T) {
130130 mockRPC := mocks .NewMockIRPCClient (t )
131131 mockMainStorage := mocks .NewMockIMainStorage (t )
132132 mockStagingStorage := mocks .NewMockIStagingStorage (t )
133+ mockOrchestratorStorage := mocks .NewMockIOrchestratorStorage (t )
133134 mockStorage := storage.IStorage {
134- MainStorage : mockMainStorage ,
135- StagingStorage : mockStagingStorage ,
135+ MainStorage : mockMainStorage ,
136+ StagingStorage : mockStagingStorage ,
137+ OrchestratorStorage : mockOrchestratorStorage ,
136138 }
137139 committer := NewCommitter (mockRPC , mockStorage )
138140
@@ -186,10 +188,12 @@ func TestStartCommitter(t *testing.T) {
186188 mockRPC := mocks .NewMockIRPCClient (t )
187189 mockMainStorage := mocks .NewMockIMainStorage (t )
188190 mockStagingStorage := mocks .NewMockIStagingStorage (t )
191+ mockOrchestratorStorage := mocks .NewMockIOrchestratorStorage (t )
189192
190193 mockStorage := storage.IStorage {
191- MainStorage : mockMainStorage ,
192- StagingStorage : mockStagingStorage ,
194+ MainStorage : mockMainStorage ,
195+ StagingStorage : mockStagingStorage ,
196+ OrchestratorStorage : mockOrchestratorStorage ,
193197 }
194198
195199 committer := NewCommitter (mockRPC , mockStorage )
@@ -218,9 +222,11 @@ func TestCommitterRespectsSIGTERM(t *testing.T) {
218222 mockRPC := mocks .NewMockIRPCClient (t )
219223 mockMainStorage := mocks .NewMockIMainStorage (t )
220224 mockStagingStorage := mocks .NewMockIStagingStorage (t )
225+ mockOrchestratorStorage := mocks .NewMockIOrchestratorStorage (t )
221226 mockStorage := storage.IStorage {
222- MainStorage : mockMainStorage ,
223- StagingStorage : mockStagingStorage ,
227+ MainStorage : mockMainStorage ,
228+ StagingStorage : mockStagingStorage ,
229+ OrchestratorStorage : mockOrchestratorStorage ,
224230 }
225231
226232 committer := NewCommitter (mockRPC , mockStorage )
0 commit comments