fixig gh template test

Mzack9999 · Mzack9999 · commit f0dfda6542b1 · 2025-07-17T15:15:56.000+02:00
diff --git a/pkg/external/customtemplates/github_test.go b/pkg/external/customtemplates/github_test.go
@@ -1,31 +1,24 @@
 package customtemplates
 
 import (
+	"bytes"
 	"context"
-	"os"
 	"path/filepath"
+	"strings"
 	"testing"
 
 	"github.com/projectdiscovery/gologger"
 	"github.com/projectdiscovery/gologger/levels"
 	"github.com/projectdiscovery/nuclei/v3/pkg/catalog/config"
 	"github.com/projectdiscovery/nuclei/v3/pkg/testutils"
+	"github.com/projectdiscovery/nuclei/v3/pkg/utils"
 	"github.com/stretchr/testify/require"
 )
 
-// stdoutWriter adapts os.Stdout to the gologger writer interface
-type stdoutWriter struct{}
-
-func (w *stdoutWriter) Write(data []byte, level levels.Level) {
-	os.Stdout.Write(data)
-}
-
 func TestDownloadCustomTemplatesFromGitHub(t *testing.T) {
-	// if osutils.IsOSX() {
-	// 	t.Skip("skipping on macos due to unknown failure (works locally)")
-	// }
-
-	gologger.DefaultLogger.SetWriter(&stdoutWriter{})
+	// Capture output to check for rate limit errors
+	outputBuffer := &bytes.Buffer{}
+	gologger.DefaultLogger.SetWriter(&utils.CaptureWriter{Buffer: outputBuffer})
 	gologger.DefaultLogger.SetMaxLevel(levels.LevelDebug)
 
 	templatesDirectory := t.TempDir()
@@ -39,5 +32,11 @@ func TestDownloadCustomTemplatesFromGitHub(t *testing.T) {
 
 	ctm.Download(context.Background())
 
+	// Check if output contains rate limit error and skip test if so
+	output := outputBuffer.String()
+	if strings.Contains(output, "API rate limit exceeded") {
+		t.Skip("GitHub API rate limit exceeded, skipping test")
+	}
+
 	require.DirExists(t, filepath.Join(templatesDirectory, "github", "projectdiscovery", "nuclei-templates-test"), "cloned directory does not exists")
 }
diff --git a/pkg/utils/capture_writer.go b/pkg/utils/capture_writer.go
@@ -0,0 +1,16 @@
+package utils
+
+import (
+	"bytes"
+
+	"github.com/projectdiscovery/gologger/levels"
+)
+
+// CaptureWriter captures log output for testing
+type CaptureWriter struct {
+	Buffer *bytes.Buffer
+}
+
+func (w *CaptureWriter) Write(data []byte, level levels.Level) {
+	w.Buffer.Write(data)
+}
