add short flag to benchmarks

Author: tac0turtle

da/compression/benchmark_test.go

@@ -12,6 +12,10 @@ import (
 
 // TestLargeBlobCompressionEfficiency tests compression efficiency for blob sizes from 20KB to 2MB
 func TestLargeBlobCompressionEfficiency(t *testing.T) {
+	if testing.Short() {
+		t.Skip("Skipping large blob compression test in short mode")
+	}
+	
 	config := DefaultConfig()
 	compressor, err := NewCompressibleDA(nil, config)
 	require.NoError(t, err)
@@ -54,7 +58,7 @@ func TestLargeBlobCompressionEfficiency(t *testing.T) {
 		},
 	}
 
-	fmt.Printf("\n=== Large Blob Compression Efficiency Test ===\n")
+	fmt.Printf("\n=== Blob Compression Efficiency Test ===\n")
 	fmt.Printf("%-15s %-10s %-12s %-12s %-10s %-15s\n",
 		"Data Type", "Size", "Compressed", "Saved", "Ratio", "Compression")
 	fmt.Printf("%-15s %-10s %-12s %-12s %-10s %-15s\n",

scripts/test.go

@@ -1,6 +1,7 @@
 package main
 
 import (
+	"flag"
 	"fmt"
 	"log"
 	"os"
@@ -10,6 +11,10 @@ import (
 )
 
 func main() {
+	// Parse command line flags
+	shortMode := flag.Bool("short", true, "Run tests in short mode (skip long-running tests)")
+	flag.Parse()
+
 	rootDir := "." // Start from the current directory
 	var testFailures bool
 	err := filepath.WalkDir(rootDir, func(path string, d os.DirEntry, err error) error {
@@ -28,8 +33,16 @@ func main() {
 			// or adjust logic if root tests are also desired.
 			// For this example, we'll run tests in all directories with go.mod.
 
-			fmt.Printf("--> Running tests in: %s\n", modDir)
-			cmd := exec.Command("go", "test", "./...", "-cover")
+			// Build test command with optional -short flag
+			testArgs := []string{"test", "./...", "-cover"}
+			if *shortMode {
+				testArgs = append([]string{"test", "./...", "-short", "-cover"}, testArgs[3:]...)
+				fmt.Printf("--> Running tests in short mode in: %s\n", modDir)
+			} else {
+				fmt.Printf("--> Running full tests in: %s\n", modDir)
+			}
+			
+			cmd := exec.Command("go", testArgs...)
 			cmd.Dir = modDir // Set the working directory for the command
 			cmd.Stdout = os.Stdout
 			cmd.Stderr = os.Stderr

scripts/test.mk

@@ -4,12 +4,18 @@ clean-testcache:
 	@go clean --testcache
 .PHONY: clean-testcache
 
-## test: Running unit tests for all go.mods
+## test: Running unit tests for all go.mods (fast mode with -short flag)
 test:
-	@echo "--> Running unit tests"
-	@go run -tags='run integration' scripts/test.go
+	@echo "--> Running unit tests (fast mode)"
+	@go run -tags='run integration' scripts/test.go -short
 .PHONY: test
 
+## test-full: Running full unit tests for all go.mods (includes long-running tests)
+test-full:
+	@echo "--> Running full unit tests (including long-running tests)"
+	@go run -tags='run integration' scripts/test.go -short=false
+.PHONY: test-full
+
 ## test-all: Running all tests including Docker E2E
 test-all: test test-docker-e2e
 	@echo "--> All tests completed"
@@ -33,12 +39,18 @@ test-integration-cover:
 	@cd node && go test -mod=readonly -failfast -timeout=15m -tags='integration' -coverprofile=coverage.txt -covermode=atomic ./...
 .PHONY: test-integration-cover
 
-## test-cover: generate code coverage report.
+## test-cover: generate code coverage report (fast mode with -short flag).
 test-cover:
-	@echo "--> Running unit tests"
-	@go run -tags=cover scripts/test_cover.go
+	@echo "--> Running unit tests with coverage (fast mode)"
+	@go run -tags=cover scripts/test_cover.go -short
 .PHONY: test-cover
 
+## test-cover-full: generate code coverage report with all tests.
+test-cover-full:
+	@echo "--> Running full unit tests with coverage"
+	@go run -tags=cover scripts/test_cover.go -short=false
+.PHONY: test-cover-full
+
 ## test-evm: Running EVM tests
 test-evm:
 	@echo "--> Running EVM tests"

scripts/test_cover.go

@@ -5,6 +5,7 @@ package main
 
 import (
 	"bufio"
+	"flag"
 	"fmt"
 	"log"
 	"os"
@@ -14,6 +15,10 @@ import (
 )
 
 func main() {
+	// Parse command line flags
+	shortMode := flag.Bool("short", true, "Run tests in short mode (skip long-running tests)")
+	flag.Parse()
+	
 	rootDir := "."
 
 	var coverFiles []string
@@ -38,8 +43,16 @@ func main() {
 			fullCoverProfilePath := filepath.Join(modDir, "cover.out")
 			relativeCoverProfileArg := "cover.out"
 
-			fmt.Printf("--> Running tests with coverage in: %s (profile: %s)\n", modDir, relativeCoverProfileArg)
-			cmd := exec.Command("go", "test", "./...", "-race", "-coverprofile="+relativeCoverProfileArg, "-covermode=atomic")
+			// Build test command with optional -short flag
+			testArgs := []string{"test", "./...", "-race", "-coverprofile=" + relativeCoverProfileArg, "-covermode=atomic"}
+			if *shortMode {
+				testArgs = []string{"test", "./...", "-short", "-race", "-coverprofile=" + relativeCoverProfileArg, "-covermode=atomic"}
+				fmt.Printf("--> Running tests with coverage in short mode in: %s (profile: %s)\n", modDir, relativeCoverProfileArg)
+			} else {
+				fmt.Printf("--> Running full tests with coverage in: %s (profile: %s)\n", modDir, relativeCoverProfileArg)
+			}
+			
+			cmd := exec.Command("go", testArgs...)
 			cmd.Dir = modDir
 			cmd.Stdout = os.Stdout
 			cmd.Stderr = os.Stderr