Merge pull request #67750 from tizhou86/newUnitTest9

Automatic merge from submit-queue (batch tested with PRs 66257, 67750). If you want to cherry-pick this change to another branch, please follow the instructions <a href="https://github.com/kubernetes/community/blob/master/contributors/devel/cherry-picks.md">here</a>.

Add unit test cases for scheduler/util.

**What this PR does / why we need it**:
Add unit test cases for scheduler/util for more code coverage.

**Which issue(s) this PR fixes** *(optional, in `fixes #<issue number>(, fixes #<issue_number>, ...)` format, will close the issue(s) when PR gets merged)*:
NONE

**Special notes for your reviewer**:
NONE

**Release note**:

```release-note
NONE
```

pkg/scheduler/util/BUILD

@@ -17,6 +17,7 @@ go_test(
         "//pkg/apis/scheduling:go_default_library",
         "//staging/src/k8s.io/api/core/v1:go_default_library",
         "//staging/src/k8s.io/apimachinery/pkg/types:go_default_library",
+        "//staging/src/k8s.io/apimachinery/pkg/util/diff:go_default_library",
     ],
 )
 

pkg/scheduler/util/utils_test.go

@@ -17,9 +17,11 @@ limitations under the License.
 package util
 
 import (
+	"reflect"
 	"testing"
 
 	"k8s.io/api/core/v1"
+	"k8s.io/apimachinery/pkg/util/diff"
 	"k8s.io/kubernetes/pkg/apis/scheduling"
 )
 
@@ -303,3 +305,115 @@ func TestHostPortInfo_Check(t *testing.T) {
 		}
 	}
 }
+
+func TestGetContainerPorts(t *testing.T) {
+	tests := []struct {
+		pod1     *v1.Pod
+		pod2     *v1.Pod
+		expected []*v1.ContainerPort
+	}{
+		{
+			pod1: &v1.Pod{
+				Spec: v1.PodSpec{
+					Containers: []v1.Container{
+						{
+							Ports: []v1.ContainerPort{
+								{
+									ContainerPort: 8001,
+									Protocol:      v1.ProtocolTCP,
+								},
+								{
+									ContainerPort: 8002,
+									Protocol:      v1.ProtocolTCP,
+								},
+							},
+						},
+						{
+							Ports: []v1.ContainerPort{
+								{
+									ContainerPort: 8003,
+									Protocol:      v1.ProtocolTCP,
+								},
+								{
+									ContainerPort: 8004,
+									Protocol:      v1.ProtocolTCP,
+								},
+							},
+						},
+					},
+				},
+			},
+			pod2: &v1.Pod{
+				Spec: v1.PodSpec{
+					Containers: []v1.Container{
+						{
+							Ports: []v1.ContainerPort{
+								{
+									ContainerPort: 8011,
+									Protocol:      v1.ProtocolTCP,
+								},
+								{
+									ContainerPort: 8012,
+									Protocol:      v1.ProtocolTCP,
+								},
+							},
+						},
+						{
+							Ports: []v1.ContainerPort{
+								{
+									ContainerPort: 8013,
+									Protocol:      v1.ProtocolTCP,
+								},
+								{
+									ContainerPort: 8014,
+									Protocol:      v1.ProtocolTCP,
+								},
+							},
+						},
+					},
+				},
+			},
+			expected: []*v1.ContainerPort{
+				{
+					ContainerPort: 8001,
+					Protocol:      v1.ProtocolTCP,
+				},
+				{
+					ContainerPort: 8002,
+					Protocol:      v1.ProtocolTCP,
+				},
+				{
+					ContainerPort: 8003,
+					Protocol:      v1.ProtocolTCP,
+				},
+				{
+					ContainerPort: 8004,
+					Protocol:      v1.ProtocolTCP,
+				},
+				{
+					ContainerPort: 8011,
+					Protocol:      v1.ProtocolTCP,
+				},
+				{
+					ContainerPort: 8012,
+					Protocol:      v1.ProtocolTCP,
+				},
+				{
+					ContainerPort: 8013,
+					Protocol:      v1.ProtocolTCP,
+				},
+				{
+					ContainerPort: 8014,
+					Protocol:      v1.ProtocolTCP,
+				},
+			},
+		},
+	}
+
+	for _, test := range tests {
+		result := GetContainerPorts(test.pod1, test.pod2)
+		if !reflect.DeepEqual(test.expected, result) {
+			t.Errorf("Got different result than expected.\nDifference detected on:\n%s", diff.ObjectGoPrintSideBySide(test.expected, result))
+		}
+	}
+}