github/kubernetes
1
0
Fork 0
mirror of https://github.com/k3s-io/kubernetes.git synced 2025-07-24 20:24:09 +00:00
Code Issues Projects Releases Wiki Activity

Move shared load balancer variables out of test.lb update/delete tests

Browse Source
This commit is contained in:
Ashley Gau 2018-01-31 17:51:01 -08:00
parent 6fedaa88ec
commit aeba01a287
1 changed files with 73 additions and 34 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

107
pkg/cloudprovider/providers/gce/gce_loadbalancer_external_test.go
View File

@ -242,27 +242,61 @@ func TestDeleteAddressWithWrongTier(t *testing.T) {
}
}
func TestEnsureExternalLoadBalancer(t *testing.T) {
const nodeName = "test-node-1"
const clusterName = "Test Cluster Name"
const clusterID = "test-cluster-id"
var apiService = &v1.Service{
Spec: v1.ServiceSpec{
SessionAffinity: v1.ServiceAffinityClientIP,
Type: v1.ServiceTypeClusterIP,
Ports: []v1.ServicePort{{Protocol: v1.ProtocolTCP, Port: int32(123)}},
},
}
var nodes = []*v1.Node{
&v1.Node{
ObjectMeta: metav1.ObjectMeta{
Name: nodeName,
Labels: map[string]string{
kubeletapis.LabelHostname: nodeName,
},
},
Status: v1.NodeStatus{
NodeInfo: v1.NodeSystemInfo{
KubeProxyVersion: "v1.7.2",
},
},
},
}
func fakeGCECloud() (*GCECloud, error) {
gceProjectId := "test-project"
gceRegion := "test-region"
zone := "zone1"
nodeName := "test-node-1"
client, clientErr := newOauthClient(nil)
require.NoError(t, clientErr)
client, err := newOauthClient(nil)
if err != nil {
return nil, err
}
service, serviceErr := compute.New(client)
require.NoError(t, serviceErr)
service, err := compute.New(client)
if err != nil {
return nil, err
}
fakeManager := newFakeManager(gceProjectId, gceRegion)
alphaFeatureGate, featureGateErr := NewAlphaFeatureGate([]string{})
require.NoError(t, featureGateErr)
alphaFeatureGate, err := NewAlphaFeatureGate([]string{})
if err != nil {
return nil, err
}
cloud := cloud.NewMockGCE()
zonesWithNodes := createNodeZones([]string{zone})
gce := GCECloud{
region: gceRegion,
service: service,
manager: fakeManager,
managedZones: []string{zone},
@ -284,35 +318,12 @@ func TestEnsureExternalLoadBalancer(t *testing.T) {
},
)
clusterName := "Test Cluster Name"
clusterID := "test-cluster-id"
apiService := &v1.Service{
Spec: v1.ServiceSpec{
SessionAffinity: v1.ServiceAffinityClientIP,
Type: v1.ServiceTypeClusterIP,
Ports: []v1.ServicePort{{Protocol: v1.ProtocolTCP, Port: int32(123)}},
},
}
return &gce, nil
}
func createExternalLoadBalancer(gce *GCECloud) (*v1.LoadBalancerStatus, error) {
existingFwdRule := &compute.ForwardingRule{}
nodes := []*v1.Node{
&v1.Node{
ObjectMeta: metav1.ObjectMeta{
Name: nodeName,
Labels: map[string]string{
kubeletapis.LabelHostname: nodeName,
},
},
Status: v1.NodeStatus{
NodeInfo: v1.NodeSystemInfo{
KubeProxyVersion: "v1.7.2",
},
},
},
}
status, err := gce.ensureExternalLoadBalancer(
clusterName,
clusterID,
@ -321,6 +332,34 @@ func TestEnsureExternalLoadBalancer(t *testing.T) {
nodes,
)
return status, err
}
func TestEnsureExternalLoadBalancer(t *testing.T) {
gce, err := fakeGCECloud()
assert.NoError(t, err)
status, err := createExternalLoadBalancer(gce)
assert.NotNil(t, status)
assert.NoError(t, err)
}
func TestUpdateExternalLoadBalancer(t *testing.T) {
gce, err := fakeGCECloud()
require.NoError(t, err)
createExternalLoadBalancer(gce)
err = gce.updateExternalLoadBalancer(clusterName, apiService, nodes)
assert.NoError(t, err)
// Assert pool is modified
}
func TestEnsureExternalLoadBalancerDeleted(t *testing.T) {
gce, err := fakeGCECloud()
require.NoError(t, err)
createExternalLoadBalancer(gce)
err = gce.ensureExternalLoadBalancerDeleted(clusterName, clusterID, apiService)
assert.NoError(t, err)
}