get nodebalancers by status in test

0xch4z · 0xch4z · commit db061c76a678 · 2021-01-28T09:28:56.000-05:00
diff --git a/cloud/linode/loadbalancers_test.go b/cloud/linode/loadbalancers_test.go
@@ -310,9 +310,9 @@ func testUpdateLoadBalancerAddAnnotation(t *testing.T, client *linodego.Client,
 		t.Errorf("UpdateLoadBalancer returned an error while updated annotations: %s", err)
 	}
 
-	nb, err := lb.getNodeBalancerByIPv4(context.TODO(), svc, lbStatus.Ingress[0].IP)
+	nb, err := lb.getNodeBalancerByStatus(context.TODO(), svc)
 	if err != nil {
-		t.Errorf("unexpected error: %v", err)
+		t.Fatalf("failed to get NodeBalancer via status: %s", err)
 	}
 
 	if nb.ClientConnThrottle != 10 {

Original file line number	Diff line number	Diff line change
`@@ -310,9 +310,9 @@ func testUpdateLoadBalancerAddAnnotation(t testing.T, client linodego.Client,`
`310`	`310`	`t.Errorf("UpdateLoadBalancer returned an error while updated annotations: %s", err)`
`311`	`311`	`}`
`312`	`312`
`313`		`- nb, err := lb.getNodeBalancerByIPv4(context.TODO(), svc, lbStatus.Ingress[0].IP)`
	`313`	`+ nb, err := lb.getNodeBalancerByStatus(context.TODO(), svc)`
`314`	`314`	`if err != nil {`
`315`		`- t.Errorf("unexpected error: %v", err)`
	`315`	`+ t.Fatalf("failed to get NodeBalancer via status: %s", err)`
`316`	`316`	`}`
`317`	`317`
`318`	`318`	`if nb.ClientConnThrottle != 10 {`