From b1ef69a870d523c119b950674ca3862f88654093 Mon Sep 17 00:00:00 2001
From: Piggy NL <piggynl@outlook.com>
Date: Tue, 8 Nov 2022 16:52:16 +0800
Subject: [PATCH] Add test timeout multiplier environment variable

---
 extra_test.go | 31 +++++++++++++++++++++----------
 1 file changed, 21 insertions(+), 10 deletions(-)

diff --git a/extra_test.go b/extra_test.go
index 0ae3d9d..0e61c67 100644
--- a/extra_test.go
+++ b/extra_test.go
@@ -2,6 +2,8 @@ package goldmark_test
 
 import (
 	"bytes"
+	"os"
+	"strconv"
 	"strings"
 	"testing"
 	"time"
@@ -13,6 +15,15 @@ import (
 	"github.com/yuin/goldmark/testutil"
 )
 
+var testTimeoutMultiplier = 1.0
+
+func init() {
+	m, err := strconv.ParseFloat(os.Getenv("GOLDMARK_TEST_TIMEOUT_MULTIPLIER"), 64)
+	if err == nil {
+		testTimeoutMultiplier = m
+	}
+}
+
 func TestExtras(t *testing.T) {
 	markdown := New(WithRendererOptions(
 		html.WithXHTML(),
@@ -108,8 +119,8 @@ func TestDeepNestedLabelPerformance(t *testing.T) {
 	var b bytes.Buffer
 	_ = markdown.Convert(source, &b)
 	finished := nowMillis()
-	if (finished - started) > 5000 {
-		t.Error("Parsing deep nested labels took more 5 secs")
+	if (finished - started) > int64(5000*testTimeoutMultiplier) {
+		t.Error("Parsing deep nested labels took too long")
 	}
 }
 
@@ -128,8 +139,8 @@ func TestManyProcessingInstructionPerformance(t *testing.T) {
 	var b bytes.Buffer
 	_ = markdown.Convert(source, &b)
 	finished := nowMillis()
-	if (finished - started) > 5000 {
-		t.Error("Parsing processing instructions took more 5 secs")
+	if (finished - started) > int64(5000*testTimeoutMultiplier) {
+		t.Error("Parsing processing instructions took too long")
 	}
 }
 
@@ -148,8 +159,8 @@ func TestManyCDATAPerformance(t *testing.T) {
 	var b bytes.Buffer
 	_ = markdown.Convert(source, &b)
 	finished := nowMillis()
-	if (finished - started) > 5000 {
-		t.Error("Parsing processing instructions took more 5 secs")
+	if (finished - started) > int64(5000*testTimeoutMultiplier) {
+		t.Error("Parsing processing instructions took too long")
 	}
 }
 
@@ -168,8 +179,8 @@ func TestManyDeclPerformance(t *testing.T) {
 	var b bytes.Buffer
 	_ = markdown.Convert(source, &b)
 	finished := nowMillis()
-	if (finished - started) > 5000 {
-		t.Error("Parsing processing instructions took more 5 secs")
+	if (finished - started) > int64(5000*testTimeoutMultiplier) {
+		t.Error("Parsing processing instructions took too long")
 	}
 }
 
@@ -188,7 +199,7 @@ func TestManyCommentPerformance(t *testing.T) {
 	var b bytes.Buffer
 	_ = markdown.Convert(source, &b)
 	finished := nowMillis()
-	if (finished - started) > 5000 {
-		t.Error("Parsing processing instructions took more 5 secs")
+	if (finished - started) > int64(5000*testTimeoutMultiplier) {
+		t.Error("Parsing processing instructions took too long")
 	}
 }