v2

handler/util_test.go

@@ -0,0 +1,43 @@
+package handler
+
+import (
+	"strings"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestGenerateId(t *testing.T) {
+	id, err := generateId()
+	assert.NoError(t, err)
+	assert.Equal(t, 10, len(id))
+	for _, char := range id {
+		assert.True(t, strings.ContainsRune(charset, char))
+	}
+}
+
+func TestHighlight(t *testing.T) {
+	goContent := `package main
+
+import "fmt"
+
+func main() {
+	fmt.Println("Hello, World!")
+}`
+
+	// Test with a specific language extension (.go) -> SHOULD highlight
+	highlighted, err := highlight(goContent, "go", "monokai")
+	assert.NoError(t, err)
+	assert.Contains(t, highlighted, `style="color:#f92672"`, "Should highlight Go code with .go extension")
+
+	// Test with a generic extension (.txt) -> SHOULD auto-detect and highlight
+	highlighted, err = highlight(goContent, "txt", "monokai")
+	assert.NoError(t, err)
+	assert.Contains(t, highlighted, `style="color:#f92672"`, "Should auto-detect Go code with .txt extension")
+
+	// NEW TEST: No extension -> SHOULD NOT highlight
+	highlighted, err = highlight(goContent, "", "monokai")
+	assert.NoError(t, err)
+	assert.NotContains(t, highlighted, `style="color:#f92672"`, "Should NOT highlight Go code when no extension is given")
+	assert.Contains(t, highlighted, "package main", "Should still contain the original text")
+}