Fix fakes to support ipv4 filtering

Author: okokes-akamai

cloud/linode/fake_linode_test.go

@@ -32,11 +32,6 @@ type fakeRequest struct {
 	Method string
 }
 
-type filterStruct struct {
-	Label string `json:"label,omitempty"`
-	NbID  string `json:"nodebalancer_id,omitempty"`
-}
-
 func newFake(t *testing.T) *fakeAPI {
 	return &fakeAPI{
 		t:        t,
@@ -156,13 +151,13 @@ func (f *fakeAPI) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 						data = append(data, *n)
 					}
 				} else {
-					var fs filterStruct
+					var fs map[string]string
 					err := json.Unmarshal([]byte(filter), &fs)
 					if err != nil {
 						f.t.Fatal(err)
 					}
 					for _, n := range f.nbc {
-						if strconv.Itoa(n.NodeBalancerID) == fs.NbID {
+						if strconv.Itoa(n.NodeBalancerID) == fs["nodebalancer_id"] {
 							data = append(data, *n)
 						}
 					}
@@ -209,13 +204,14 @@ func (f *fakeAPI) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 						data = append(data, *n)
 					}
 				} else {
-					var fs filterStruct
+					var fs map[string]string
 					err := json.Unmarshal([]byte(filter), &fs)
 					if err != nil {
 						f.t.Fatal(err)
 					}
 					for _, n := range f.nb {
-						if *n.Label == fs.Label {
+						if (n.Label != nil && fs["label"] != "" && *n.Label == fs["label"]) ||
+							(fs["ipv4"] != "" && n.IPv4 != nil && *n.IPv4 == fs["ipv4"]) {
 							data = append(data, *n)
 						}
 					}

cloud/linode/loadbalancers_test.go

@@ -379,17 +379,17 @@ func testUpdateLoadBalancerAddPortAnnotation(t *testing.T, client *linodego.Clie
 
 	err = lb.UpdateLoadBalancer(context.TODO(), "linodelb", svc, nodes)
 	if err != nil {
-		t.Errorf("UpdateLoadBalancer returned an error while updated annotations: %s", err)
+		t.Fatalf("UpdateLoadBalancer returned an error while updated annotations: %s", err)
 	}
 
 	nb, err := lb.getNodeBalancerByStatus(context.TODO(), svc)
 	if err != nil {
-		t.Errorf("failed to get NodeBalancer by status: %v", err)
+		t.Fatalf("failed to get NodeBalancer by status: %v", err)
 	}
 
 	cfgs, errConfigs := client.ListNodeBalancerConfigs(context.TODO(), nb.ID, nil)
 	if errConfigs != nil {
-		t.Errorf("error getting NodeBalancer configs: %v", errConfigs)
+		t.Fatalf("error getting NodeBalancer configs: %v", errConfigs)
 	}
 
 	expectedPortConfigs := map[int]string{
@@ -468,17 +468,17 @@ func testUpdateLoadBalancerAddTLSPort(t *testing.T, client *linodego.Client, _ *
 	})
 	err = lb.UpdateLoadBalancer(context.TODO(), "linodelb", svc, nodes)
 	if err != nil {
-		t.Errorf("UpdateLoadBalancer returned an error while updated annotations: %s", err)
+		t.Fatalf("UpdateLoadBalancer returned an error while updated annotations: %s", err)
 	}
 
 	nb, err := lb.getNodeBalancerByStatus(context.TODO(), svc)
 	if err != nil {
-		t.Errorf("unexpected error: %v", err)
+		t.Fatalf("unexpected error: %v", err)
 	}
 
 	cfgs, errConfigs := client.ListNodeBalancerConfigs(context.TODO(), nb.ID, nil)
 	if errConfigs != nil {
-		t.Errorf("error getting NodeBalancer configs: %v", errConfigs)
+		t.Fatalf("error getting NodeBalancer configs: %v", errConfigs)
 	}
 
 	expectedPorts := map[int]struct{}{