[TO] Added unit tests for server folder (#7406)

* Added unit tests for server folder.

* Fix go vet issue.

* updated test case(insertProfile).

* updated test case(updateServer).

* Arranged import and checking the correct error.

traffic_ops/traffic_ops_golang/server/put_status_test.go

@@ -29,6 +29,26 @@ import (
 	"gopkg.in/DATA-DOG/go-sqlmock.v1"
 )
 
+func TestInvalidStatusForDeliveryServicesAlertText(t *testing.T) {
+	type testStruct struct {
+		dsId     []int
+		expected string
+	}
+
+	var testData = []testStruct{
+		{[]int{0}, " #0 with no 'ONLINE' or 'REPORTED' EDGE servers"},
+		{[]int{0, 1}, "s #0 and #1 with no 'ONLINE' or 'REPORTED' EDGE servers"},
+		{[]int{0, 1, 2}, "s #0, #1, and #2 with no 'ONLINE' or 'REPORTED' EDGE servers"},
+	}
+
+	for i, _ := range testData {
+		desc := InvalidStatusForDeliveryServicesAlertText("", "EDGE", testData[i].dsId)
+		if testData[i].expected != desc {
+			t.Errorf("strings don't't match, got:%s; expected:%s", desc, testData[i].expected)
+		}
+	}
+}
+
 func TestCheckExistingStatusInfo(t *testing.T) {
 	mockDB, mock, err := sqlmock.New()
 	if err != nil {

traffic_ops/traffic_ops_golang/server/servers_assignment.go

@@ -310,7 +310,7 @@ func ValidateDSCapabilities(dsIDs []int, serverName string, tx *sql.Tx) (error,
 		}
 	}
 
-	return nil, nil, 0
+	return nil, nil, http.StatusOK
 }
 
 func assignDeliveryServicesToServer(server int, dses []int, replace bool, tx *sql.Tx) ([]int, error) {

traffic_ops/traffic_ops_golang/server/servers_assignment_test.go

@@ -183,3 +183,36 @@ func TestCheckTenancyAndCDN(t *testing.T) {
 		t.Errorf("tenancy and cdn check for a given user failed with status code:%d", code)
 	}
 }
+
+func TestValidateDSCapabilities(t *testing.T) {
+	mockDB, mock, err := sqlmock.New()
+	if err != nil {
+		t.Fatalf("an error '%v' was not expected when opening a stub database connection", err)
+	}
+	defer mockDB.Close()
+
+	db := sqlx.NewDb(mockDB, "sqlmock")
+	defer db.Close()
+
+	mock.ExpectBegin()
+	rows := sqlmock.NewRows([]string{"server_capability"})
+	rows.AddRow([]byte("{eas}"))
+	mock.ExpectQuery("SELECT").WithArgs("eas").WillReturnRows(rows)
+
+	dsIDs := []int64{1}
+	rows1 := sqlmock.NewRows([]string{"id", "required_capabilities"})
+	rows1.AddRow(1, []byte("{eas}"))
+	mock.ExpectQuery("SELECT ").WithArgs(pq.Array(dsIDs)).WillReturnRows(rows1)
+	mock.ExpectCommit()
+
+	usrErr, sysErr, code := ValidateDSCapabilities([]int{1}, "eas", db.MustBegin().Tx)
+	if usrErr != nil {
+		t.Errorf("unable to validate DS capability, most likely cache doesn't have the DS capability, error: %v", usrErr)
+	}
+	if sysErr != nil {
+		t.Errorf("unable to validate DS Capability, system error: %v", sysErr)
+	}
+	if code != http.StatusOK {
+		t.Errorf("DS validation failed with status code:%d", code)
+	}
+}