p2p-rendering-computation/client/GroupTrackContainer_test.go

package client

import (
	"fmt"
	"github.com/Akilan1999/p2p-rendering-computation/server/docker"
	"testing"
)

// Testing out if a new group is getting created
func TestCreateGroup(t *testing.T) {
	group, err := CreateGroup()
	if err != nil {
		fmt.Println(err)
		t.Fail()
	}
	PrettyPrint(group)
}

// Testing if the group gets removed when a
// group ID is provided
func TestRemoveGroup(t *testing.T) {
	// Creates a new group
	group, err := CreateGroup()
	if err != nil {
		fmt.Println(err)
		t.Fail()
	}
	// Removes the new group
	// it created
	err = RemoveGroup(group.ID)
	if err != nil {
		fmt.Println(err)
		t.Fail()
	}
}

// Testing if container information is added
// to the created group
func TestAddContainerToGroup(t *testing.T) {
	// Creates a new group
	group, err := CreateGroup()
	if err != nil {
		fmt.Println(err)
		t.Fail()
	}

	// Creating and adding the container to the
	// tracked list
	container1, err := docker.BuildRunContainer(0, "false", "")
	if err != nil {
		fmt.Println(err)
		t.Fail()
	}

	// Testing the AddTrackContainer Function and adding the first container created
	err = AddTrackContainer(container1, "0.0.0.0")
	if err != nil {
		// Killing docker container created
		err = docker.StopAndRemoveContainer(container1.ID)
		if err != nil {
			fmt.Println(err)
			t.Fail()
		}
		fmt.Println(err)
		t.Fail()
	}

	// Adds container information to the group
	Group, err := AddContainerToGroup(container1.ID, group.ID)
	if err != nil {
		fmt.Println(err)
		t.Fail()
	}

	PrettyPrint(Group)

	// Killing docker container created
	err = docker.StopAndRemoveContainer(container1.ID)
	if err != nil {
		fmt.Println(err)
		t.Fail()
	}

	// Removing container 1 from the tracked list
	err = RemoveTrackedContainer(container1.ID)
	if err != nil {
		fmt.Println(err)
		t.Fail()
	}

	// Removes the new group
	// it created
	err = RemoveGroup(group.ID)
	if err != nil {
		fmt.Println(err)
		t.Fail()
	}

}

// Testing if the container information is removed from the group
func TestGroup_RemoveContainerGroup(t *testing.T) {
	// Creates a new group
	group, err := CreateGroup()
	if err != nil {
		fmt.Println(err)
		t.Fail()
	}

	// Creating and adding the container to the
	// tracked list
	container1, err := docker.BuildRunContainer(0, "false", "")
	if err != nil {
		fmt.Println(err)
		t.Fail()
	}

	// Testing the AddTrackContainer Function and adding the first container created
	err = AddTrackContainer(container1, "0.0.0.0")
	if err != nil {
		// Killing docker container created
		err = docker.StopAndRemoveContainer(container1.ID)
		if err != nil {
			fmt.Println(err)
			t.Fail()
		}
		fmt.Println(err)
		t.Fail()
	}

	// Adds container information to the group
	Group, err := AddContainerToGroup(container1.ID, group.ID)
	if err != nil {
		fmt.Println(err)
		t.Fail()
	}
	fmt.Println("Container added")
	PrettyPrint(Group)

	// Removing docker container from the group
	Group, err = RemoveContainerGroup(container1.ID, group.ID)
	if err != nil {
		fmt.Println(err)
		t.Fail()
	}
	fmt.Println("Container removed")
	PrettyPrint(Group)

	// Killing docker container created
	err = docker.StopAndRemoveContainer(container1.ID)
	if err != nil {
		fmt.Println(err)
		t.Fail()
	}

	// Removing container 1 from the tracked list
	err = RemoveTrackedContainer(container1.ID)
	if err != nil {
		fmt.Println(err)
		t.Fail()
	}

	// Removes the new group
	// it created
	err = RemoveGroup(group.ID)
	if err != nil {
		fmt.Println(err)
		t.Fail()
	}
}

// Testing that container are removed from all
// created groups
// Scenario:
// - Create 2 groups
// - Add Container information to each group
// - Remove Container information from each group
func TestGroups_RemoveContainerGroups(t *testing.T) {
	// Creates a new group
	group, err := CreateGroup()
	if err != nil {
		fmt.Println(err)
		t.Fail()
	}
	// Created another group assigned to variable group 1
	group1, err := CreateGroup()
	if err != nil {
		fmt.Println(err)
		t.Fail()
	}

	// Creating and adding the container to the
	// tracked list
	container1, err := docker.BuildRunContainer(0, "false", "")
	if err != nil {
		fmt.Println(err)
		t.Fail()
	}

	// Testing the AddTrackContainer Function and adding the first container created
	err = AddTrackContainer(container1, "0.0.0.0")
	if err != nil {
		// Killing docker container created
		err = docker.StopAndRemoveContainer(container1.ID)
		if err != nil {
			fmt.Println(err)
			t.Fail()
		}
		fmt.Println(err)
		t.Fail()
	}

	// Adds container information to the group
	Group, err := AddContainerToGroup(container1.ID, group.ID)
	if err != nil {
		fmt.Println(err)
		t.Fail()
	}
	fmt.Println("Container added")
	PrettyPrint(Group)

	// Adds container information to the group
	Group1, err := AddContainerToGroup(container1.ID, group1.ID)
	if err != nil {
		fmt.Println(err)
		t.Fail()
	}
	fmt.Println("Container added")
	PrettyPrint(Group1)

	// Removing docker container from the group
	err = RemoveContainerGroups(container1.ID)
	if err != nil {
		fmt.Println(err)
		t.Fail()
	}

	// Killing docker container created
	err = docker.StopAndRemoveContainer(container1.ID)
	if err != nil {
		fmt.Println(err)
		t.Fail()
	}

	// Removing container 1 from the tracked list
	err = RemoveTrackedContainer(container1.ID)
	if err != nil {
		fmt.Println(err)
		t.Fail()
	}

	// Removes the new group
	// it created
	err = RemoveGroup(group.ID)
	if err != nil {
		fmt.Println(err)
		t.Fail()
	}

	// Removes the new group
	// it created
	err = RemoveGroup(group1.ID)
	if err != nil {
		fmt.Println(err)
		t.Fail()
	}
}