refactor: get config file path as a parameter

grid-proxy/tests/load_testing/load.go

@@ -5,6 +5,8 @@ import (
 	"log"
 	"net/http"
 	"os"
+	"path/filepath"
+	"slices"
 	"time"
 
 	prometheus_integration "github.com/threefoldtech/tfgrid-sdk-go/grid-proxy/tests/load_testing/prometheus"
@@ -14,9 +16,12 @@ import (
 	"github.com/prometheus/client_golang/prometheus"
 )
 
-func PerformLoadTesting() error {
-
-	data, err := os.ReadFile("test.yml")
+func PerformLoadTesting(path string) error {
+	extension := filepath.Ext(path)
+	if !slices.Contains([]string{".yml", ".yaml"}, extension) {
+		return fmt.Errorf("Unsupported file extension")
+	}
+	data, err := os.ReadFile(path)
 	if err != nil {
 		return fmt.Errorf("Failed to read YAML file: %v", err)
 	}

grid-proxy/tests/load_testing/load_test.go

@@ -3,7 +3,7 @@ package load_test
 import "testing"
 
 func TestLoad(t *testing.T) {
-	err := PerformLoadTesting()
+	err := PerformLoadTesting("test.yml")
 	if err != nil {
 		t.Fatalf("Load test failed: %v", err)
 	}