@@ -130,9 +130,11 @@ func TestCommit(t *testing.T) {
130
130
mockRPC := mocks .NewMockIRPCClient (t )
131
131
mockMainStorage := mocks .NewMockIMainStorage (t )
132
132
mockStagingStorage := mocks .NewMockIStagingStorage (t )
133
+ mockOrchestratorStorage := mocks .NewMockIOrchestratorStorage (t )
133
134
mockStorage := storage.IStorage {
134
- MainStorage : mockMainStorage ,
135
- StagingStorage : mockStagingStorage ,
135
+ MainStorage : mockMainStorage ,
136
+ StagingStorage : mockStagingStorage ,
137
+ OrchestratorStorage : mockOrchestratorStorage ,
136
138
}
137
139
committer := NewCommitter (mockRPC , mockStorage )
138
140
@@ -186,10 +188,12 @@ func TestStartCommitter(t *testing.T) {
186
188
mockRPC := mocks .NewMockIRPCClient (t )
187
189
mockMainStorage := mocks .NewMockIMainStorage (t )
188
190
mockStagingStorage := mocks .NewMockIStagingStorage (t )
191
+ mockOrchestratorStorage := mocks .NewMockIOrchestratorStorage (t )
189
192
190
193
mockStorage := storage.IStorage {
191
- MainStorage : mockMainStorage ,
192
- StagingStorage : mockStagingStorage ,
194
+ MainStorage : mockMainStorage ,
195
+ StagingStorage : mockStagingStorage ,
196
+ OrchestratorStorage : mockOrchestratorStorage ,
193
197
}
194
198
195
199
committer := NewCommitter (mockRPC , mockStorage )
@@ -218,9 +222,11 @@ func TestCommitterRespectsSIGTERM(t *testing.T) {
218
222
mockRPC := mocks .NewMockIRPCClient (t )
219
223
mockMainStorage := mocks .NewMockIMainStorage (t )
220
224
mockStagingStorage := mocks .NewMockIStagingStorage (t )
225
+ mockOrchestratorStorage := mocks .NewMockIOrchestratorStorage (t )
221
226
mockStorage := storage.IStorage {
222
- MainStorage : mockMainStorage ,
223
- StagingStorage : mockStagingStorage ,
227
+ MainStorage : mockMainStorage ,
228
+ StagingStorage : mockStagingStorage ,
229
+ OrchestratorStorage : mockOrchestratorStorage ,
224
230
}
225
231
226
232
committer := NewCommitter (mockRPC , mockStorage )
0 commit comments