added more tests

main_test.go

@@ -0,0 +1,27 @@
+package main
+
+import (
+	"testing"
+)
+
+func TestMainFunction(t *testing.T) {
+	// Сохраняем оригинальную функцию
+	originalExecute := execute
+	defer func() {
+		execute = originalExecute // Восстанавливаем после теста
+	}()
+
+	// Флаг для проверки вызова
+	called := false
+	execute = func() {
+		called = true // Подменяем функцию
+	}
+
+	// Запускаем main
+	main()
+
+	// Проверяем, что функция была вызвана
+	if !called {
+		t.Fatal("Expected cmd.Execute to be called, but it wasn't")
+	}
+}