From 2478398b3baae5b45b83eb0a81508848da05d6ec Mon Sep 17 00:00:00 2001
From: Aykhan Shahsuvarov <aykhan.shahs@gmail.com>
Date: Fri, 18 Apr 2025 15:29:31 +0400
Subject: [PATCH] =?UTF-8?q?=F0=9F=9A=A8=20Update=20test?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 Taskfile.yaml      |  2 ++
 config/cli_test.go | 30 ++++++------------------------
 2 files changed, 8 insertions(+), 24 deletions(-)

diff --git a/Taskfile.yaml b/Taskfile.yaml
index f47e679..eb1ef57 100644
--- a/Taskfile.yaml
+++ b/Taskfile.yaml
@@ -24,6 +24,8 @@ tasks:
 
     lint: golangci-lint run
 
+    test: go test ./...
+
     build: go build -ldflags "-s -w" -o "dodo"
 
     build-all:
diff --git a/config/cli_test.go b/config/cli_test.go
index b055304..9adc205 100644
--- a/config/cli_test.go
+++ b/config/cli_test.go
@@ -42,13 +42,13 @@ func TestReadCLI(t *testing.T) {
 			expectFile:  "/path/to/config.json",
 			expectError: false,
 			expectedConfig: &Config{
-				Method:       stringPtr("POST"),
+				Method:       toPtr("POST"),
 				URL:          &types.RequestURL{},
-				DodosCount:   uintPtr(10),
-				RequestCount: uintPtr(1000),
+				DodosCount:   toPtr[uint](10),
+				RequestCount: toPtr[uint](1000),
 				Duration:     &types.Duration{Duration: 3 * time.Minute},
 				Timeout:      &types.Timeout{Duration: 3 * time.Second},
-				Yes:          boolPtr(true),
+				Yes:          toPtr(true),
 			},
 		},
 		{
@@ -60,10 +60,6 @@ func TestReadCLI(t *testing.T) {
 		},
 	}
 
-	// Save original command-line arguments
-	origArgs := os.Args
-	origFlagCommandLine := flag.CommandLine
-
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {
 			// Reset flag.CommandLine to its original state
@@ -85,9 +81,6 @@ func TestReadCLI(t *testing.T) {
 			// Call the function being tested
 			file, err := config.ReadCLI()
 
-			// Reset os.Args after test
-			os.Args = origArgs
-
 			// Assert expected results
 			assert.Equal(t, tt.expectFile, file)
 			if tt.expectError {
@@ -126,9 +119,6 @@ func TestReadCLI(t *testing.T) {
 			}
 		})
 	}
-
-	// Restore original flag.CommandLine
-	flag.CommandLine = origFlagCommandLine
 }
 
 // Skip the prompt tests as they require interactive input/output handling
@@ -168,14 +158,6 @@ func TestCLIYesOrNoReaderBasic(t *testing.T) {
 }
 
 // Helper types and functions for testing
-func stringPtr(s string) *string {
-	return &s
-}
-
-func uintPtr(u uint) *uint {
-	return &u
-}
-
-func boolPtr(b bool) *bool {
-	return &b
+func toPtr[T any](value T) *T {
+	return &value
 }