test: use seelog in agent introspection validator func test

Author: haikuoliu

misc/agent-introspection-validator/agent-introspection-validator.go

@@ -20,6 +20,8 @@ import (
 	"net/http"
 	"os"
 	"time"
+
+	"github.com/cihub/seelog"
 )
 
 const (
@@ -76,17 +78,17 @@ func getTasksMetadata(client *http.Client, path string) (*TasksResponse, error)
 		return nil, err
 	}
 
-	fmt.Printf("Received tasks metadata: %s \n", string(body))
+	seelog.Infof("Received tasks metadata: %s \n", string(body))
 
-	var tasksMetadata TasksResponse
-	err = json.Unmarshal(body, &tasksMetadata)
+	err = verifyTasksMetadata(body)
 	if err != nil {
-		return nil, fmt.Errorf("%s: unable to parse response body: %v", tasksMetadataRespType, err)
+		return nil, fmt.Errorf("%s: unable to verify response: %v", tasksMetadataRespType, err)
 	}
 
-	err = verifyTasksMetadata(&tasksMetadata)
+	var tasksMetadata TasksResponse
+	err = json.Unmarshal(body, &tasksMetadata)
 	if err != nil {
-		return nil, fmt.Errorf("%s: unable to verify response: %v", tasksMetadataRespType, err)
+		return nil, fmt.Errorf("%s: unable to parse response body: %v", tasksMetadataRespType, err)
 	}
 
 	return &tasksMetadata, nil
@@ -98,90 +100,103 @@ func getTaskMetadata(client *http.Client, path string) (*TaskResponse, error) {
 		return nil, err
 	}
 
-	fmt.Printf("Received task metadata: %s \n", string(body))
+	seelog.Infof("Received task metadata: %s \n", string(body))
 
-	var taskMetadata TaskResponse
-	err = json.Unmarshal(body, &taskMetadata)
+	err = verifyTaskMetadata(body)
 	if err != nil {
-		return nil, fmt.Errorf("%s: unable to parse response body: %v", taskMetadataRespType, err)
+		return nil, fmt.Errorf("%s: unable to verify response: %v", taskMetadataRespType, err)
 	}
 
-	err = verifyTaskMetadata(&taskMetadata)
+	var taskMetadata TaskResponse
+	err = json.Unmarshal(body, &taskMetadata)
 	if err != nil {
-		return nil, fmt.Errorf("%s: unable to verify response: %v", taskMetadataRespType, err)
+		return nil, fmt.Errorf("%s: unable to parse response body: %v", taskMetadataRespType, err)
 	}
 
 	return &taskMetadata, nil
 }
 
 // verifyTasksMetadata verifies the number of tasks in tasks metadata.
-func verifyTasksMetadata(tasksMetadata *TasksResponse) error {
-	if len(tasksMetadata.Tasks) != 1 {
+func verifyTasksMetadata(tasksMetadataRawMsg json.RawMessage) error {
+	var tasksMetadataArray []json.RawMessage
+	json.Unmarshal(tasksMetadataRawMsg, &tasksMetadataArray)
+
+	if len(tasksMetadataArray) != 1 {
 		return fmt.Errorf("incorrect number of tasks, expected 1, received %d",
-			len(tasksMetadata.Tasks))
+			len(tasksMetadataArray))
 	}
 
-	return nil
+	return verifyTaskMetadata(tasksMetadataArray[0])
 }
 
 // verifyTaskMetadata verifies the number of containers in task metadata, make
 // sure each necessary field is not empty, we cannot check the values of those fields
 // since they are dynamic. It also verifies the container metadata.
-func verifyTaskMetadata(taskMetadata *TaskResponse) error {
-	if len(taskMetadata.Containers) != 1 {
-		return fmt.Errorf("incorrect number of containers, expected 1, received %d",
-			len(taskMetadata.Containers))
-	}
+func verifyTaskMetadata(taskMetadataRawMsg json.RawMessage) error {
+	taskMetadataMap := make(map[string]json.RawMessage)
+	json.Unmarshal(taskMetadataRawMsg, &taskMetadataMap)
 
-	if isEmpty(taskMetadata.Arn) {
+	if taskMetadataMap["Arn"] == nil {
 		return notEmptyErrMsg("Arn")
 	}
 
-	if isEmpty(taskMetadata.Version) {
-		return notEmptyErrMsg("Version")
+	if taskMetadataMap["DesiredStatus"] == nil {
+		return notEmptyErrMsg("DesiredStatus")
 	}
 
-	if isEmpty(taskMetadata.KnownStatus) {
+	if taskMetadataMap["KnownStatus"] == nil {
 		return notEmptyErrMsg("KnownStatus")
 	}
 
-	if isEmpty(taskMetadata.DesiredStatus) {
-		return notEmptyErrMsg("DesiredStatus")
+	if taskMetadataMap["Family"] == nil {
+		return notEmptyErrMsg("Family")
 	}
 
-	if isEmpty(taskMetadata.Family) {
-		return notEmptyErrMsg("Family")
+	if taskMetadataMap["Version"] == nil {
+		return notEmptyErrMsg("Version")
+	}
+
+	if taskMetadataMap["Containers"] == nil {
+		return notEmptyErrMsg("Containers")
+	}
+
+	var containersMetadataArray []json.RawMessage
+	json.Unmarshal(taskMetadataMap["Containers"], &containersMetadataArray)
+	if len(containersMetadataArray) != 1 {
+		return fmt.Errorf("incorrect number of containers, expected 1, received %d",
+			len(containersMetadataArray))
 	}
 
-	container := taskMetadata.Containers[0]
-	return verifyContainerMetadata(container)
+	return verifyContainerMetadata(containersMetadataArray[0])
 }
 
 // verifyContainerMetadata verifies the container name of the container metadata,
 // and each necessary field is not empty.
-func verifyContainerMetadata(containerMetadata ContainerResponse) error {
-	if containerMetadata.Name != containerName {
-		return fmt.Errorf("incorrect container name, expected %s, received %s",
-			containerName, containerMetadata.Name)
-	}
+func verifyContainerMetadata(containerMetadataRawMsg json.RawMessage) error {
+	containerMetadataMap := make(map[string]json.RawMessage)
+	json.Unmarshal(containerMetadataRawMsg, &containerMetadataMap)
 
-	if isEmpty(containerMetadata.DockerID) {
-		return notEmptyErrMsg("DockerID")
+	if containerMetadataMap["Name"] == nil {
+		return notEmptyErrMsg("Containers")
 	}
 
-	if isEmpty(containerMetadata.DockerName) {
-		return notEmptyErrMsg("DockerName")
+	var actualContainerName string
+	json.Unmarshal(containerMetadataMap["Name"] ,&actualContainerName)
+
+	if actualContainerName != containerName {
+		return fmt.Errorf("incorrect container name, expected %s, received %s",
+			containerName, actualContainerName)
 	}
 
-	return nil
-}
+	if containerMetadataMap["DockerID"] == nil {
+		return notEmptyErrMsg("Containers")
+	}
 
-func isEmpty(field string) bool {
-	if field == "" {
-		return true
+	if containerMetadataMap["DockerName"] == nil {
+		return notEmptyErrMsg("Containers")
 	}
 
-	return false
+	return nil
 }
 
 func notEmptyErrMsg(fieldName string) error {
@@ -196,7 +211,7 @@ func metadataResponse(client *http.Client, endpoint string, respType string) ([]
 		if err == nil {
 			return resp, nil
 		}
-		fmt.Fprintf(os.Stderr, "Attempt [%d/%d]: unable to get metadata response for '%s' from '%s': %v",
+		seelog.Infof(os.Stderr, "Attempt [%d/%d]: unable to get metadata response for '%s' from '%s': %v",
 			i, maxRetries, respType, endpoint, err)
 		time.Sleep(durationBetweenRetries)
 	}