diff --git a/staging/src/k8s.io/legacy-cloud-providers/azure/azure_loadbalancer.go b/staging/src/k8s.io/legacy-cloud-providers/azure/azure_loadbalancer.go index 973c1097aa5..de5f179f01e 100644 --- a/staging/src/k8s.io/legacy-cloud-providers/azure/azure_loadbalancer.go +++ b/staging/src/k8s.io/legacy-cloud-providers/azure/azure_loadbalancer.go @@ -1005,9 +1005,9 @@ func (az *Cloud) reconcileLoadBalancerRule( }) } - loadDistribution := network.Default + loadDistribution := network.LoadDistributionDefault if service.Spec.SessionAffinity == v1.ServiceAffinityClientIP { - loadDistribution = network.SourceIP + loadDistribution = network.LoadDistributionSourceIP } expectedRule := network.LoadBalancingRule{ diff --git a/staging/src/k8s.io/legacy-cloud-providers/azure/azure_loadbalancer_test.go b/staging/src/k8s.io/legacy-cloud-providers/azure/azure_loadbalancer_test.go index 07d622297ea..5b434e35231 100644 --- a/staging/src/k8s.io/legacy-cloud-providers/azure/azure_loadbalancer_test.go +++ b/staging/src/k8s.io/legacy-cloud-providers/azure/azure_loadbalancer_test.go @@ -176,14 +176,14 @@ func TestFindRule(t *testing.T) { { Name: to.StringPtr("probe1"), LoadBalancingRulePropertiesFormat: &network.LoadBalancingRulePropertiesFormat{ - LoadDistribution: network.SourceIP, + LoadDistribution: network.LoadDistributionSourceIP, }, }, }, curRule: network.LoadBalancingRule{ Name: to.StringPtr("probe2"), LoadBalancingRulePropertiesFormat: &network.LoadBalancingRulePropertiesFormat{ - LoadDistribution: network.SourceIP, + LoadDistribution: network.LoadDistributionSourceIP, }, }, expected: false, @@ -196,7 +196,7 @@ func TestFindRule(t *testing.T) { LoadBalancingRulePropertiesFormat: &network.LoadBalancingRulePropertiesFormat{ BackendPort: to.Int32Ptr(2), FrontendPort: to.Int32Ptr(2), - LoadDistribution: network.SourceIP, + LoadDistribution: network.LoadDistributionSourceIP, }, }, }, @@ -205,7 +205,7 @@ func TestFindRule(t *testing.T) { LoadBalancingRulePropertiesFormat: &network.LoadBalancingRulePropertiesFormat{ BackendPort: to.Int32Ptr(2), FrontendPort: to.Int32Ptr(2), - LoadDistribution: network.SourceIP, + LoadDistribution: network.LoadDistributionSourceIP, }, }, expected: true, diff --git a/staging/src/k8s.io/legacy-cloud-providers/azure/azure_test.go b/staging/src/k8s.io/legacy-cloud-providers/azure/azure_test.go index ee4d1d65b6f..8e402aee3d5 100644 --- a/staging/src/k8s.io/legacy-cloud-providers/azure/azure_test.go +++ b/staging/src/k8s.io/legacy-cloud-providers/azure/azure_test.go @@ -709,7 +709,7 @@ func TestServiceDefaultsToNoSessionPersistence(t *testing.T) { t.Error(err) } - if lbRule.LoadDistribution != network.Default { + if lbRule.LoadDistribution != network.LoadDistributionDefault { t.Errorf("Expected LB rule to have default load distribution but was %s", lbRule.LoadDistribution) } } @@ -732,7 +732,7 @@ func TestServiceRespectsNoSessionAffinity(t *testing.T) { t.Error(err) } - if lbRule.LoadDistribution != network.Default { + if lbRule.LoadDistribution != network.LoadDistributionDefault { t.Errorf("Expected LB rule to have default load distribution but was %s", lbRule.LoadDistribution) } } @@ -755,7 +755,7 @@ func TestServiceRespectsClientIPSessionAffinity(t *testing.T) { t.Error(err) } - if lbRule.LoadDistribution != network.SourceIP { + if lbRule.LoadDistribution != network.LoadDistributionSourceIP { t.Errorf("Expected LB rule to have SourceIP load distribution but was %s", lbRule.LoadDistribution) } }