From 9c047e9df25a4f630a5ff4caf9ad9a4125ca0422 Mon Sep 17 00:00:00 2001
From: j178 <10510431+j178@users.noreply.github.com>
Date: Mon, 20 Nov 2023 19:59:34 +0800
Subject: [PATCH] fix: truncate long test case for local test

---
 lang/test.go | 36 ++++++++++++++++++++++++++++--------
 1 file changed, 28 insertions(+), 8 deletions(-)

diff --git a/lang/test.go b/lang/test.go
index 915aa15c..1f74c8bd 100644
--- a/lang/test.go
+++ b/lang/test.go
@@ -203,14 +203,19 @@ func runTest(q *leetcode.QuestionData, genResult *GenerateResult, args []string,
 			actualOutput, stdout := extractOutput(outputBuf.String())
 			mayAppendStdout := func() {
 				if stdout != "" {
-					out := config.StdoutStyle.Render(strings.ReplaceAll(stdout, "\n", "↩ "))
+					out := config.StdoutStyle.Render(utils.TruncateString(strings.ReplaceAll(stdout, "\n", "↩ "), 100))
 					l.AppendItem(fmt.Sprintf("Stdout:     %s", out))
 				}
 			}
 			if ctx.Err() != nil {
 				l.AppendItem(fmt.Sprintf("Case %d:    %s", c.No, config.ErrorStyle.Render("Time limit exceeded")))
 				l.Indent()
-				l.AppendItem(fmt.Sprintf("Input:      %s", strings.ReplaceAll(c.InputString(), "\n", "↩ ")))
+				l.AppendItem(
+					fmt.Sprintf(
+						"Input:      %s",
+						utils.TruncateString(strings.ReplaceAll(c.InputString(), "\n", "↩ "), 100),
+					),
+				)
 				mayAppendStdout()
 				l.UnIndent()
 				return
@@ -218,7 +223,12 @@ func runTest(q *leetcode.QuestionData, genResult *GenerateResult, args []string,
 			if err != nil {
 				l.AppendItem(fmt.Sprintf("Case %d:    %s", c.No, config.ErrorStyle.Render("Runtime error")))
 				l.Indent()
-				l.AppendItem(fmt.Sprintf("Input:      %s", strings.ReplaceAll(c.InputString(), "\n", "↩ ")))
+				l.AppendItem(
+					fmt.Sprintf(
+						"Input:      %s",
+						utils.TruncateString(strings.ReplaceAll(c.InputString(), "\n", "↩ "), 100),
+					),
+				)
 				mayAppendStdout()
 				l.UnIndent()
 				return
@@ -227,8 +237,13 @@ func runTest(q *leetcode.QuestionData, genResult *GenerateResult, args []string,
 			if err != nil {
 				l.AppendItem(fmt.Sprintf("Case %d:    %s", c.No, config.ErrorStyle.Render("Invalid output")))
 				l.Indent()
-				l.AppendItem(fmt.Sprintf("Input:      %s", strings.ReplaceAll(c.InputString(), "\n", "↩ ")))
-				l.AppendItem(fmt.Sprintf("Output:     %s", actualOutput))
+				l.AppendItem(
+					fmt.Sprintf(
+						"Input:      %s",
+						utils.TruncateString(strings.ReplaceAll(c.InputString(), "\n", "↩ "), 100),
+					),
+				)
+				l.AppendItem(fmt.Sprintf("Output:     %s", utils.TruncateString(actualOutput, 100)))
 				mayAppendStdout()
 				l.UnIndent()
 				return
@@ -241,9 +256,14 @@ func runTest(q *leetcode.QuestionData, genResult *GenerateResult, args []string,
 				l.AppendItem(fmt.Sprintf("Case %d:    %s", c.No, config.FailedStyle.Render("Wrong answer")))
 				l.Indent()
 				l.AppendItem(fmt.Sprintf("Reason:     %s", r.GetInfo()))
-				l.AppendItem(fmt.Sprintf("Input:      %s", strings.ReplaceAll(c.InputString(), "\n", "↩ ")))
-				l.AppendItem(fmt.Sprintf("Output:     %s", actualOutput))
-				l.AppendItem(fmt.Sprintf("Expected:   %s", c.Output))
+				l.AppendItem(
+					fmt.Sprintf(
+						"Input:      %s",
+						utils.TruncateString(strings.ReplaceAll(c.InputString(), "\n", "↩ "), 100),
+					),
+				)
+				l.AppendItem(fmt.Sprintf("Output:     %s", utils.TruncateString(actualOutput, 100)))
+				l.AppendItem(fmt.Sprintf("Expected:   %s", utils.TruncateString(c.Output, 100)))
 				mayAppendStdout()
 				l.UnIndent()
 			}