Add fix for non-default worker pool size

vmware-tanzu · May 3, 2023 · 8ec2304 · 8ec2304
1 parent a59e110
commit 8ec2304
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/pkg/migrate/foundation.go b/pkg/migrate/foundation.go
@@ -120,6 +120,7 @@ func NewFoundationMigratorFromConfig(c config.Config) (*FoundationMigrator, erro
  l.Debug("Creating foundation migrator")
  fm := NewFoundationMigrator(clientPool, boshClient, vmMigrator, srcAZsToClusters, out)
  fm.AdditionalVMs = ConfigToAdditionalVMs(c)
+ fm.WorkerCount = c.WorkerPoolSize
  return fm, nil
 }
 

diff --git a/pkg/migrate/foundation_test.go b/pkg/migrate/foundation_test.go
@@ -73,8 +73,14 @@ func baseConfig() config.Config {
 }
 
 func TestNewFoundationMigratorFromConfig(t *testing.T) {
- _, err := migrate.NewFoundationMigratorFromConfig(baseConfig())
+ fm, err := migrate.NewFoundationMigratorFromConfig(baseConfig())
  require.NoError(t, err)
+ require.NotNil(t, fm)
+ require.Equal(t, 1, fm.WorkerCount)
+ require.Equal(t, []bosh.VM{{
+ Name: "additional-vm1",
+ AZ: "az1",
+ }}, fm.AdditionalVMs)
 }
 
 func TestConfigToSourceClustersByAZ(t *testing.T) {