From ea8e7d4073f27e1ff83bb273da262c715d18aea0 Mon Sep 17 00:00:00 2001 From: xecarlox94 Date: Tue, 7 Jan 2025 20:19:47 +0000 Subject: [PATCH] uncommenting testing code --- client/GroupTrackContainer_test.go | 528 +++++++++++++-------------- client/TrackContianers_test.go | 234 ++++++------ client/clientIPTable/iptable_test.go | 22 +- config/generate/config_test.go | 108 +++--- config/generate/gernerate_test.go | 76 ++-- p2p/frp/server_test.go | 88 ++--- p2p/iptable_test.go | 151 ++++---- p2p/speedtest_test.go | 28 +- p2p/upnp_test.go | 26 +- server/gpu_test.go | 28 +- 10 files changed, 645 insertions(+), 644 deletions(-) diff --git a/client/GroupTrackContainer_test.go b/client/GroupTrackContainer_test.go index f53a433..8298ca4 100644 --- a/client/GroupTrackContainer_test.go +++ b/client/GroupTrackContainer_test.go @@ -1,267 +1,267 @@ package client -// import ( -// "fmt" -// "github.com/Akilan1999/p2p-rendering-computation/server/docker" -// "testing" -// ) +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() -// } -// } +// 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() + } +} diff --git a/client/TrackContianers_test.go b/client/TrackContianers_test.go index 8b1920b..3ec98a2 100644 --- a/client/TrackContianers_test.go +++ b/client/TrackContianers_test.go @@ -1,120 +1,120 @@ package client -// import ( -// "fmt" -// "github.com/Akilan1999/p2p-rendering-computation/server/docker" -// "testing" -// ) +import ( + "fmt" + "github.com/Akilan1999/p2p-rendering-computation/server/docker" + "testing" +) -// // Tests a scenario where the container are getting tracked -// func TestAddTrackContainer(t *testing.T) { -// // Create docker container and get SSH port -// container1, err := docker.BuildRunContainer(0, "false", "") -// if err != nil { -// fmt.Println(err) -// t.Fail() -// } -// // Testing the AddTrackContainer Function -// 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() -// } -// // Killing docker container created -// err = docker.StopAndRemoveContainer(container1.ID) -// if err != nil { -// fmt.Println(err) -// t.Fail() -// } -// } -// -// // Testing the remove container function -// // NOTE: This test can also be considered as a whole flow on the process of -// // tracked containers -// func TestRemoveTrackedContainer(t *testing.T) { -// container1, err := docker.BuildRunContainer(0, "false", "") -// if err != nil { -// fmt.Println(err) -// t.Fail() -// } -// -// container2, 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() -// } -// // Killing docker container created -// err = docker.StopAndRemoveContainer(container1.ID) -// if err != nil { -// fmt.Println(err) -// t.Fail() -// } -// -// // Testing the AddTrackContainer Function and the adding the second container created -// err = AddTrackContainer(container2, "0.0.0.0") -// if err != nil { -// // Killing docker container created -// err = docker.StopAndRemoveContainer(container2.ID) -// if err != nil { -// fmt.Println(err) -// t.Fail() -// } -// fmt.Println(err) -// t.Fail() -// } -// // Killing docker container created -// err = docker.StopAndRemoveContainer(container2.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() -// } -// -// // Removing container 2 from the tracked list -// err = RemoveTrackedContainer(container2.ID) -// if err != nil { -// fmt.Println(err) -// 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() -// } -// } +// Tests a scenario where the container are getting tracked +func TestAddTrackContainer(t *testing.T) { + // Create docker container and get SSH port + container1, err := docker.BuildRunContainer(0, "false", "") + if err != nil { + fmt.Println(err) + t.Fail() + } + // Testing the AddTrackContainer Function + 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() + } + // Killing docker container created + err = docker.StopAndRemoveContainer(container1.ID) + if err != nil { + fmt.Println(err) + t.Fail() + } +} + +// Testing the remove container function +// NOTE: This test can also be considered as a whole flow on the process of +// tracked containers +func TestRemoveTrackedContainer(t *testing.T) { + container1, err := docker.BuildRunContainer(0, "false", "") + if err != nil { + fmt.Println(err) + t.Fail() + } + + container2, 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() + } + // Killing docker container created + err = docker.StopAndRemoveContainer(container1.ID) + if err != nil { + fmt.Println(err) + t.Fail() + } + + // Testing the AddTrackContainer Function and the adding the second container created + err = AddTrackContainer(container2, "0.0.0.0") + if err != nil { + // Killing docker container created + err = docker.StopAndRemoveContainer(container2.ID) + if err != nil { + fmt.Println(err) + t.Fail() + } + fmt.Println(err) + t.Fail() + } + // Killing docker container created + err = docker.StopAndRemoveContainer(container2.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() + } + + // Removing container 2 from the tracked list + err = RemoveTrackedContainer(container2.ID) + if err != nil { + fmt.Println(err) + 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() + } +} diff --git a/client/clientIPTable/iptable_test.go b/client/clientIPTable/iptable_test.go index 97f9acb..53a755a 100644 --- a/client/clientIPTable/iptable_test.go +++ b/client/clientIPTable/iptable_test.go @@ -1,13 +1,13 @@ package clientIPTable -// import ( -// "testing" -// ) -// -// func TestUpdateIpTableListClient(t *testing.T) { -// err := UpdateIpTableListClient() -// -// if err != nil { -// t.Error(err) -// } -// } +import ( + "testing" +) + +func TestUpdateIpTableListClient(t *testing.T) { + err := UpdateIpTableListClient() + + if err != nil { + t.Error(err) + } +} diff --git a/config/generate/config_test.go b/config/generate/config_test.go index 28ad884..e3a0408 100644 --- a/config/generate/config_test.go +++ b/config/generate/config_test.go @@ -1,64 +1,64 @@ package generate -// import ( -// "fmt" -// "github.com/Akilan1999/p2p-rendering-computation/config" -// "os" -// "testing" -// ) -// -// // func TestConfigInit(t *testing.T) { -// // _, err := config.ConfigInit(nil) -// // if err != nil { -// // t.Error(err) -// // } -// // } -// -// // func TestSetDefaults(t *testing.T) { -// // _, err := SetDefaults("", false) -// // if err != nil { -// // t.Error(err) -// // } -// // } -// -// func TestGetCurrentPath(t *testing.T) { -// path, err := GetCurrentPath() +import ( + "fmt" + "github.com/Akilan1999/p2p-rendering-computation/config" + "os" + "testing" +) + +// func TestConfigInit(t *testing.T) { +// _, err := config.ConfigInit(nil) // if err != nil { -// fmt.Println(err) // t.Error(err) // } -// fmt.Println(path) // } -// -// func TestGetPathP2PRC(t *testing.T) { -// path, err := config.GetPathP2PRC("") + +// func TestSetDefaults(t *testing.T) { +// _, err := SetDefaults("", false) // if err != nil { -// fmt.Println(err) // t.Error(err) // } -// fmt.Println(path) -// } -// -// func TestSetEnvName(t *testing.T) { -// // Create an Env variable TEST with the value "lol" -// err := os.Setenv("TEST", "lol") -// if err != nil { -// fmt.Println(err) -// t.Error(err) -// } -// // Sets the environment variable as the default to read -// // for P2PRC -// err = config.SetEnvName("TEST") -// if err != nil { -// fmt.Println(err) -// t.Error(err) -// } -// -// // Checks if the output for the default read is "lol" -// path, err := config.GetPathP2PRC("") -// if err != nil { -// fmt.Println(err) -// t.Error(err) -// } -// fmt.Println(path) // } + +func TestGetCurrentPath(t *testing.T) { + path, err := GetCurrentPath() + if err != nil { + fmt.Println(err) + t.Error(err) + } + fmt.Println(path) +} + +func TestGetPathP2PRC(t *testing.T) { + path, err := config.GetPathP2PRC("") + if err != nil { + fmt.Println(err) + t.Error(err) + } + fmt.Println(path) +} + +func TestSetEnvName(t *testing.T) { + // Create an Env variable TEST with the value "lol" + err := os.Setenv("TEST", "lol") + if err != nil { + fmt.Println(err) + t.Error(err) + } + // Sets the environment variable as the default to read + // for P2PRC + err = config.SetEnvName("TEST") + if err != nil { + fmt.Println(err) + t.Error(err) + } + + // Checks if the output for the default read is "lol" + path, err := config.GetPathP2PRC("") + if err != nil { + fmt.Println(err) + t.Error(err) + } + fmt.Println(path) +} diff --git a/config/generate/gernerate_test.go b/config/generate/gernerate_test.go index f12cfb8..b48edf2 100644 --- a/config/generate/gernerate_test.go +++ b/config/generate/gernerate_test.go @@ -1,40 +1,40 @@ package generate -// import ( -// "fmt" -// "github.com/Akilan1999/p2p-rendering-computation/config" -// "testing" -// ) -// -// type CustomConfig struct { -// Test string -// } -// -// // Test case to generate defaults with custom data-structure -// func TestSetDefaults(t *testing.T) { -// setDefaults, err := SetDefaults("", true, &CustomConfig{Test: "lol"}, true) -// if err != nil { -// fmt.Println(err) -// t.Fail() -// return -// } -// -// fmt.Println(setDefaults) -// -// var c CustomConfig -// -// _, err = config.ConfigInit(nil, &c) -// if err != nil { -// fmt.Println(err) -// t.Fail() -// return -// } -// -// fmt.Println(c) -// -// } -// -// // Test case to generate public and private keys -// func TestGeneratePublicAndPrivateKeys(t *testing.T) { -// MakeSSHKeyPair("test.pub", "test.prv") -// } +import ( + "fmt" + "github.com/Akilan1999/p2p-rendering-computation/config" + "testing" +) + +type CustomConfig struct { + Test string +} + +// Test case to generate defaults with custom data-structure +func TestSetDefaults(t *testing.T) { + setDefaults, err := SetDefaults("", true, &CustomConfig{Test: "lol"}, true) + if err != nil { + fmt.Println(err) + t.Fail() + return + } + + fmt.Println(setDefaults) + + var c CustomConfig + + _, err = config.ConfigInit(nil, &c) + if err != nil { + fmt.Println(err) + t.Fail() + return + } + + fmt.Println(c) + +} + +// Test case to generate public and private keys +func TestGeneratePublicAndPrivateKeys(t *testing.T) { + MakeSSHKeyPair("test.pub", "test.prv") +} diff --git a/p2p/frp/server_test.go b/p2p/frp/server_test.go index c084ac3..04867d1 100644 --- a/p2p/frp/server_test.go +++ b/p2p/frp/server_test.go @@ -1,46 +1,46 @@ package frp -// import ( -// "fmt" -// "testing" -// "time" -// ) -// -// // Testing scenario FRPServer -// func TestStartFRPServer(t *testing.T) { -// var s Server -// s.address = "127.0.0.1" -// s.port = 8808 -// err := s.StartFRPServer() -// if err != nil { -// fmt.Println(err) -// t.Fail() -// } -// } -// -// // Testing scenario FRPServer and FRPClient connection -// func TestStartFRPClient(t *testing.T) { -// var s Server -// s.address = "127.0.0.1" -// s.port = 8808 -// go s.StartFRPServer() -// -// time.Sleep(3 * time.Second) -// -// // Sample test client -// var c Client -// c.Server = &s -// c.ClientMappings = []ClientMapping{ -// { -// LocalIP: "127.0.0.1", -// LocalPort: 22, -// RemotePort: 3301, -// }, -// } -// -// err := c.StartFRPClient() -// if err != nil { -// fmt.Println(err) -// t.Fail() -// } -// } +import ( + "fmt" + "testing" + "time" +) + +// Testing scenario FRPServer +func TestStartFRPServer(t *testing.T) { + var s Server + s.address = "127.0.0.1" + s.port = 8808 + err := s.StartFRPServer() + if err != nil { + fmt.Println(err) + t.Fail() + } +} + +// Testing scenario FRPServer and FRPClient connection +func TestStartFRPClient(t *testing.T) { + var s Server + s.address = "127.0.0.1" + s.port = 8808 + go s.StartFRPServer() + + time.Sleep(3 * time.Second) + + // Sample test client + var c Client + c.Server = &s + c.ClientMappings = []ClientMapping{ + { + LocalIP: "127.0.0.1", + LocalPort: 22, + RemotePort: 3301, + }, + } + + err := c.StartFRPClient() + if err != nil { + fmt.Println(err) + t.Fail() + } +} diff --git a/p2p/iptable_test.go b/p2p/iptable_test.go index d857fbb..7ca18ce 100644 --- a/p2p/iptable_test.go +++ b/p2p/iptable_test.go @@ -1,76 +1,77 @@ package p2p -// import ( -// "fmt" -// "testing" -// ) -// -// func TestReadIpTable(t *testing.T) { -// json, err := ReadIpTable() -// if err != nil { -// t.Fatal(err) -// } -// -// err = json.WriteIpTable() -// if err != nil { -// t.Fatal(err) -// } -// -// err = PrintIpTable() -// if err != nil { -// t.Fatal(err) -// } -// } -// -// // Testing is a IPV6 address is returned -// func TestGetCurrentIPV6(t *testing.T) { -// res, err := GetCurrentIPV6() -// -// if err != nil { -// t.Error(err) -// } -// -// fmt.Println(res) -// } -// -// // This test ensures that the duplicate function works as intended -// func TestIpAddresses_RemoveDuplicates(t *testing.T) { -// var testduplicates IpAddresses -// var duplicateaddress1 IpAddress -// var duplicateaddress2 IpAddress -// -// duplicateaddress1.Ipv6="2001:8f8:172d:ee93:7588:ad57:c351:3309" -// duplicateaddress1.Ipv4="0.0.0.0" -// -// duplicateaddress2.Ipv6="2001:8f8:172d:ee93:7588:ad57:c351:3309" -// duplicateaddress2.Ipv4="0.0.0.0" -// -// testduplicates.IpAddress = append(testduplicates.IpAddress, duplicateaddress1) -// testduplicates.IpAddress = append(testduplicates.IpAddress, duplicateaddress2) -// -// err := testduplicates.RemoveDuplicates() -// if err != nil { -// t.Error(err) -// } -// -// if len(testduplicates.IpAddress) == 2 { -// t.Fail() -// } -// -// } -// -// func TestViewNetworkInterface(t *testing.T) { -// err := ViewNetworkInterface() -// if err != nil { -// t.Error() -// } -// } -// -// func TestIp4or6(t *testing.T) { -// // This test ensures that the ipv6 address gets detected -// test := "2001:8f8:172d:7e27:4f23:ae4:bce5:e037" -// res := Ip4or6(test) -// if res != "version 6" { -// t.Fail() -// } -// -// } + +import ( + "fmt" + "testing" +) + +func TestReadIpTable(t *testing.T) { + json, err := ReadIpTable() + if err != nil { + t.Fatal(err) + } + + err = json.WriteIpTable() + if err != nil { + t.Fatal(err) + } + + err = PrintIpTable() + if err != nil { + t.Fatal(err) + } +} + +// Testing is a IPV6 address is returned +func TestGetCurrentIPV6(t *testing.T) { + res, err := GetCurrentIPV6() + + if err != nil { + t.Error(err) + } + + fmt.Println(res) +} + +// This test ensures that the duplicate function works as intended +func TestIpAddresses_RemoveDuplicates(t *testing.T) { + var testduplicates IpAddresses + var duplicateaddress1 IpAddress + var duplicateaddress2 IpAddress + + duplicateaddress1.Ipv6="2001:8f8:172d:ee93:7588:ad57:c351:3309" + duplicateaddress1.Ipv4="0.0.0.0" + + duplicateaddress2.Ipv6="2001:8f8:172d:ee93:7588:ad57:c351:3309" + duplicateaddress2.Ipv4="0.0.0.0" + + testduplicates.IpAddress = append(testduplicates.IpAddress, duplicateaddress1) + testduplicates.IpAddress = append(testduplicates.IpAddress, duplicateaddress2) + + err := testduplicates.RemoveDuplicates() + if err != nil { + t.Error(err) + } + + if len(testduplicates.IpAddress) == 2 { + t.Fail() + } + +} + +func TestViewNetworkInterface(t *testing.T) { + err := ViewNetworkInterface() + if err != nil { + t.Error() + } +} + +func TestIp4or6(t *testing.T) { + // This test ensures that the ipv6 address gets detected + test := "2001:8f8:172d:7e27:4f23:ae4:bce5:e037" + res := Ip4or6(test) + if res != "version 6" { + t.Fail() + } + +} diff --git a/p2p/speedtest_test.go b/p2p/speedtest_test.go index 3d2b55c..12bc2ef 100644 --- a/p2p/speedtest_test.go +++ b/p2p/speedtest_test.go @@ -1,16 +1,16 @@ package p2p -// import ( -// "testing" -// ) -// -// // To run this test ip_table.json must be populated -// func TestServer_SpeedTest(t *testing.T) { -// err := LocalSpeedTestIpTable() -// if err != nil { -// t.Fatal(err) -// } -// -// //HumaidTest("http://localhost:8088/50") -// //HumaidTest("http://ipv4.download.thinkbroadband.com/50MB.zip") -// } +import ( + "testing" +) + +// To run this test ip_table.json must be populated +func TestServer_SpeedTest(t *testing.T) { + err := LocalSpeedTestIpTable() + if err != nil { + t.Fatal(err) + } + + //HumaidTest("http://localhost:8088/50") + //HumaidTest("http://ipv4.download.thinkbroadband.com/50MB.zip") +} diff --git a/p2p/upnp_test.go b/p2p/upnp_test.go index 9e58490..6ac79cd 100644 --- a/p2p/upnp_test.go +++ b/p2p/upnp_test.go @@ -1,15 +1,15 @@ package p2p -// import ( -// "fmt" -// "testing" -// ) -// -// // Tests if the current has UPNP support -// func TestForwardUPNPPort(t *testing.T) { -// err := ForwardPort(6586) -// if err != nil { -// fmt.Println(err) -// t.Fail() -// } -// } +import ( + "fmt" + "testing" +) + +// Tests if the current has UPNP support +func TestForwardUPNPPort(t *testing.T) { + err := ForwardPort(6586) + if err != nil { + fmt.Println(err) + t.Fail() + } +} diff --git a/server/gpu_test.go b/server/gpu_test.go index 9a839ca..78f0a7b 100644 --- a/server/gpu_test.go +++ b/server/gpu_test.go @@ -1,16 +1,16 @@ package server -// import ( -// "fmt" -// "testing" -// ) -// -// func TestGpuOutput(t *testing.T) { -// gpu, err := GPUInfo() -// -// if err != nil { -// t.Error(err) -// } -// -// fmt.Print(gpu.Gpu.GpuName) -// } +import ( + "fmt" + "testing" +) + +func TestGpuOutput(t *testing.T) { + gpu, err := GPUInfo() + + if err != nil { + t.Error(err) + } + + fmt.Print(gpu.Gpu.GpuName) +}