added quota tree unit tests 1

Author: atantawi

pkg/quotaplugins/quota-forest/quota-manager/quota/core/allocation_test.go

@@ -185,9 +185,76 @@ func TestAllocation_SetValue(t *testing.T) {
 	}
 }
 
+func TestAllocation_Add(t *testing.T) {
+	type fields struct {
+		value1 []int
+		value2 []int
+	}
+	type args struct {
+		other *Allocation
+	}
+	tests := []struct {
+		name   string
+		fields fields
+		args   args
+		want   bool
+	}{
+		{
+			name: "test1",
+			fields: fields{
+				value1: []int{1, 2},
+				value2: []int{4, 6},
+			},
+			args: args{
+				other: &Allocation{
+					x: []int{3, 4},
+				},
+			},
+			want: true,
+		},
+		{
+			name: "test2",
+			fields: fields{
+				value1: []int{1, 2},
+				value2: []int{1, 2},
+			},
+			args: args{
+				other: &Allocation{
+					x: []int{3, 4, 5},
+				},
+			},
+			want: false,
+		},
+		{
+			name: "test3",
+			fields: fields{
+				value1: []int{1, 2},
+				value2: []int{4, 9},
+			},
+			args: args{
+				other: &Allocation{
+					x: []int{3, 4},
+				},
+			},
+			want: false,
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			a := &Allocation{
+				x: tt.fields.value1,
+			}
+			if got := a.Add(tt.args.other) && reflect.DeepEqual(a.x, tt.fields.value2); got != tt.want {
+				t.Errorf("Allocation.Add() = %v, want %v; result = %v, want %v",
+					got, tt.want, a.x, tt.fields.value2)
+			}
+		})
+	}
+}
+
 func TestAllocation_Fit(t *testing.T) {
 	type fields struct {
-		x []int
+		value []int
 	}
 	type args struct {
 		allocated *Allocation
@@ -201,7 +268,7 @@ func TestAllocation_Fit(t *testing.T) {
 	}{
 		{name: "test1",
 			fields: fields{
-				x: []int{1, 2, 3},
+				value: []int{1, 2, 3},
 			},
 			args: args{
 				allocated: &Allocation{
@@ -215,7 +282,7 @@ func TestAllocation_Fit(t *testing.T) {
 		},
 		{name: "test2",
 			fields: fields{
-				x: []int{1, 2, 3},
+				value: []int{1, 2, 3},
 			},
 			args: args{
 				allocated: &Allocation{
@@ -229,7 +296,7 @@ func TestAllocation_Fit(t *testing.T) {
 		},
 		{name: "test3",
 			fields: fields{
-				x: []int{1, 2, 3},
+				value: []int{1, 2, 3},
 			},
 			args: args{
 				allocated: &Allocation{
@@ -243,7 +310,7 @@ func TestAllocation_Fit(t *testing.T) {
 		},
 		{name: "test4",
 			fields: fields{
-				x: []int{1, 2, 3},
+				value: []int{1, 2, 3},
 			},
 			args: args{
 				allocated: &Allocation{
@@ -257,7 +324,7 @@ func TestAllocation_Fit(t *testing.T) {
 		},
 		{name: "test5",
 			fields: fields{
-				x: []int{1, 2, 3},
+				value: []int{1, 2, 3},
 			},
 			args: args{
 				allocated: &Allocation{
@@ -273,7 +340,7 @@ func TestAllocation_Fit(t *testing.T) {
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {
 			a := &Allocation{
-				x: tt.fields.x,
+				x: tt.fields.value,
 			}
 			if got := a.Fit(tt.args.allocated, tt.args.capacity); got != tt.want {
 				t.Errorf("Allocation.Fit() = %v, want %v", got, tt.want)
@@ -282,9 +349,125 @@ func TestAllocation_Fit(t *testing.T) {
 	}
 }
 
+func TestAllocation_IsZero(t *testing.T) {
+	type fields struct {
+		value []int
+	}
+	tests := []struct {
+		name   string
+		fields fields
+		want   bool
+	}{
+		{
+			name: "test1",
+			fields: fields{
+				value: []int{0, 0, 0},
+			},
+			want: true,
+		},
+		{
+			name: "test2",
+			fields: fields{
+				value: []int{0, 1},
+			},
+			want: false,
+		},
+		{
+			name: "test3",
+			fields: fields{
+				value: []int{1, 2, -4},
+			},
+			want: false,
+		},
+		{
+			name: "test4",
+			fields: fields{
+				value: []int{},
+			},
+			want: true,
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			a := &Allocation{
+				x: tt.fields.value,
+			}
+			if got := a.IsZero(); got != tt.want {
+				t.Errorf("Allocation.IsZero() = %v, want %v", got, tt.want)
+			}
+		})
+	}
+}
+
+func TestAllocation_Equal(t *testing.T) {
+	type fields struct {
+		value []int
+	}
+	type args struct {
+		other *Allocation
+	}
+	tests := []struct {
+		name   string
+		fields fields
+		args   args
+		want   bool
+	}{
+		{
+			name: "test1",
+			fields: fields{
+				value: []int{1, 2, 3},
+			},
+			args: args{
+				other: &Allocation{x: []int{1, 2, 3}},
+			},
+			want: true,
+		},
+		{
+			name: "test2",
+			fields: fields{
+				value: []int{},
+			},
+			args: args{
+				other: &Allocation{x: []int{}},
+			},
+			want: true,
+		},
+		{
+			name: "test3",
+			fields: fields{
+				value: []int{1, 2, 3},
+			},
+			args: args{
+				other: &Allocation{x: []int{4, 5, 6}},
+			},
+			want: false,
+		},
+		{
+			name: "test4",
+			fields: fields{
+				value: []int{1, 2},
+			},
+			args: args{
+				other: &Allocation{x: []int{1, 2, 3}},
+			},
+			want: false,
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			a := &Allocation{
+				x: tt.fields.value,
+			}
+			if got := a.Equal(tt.args.other); got != tt.want {
+				t.Errorf("Allocation.Equal() = %v, want %v", got, tt.want)
+			}
+		})
+	}
+}
+
 func TestAllocation_StringPretty(t *testing.T) {
 	type fields struct {
-		x []int
+		value []int
 	}
 	type args struct {
 		resourceNames []string
@@ -298,7 +481,7 @@ func TestAllocation_StringPretty(t *testing.T) {
 		{
 			name: "test1",
 			fields: fields{
-				x: []int{1, 2, 3},
+				value: []int{1, 2, 3},
 			},
 			args: args{
 				resourceNames: []string{"cpu", "memory", "gpu"},
@@ -308,7 +491,7 @@ func TestAllocation_StringPretty(t *testing.T) {
 		{
 			name: "test2",
 			fields: fields{
-				x: []int{1, 2, 3},
+				value: []int{1, 2, 3},
 			},
 			args: args{
 				resourceNames: []string{"cpu", "memory"},
@@ -320,7 +503,7 @@ func TestAllocation_StringPretty(t *testing.T) {
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {
 			a := &Allocation{
-				x: tt.fields.x,
+				x: tt.fields.value,
 			}
 			if got := a.StringPretty(tt.args.resourceNames); got != tt.want {
 				t.Errorf("Allocation.StringPretty() = %v, want %v", got, tt.want)