added a couple of tests to gameserverallocationpolicy (googleforgames…

…#1503)

pkg/apis/multicluster/v1/gameserverallocationpolicy_test.go

@@ -256,6 +256,34 @@ func TestConnectionInfoIterator(t *testing.T) {
 			in:   nil,
 			want: nil,
 		},
+		{
+			name: "Same clusters and same priorities",
+			in: []*GameServerAllocationPolicy{
+				{
+					Spec: GameServerAllocationPolicySpec{
+						Priority: 1,
+						Weight:   100,
+						ConnectionInfo: ClusterConnectionInfo{
+							ClusterName: "cluster-name",
+						},
+					},
+				},
+				{
+					Spec: GameServerAllocationPolicySpec{
+						Priority: 1,
+						Weight:   300,
+						ConnectionInfo: ClusterConnectionInfo{
+							ClusterName: "cluster-name",
+						},
+					},
+				},
+			},
+			want: []ClusterConnectionInfo{
+				{
+					ClusterName: "cluster-name",
+				},
+			},
+		},
 	}
 	for _, tc := range testCases {
 		t.Run(tc.name, func(t *testing.T) {
@@ -277,3 +305,37 @@ func TestConnectionInfoIterator(t *testing.T) {
 		})
 	}
 }
+
+func TestConnectionInfoIterator_SameClustersAndPriorities(t *testing.T) {
+	in := []*GameServerAllocationPolicy{
+		{
+			Spec: GameServerAllocationPolicySpec{
+				Priority: 444,
+				Weight:   100,
+				ConnectionInfo: ClusterConnectionInfo{
+					ClusterName: "cluster-name",
+				},
+			},
+		},
+		{
+			Spec: GameServerAllocationPolicySpec{
+				Priority: 444,
+				Weight:   300,
+				ConnectionInfo: ClusterConnectionInfo{
+					ClusterName: "cluster-name",
+				},
+			},
+		},
+	}
+
+	iterator := NewConnectionInfoIterator(in)
+	res := iterator.priorityToCluster[444]["cluster-name"]
+
+	// check an internal slice of policies
+	if assert.Equal(t, 2, len(res)) {
+		assert.Equal(t, "cluster-name", res[0].Spec.ConnectionInfo.ClusterName)
+		assert.Equal(t, int32(444), res[0].Spec.Priority)
+		assert.Equal(t, "cluster-name", res[1].Spec.ConnectionInfo.ClusterName)
+		assert.Equal(t, int32(444), res[1].Spec.Priority)
+	}
+}