Тестовый файл ищется в дире: *_test.go,

но не init_test.go и не на _mock_test.go

pkg/formatters/ast/ast.go

@@ -46,18 +46,20 @@ func (a *ASTer) ДобавитьШаг(шаг, f, ps string) error {
 }
 
 func (a *ASTer) найтиТестовыйФайл() error {
-	a.сгенеритьИмяФайла()
 	_, err := os.Stat(INIT_TEST_GO_FNAME)
 	if err != nil {
 		return err
 	}
 
-	_, err = os.Stat(a.pkg_test_go_fname)
+	имя, err := найтиТестовыйФайлВЭтойДире()
+	if err != nil {
+		return errors.Join(errors.New("Не могу найти *_test.go файл"), err)
+	}
+
+	a.pkg_test_go_fname = имя
+
 	return err
 }
-func (a *ASTer) сгенеритьИмяФайла() {
-	a.pkg_test_go_fname = a.pkg + "_test.go"
-}
 
 func (a *ASTer) добавитьШаг(шаг, f, ps string) error {
 	err := a.добавитьШагвInitФайл(шаг, f)

pkg/formatters/ast/ast/features/app.feature

@@ -54,6 +54,58 @@
       
       ```
 
+  Сценарий: Добавление шага, с другим тестовым файлом
+    Дано Файл "init_test.go":
+      ```
+      package mypkg
+      
+      func InitializeScenario(ctx *godog.ScenarioContext) {
+      
+        ctx.Before(func(ctx context.Context, sc *godog.Scenario) (context.Context, error) {
+          beforeScenario()
+          return ctx, nil
+        })
+        ctx.After(func(ctx context.Context, sc *godog.Scenario, err error) (context.Context, error) {
+          afterScenario()
+          return ctx, nil
+        })
+        InitializeGomegaForGodog(ctx)
+      }
+      
+      ```
+    Дано Файл "system_test.go":
+      ```
+      package mypkg
+      ```
+    Когда Добавляется шаг: "^Привет Мир!$" с функцией "ПриветМир" с ""
+    То Файл "init_test.go" должен содержать:
+      ```
+      package mypkg
+      
+      func InitializeScenario(ctx *godog.ScenarioContext) {
+      	ctx.Step(`^Привет Мир!$`, ПриветМир)
+      
+      	ctx.Before(func(ctx context.Context, sc *godog.Scenario) (context.Context, error) {
+      		beforeScenario()
+      		return ctx, nil
+      	})
+      	ctx.After(func(ctx context.Context, sc *godog.Scenario, err error) (context.Context, error) {
+      		afterScenario()
+      		return ctx, nil
+      	})
+      	InitializeGomegaForGodog(ctx)
+      }
+      
+      ```
+    То Файл "system_test.go" должен содержать:
+      ```
+      package mypkg
+      
+      func ПриветМир() {
+      }
+      
+      ```
+
   Сценарий: Добавление шага с параметрами: int
     Дано Файл "init_test.go":
       ```

pkg/formatters/ast/global.go

@@ -1,10 +1,19 @@
 package ast
 
+import "fmt"
+
 // 0: имя, 1: тип
 type Параметр []string
 type Параметры []Параметр
 
 func ДобавитьШаг(шаг, f string, ps string) error {
+	err := добавитьШаг(шаг, f, ps)
+	if err != nil {
+		fmt.Println("Ошибка добавления тестового шага:", err)
+	}
+	return err
+}
+func добавитьШаг(шаг, f string, ps string) error {
 	a, err := NewASTer()
 	if err != nil {
 		return err

pkg/formatters/ast/helpers.go

@@ -5,6 +5,7 @@ import (
 	"go/ast"
 	"os"
 	"path/filepath"
+	"strings"
 	"time"
 
 	"errors"
@@ -45,6 +46,33 @@ func получитьИмяGoПакетаВЭтойДире() (pkg_name string,
 	return
 }
 
+func найтиТестовыйФайлВЭтойДире() (name string, err_ret error) {
+	err_ret = errors.New("not found")
+
+	err := filepath.Walk(".", func(path string, info os.FileInfo, err error) error {
+		if err != nil {
+			return err
+		}
+
+		if !info.IsDir() && strings.HasSuffix(info.Name(), "_test.go") {
+			if info.Name() == "init_test.go" ||
+				strings.HasSuffix(info.Name(), "_mock_test.go") {
+				return nil
+			}
+			name = info.Name()
+			err_ret = nil
+			return filepath.SkipDir
+		}
+		return nil
+	})
+
+	if err != nil {
+		return "", err
+	}
+
+	return
+}
+
 func забекапитьФайлСоВременем(fname string) error {
 	os.MkdirAll(BACKUP_DIR, 0755)
 	currentTime := time.Now()