added method to check if the ID belongs to a group or container
This commit is contained in:
@@ -162,4 +162,15 @@ func GetContainerInformation(ID string) (*TrackContainer, error) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
return nil, errors.New("Container not found. ")
|
return nil, errors.New("Container not found. ")
|
||||||
|
}
|
||||||
|
|
||||||
|
// CheckID Checks if the ID belongs to a group or a single container
|
||||||
|
func CheckID(ID string) (string,error) {
|
||||||
|
// For group checks if the 1st characters is "grp"
|
||||||
|
if ID[0:3] == "grp" {
|
||||||
|
return "group", nil
|
||||||
|
} else {
|
||||||
|
return "container", nil
|
||||||
|
}
|
||||||
|
return "",nil
|
||||||
}
|
}
|
||||||
@@ -102,3 +102,19 @@ func TestRemoveTrackedContainer(t *testing.T) {
|
|||||||
t.Fail()
|
t.Fail()
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Test function that checks if the ID belongs to
|
||||||
|
// a group or container running
|
||||||
|
func TestCheckID(t *testing.T) {
|
||||||
|
id := "grp123"
|
||||||
|
checkID, err := CheckID(id)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println(err)
|
||||||
|
t.Fail()
|
||||||
|
}
|
||||||
|
if checkID == "group" {
|
||||||
|
fmt.Println("pass")
|
||||||
|
} else {
|
||||||
|
t.Fail()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user