Add the cleaning to the tests

Author: thewizardplusplus

check_imports_test.go

@@ -1,6 +1,8 @@
 package coderunner
 
 import (
+	"os"
+	"path/filepath"
 	"testing"
 
 	mapset "github.com/deckarep/golang-set"
@@ -80,6 +82,7 @@ func TestCheckImports(test *testing.T) {
 		test.Run(data.name, func(test *testing.T) {
 			pathToCode, err := systemutils.SaveTemporaryText(data.args.code, ".go")
 			require.NoError(test, err)
+			defer os.RemoveAll(filepath.Dir(pathToCode)) // nolint: errcheck
 
 			receivedErr := CheckImports(pathToCode, data.args.allowedImports)
 

compile_code_test.go

@@ -3,6 +3,8 @@ package coderunner
 import (
 	"context"
 	"io/ioutil"
+	"os"
+	"path/filepath"
 	"testing"
 
 	mapset "github.com/deckarep/golang-set"
@@ -142,6 +144,7 @@ func TestCompileCode(test *testing.T) {
 		test.Run(data.name, func(test *testing.T) {
 			pathToCode, err := systemutils.SaveTemporaryText(data.args.code, ".go")
 			require.NoError(test, err)
+			defer os.RemoveAll(filepath.Dir(pathToCode)) // nolint: errcheck
 
 			pathToExecutable, receivedErr :=
 				CompileCode(data.args.ctx, pathToCode, data.args.allowedImports)

system-utils/save_temporary_text_test.go

@@ -13,6 +13,7 @@ import (
 func TestSaveTemporaryText(test *testing.T) {
 	path, err := SaveTemporaryText("test", ".txt")
 	require.NoError(test, err)
+	defer os.RemoveAll(filepath.Dir(path)) // nolint: errcheck
 
 	content, err := ioutil.ReadFile(path)
 	require.NoError(test, err)