Merge pull request #488 from forta-protocol/fix-unit-tests-after-cont…

…ainer-count-fix fFx unit tests (after container count check fix)
forta-network · Apr 14, 2022 · f93bfcd · f93bfcd
2 parents d3a468a + 7ce72cd
commit f93bfcd
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/services/supervisor/start_test.go b/services/supervisor/start_test.go
@@ -151,12 +151,12 @@ func (s *Suite) initialContainerCheck() {
  },
  }, nil)
 
- // 4 service containers + 1 old agent
- for i := 0; i < 5; i++ {
+ // service containers + 1 old agent
+ for i := 0; i < config.DockerSupervisorManagedContainers+1; i++ {
  s.dockerClient.EXPECT().RemoveContainer(s.service.ctx, testGenericContainerID).Return(nil)
  s.dockerClient.EXPECT().WaitContainerPrune(s.service.ctx, testGenericContainerID).Return(nil)
  }
- for i := 0; i < 5; i++ {
+ for i := 0; i < config.DockerSupervisorManagedContainers+1; i++ {
  s.dockerClient.EXPECT().RemoveNetworkByName(s.service.ctx, gomock.Any()).Return(nil)
  }
 }