diff --git a/client/TrackContainers.go b/client/TrackContainers.go index 389675d..b61d7f2 100644 --- a/client/TrackContainers.go +++ b/client/TrackContainers.go @@ -162,4 +162,15 @@ func GetContainerInformation(ID string) (*TrackContainer, error) { } } 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 } \ No newline at end of file diff --git a/client/TrackContianers_test.go b/client/TrackContianers_test.go index 0f1f7c4..ec8a90d 100644 --- a/client/TrackContianers_test.go +++ b/client/TrackContianers_test.go @@ -102,3 +102,19 @@ func TestRemoveTrackedContainer(t *testing.T) { 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() + } +}