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

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

.gitignore

@@ -6,7 +6,6 @@ Gopkg.toml
 
 .DS_Store
 .idea
-.vscode
 
 _artifacts
 

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)
@@ -252,7 +254,7 @@ func (a *ASTer) сконвертитьСтрокувПараметры(in string
 }
 
 func (a *ASTer) перезаписатьInitФайл() error {
-	err := переименоватьФайлСоВременем(INIT_TEST_GO_FNAME)
+	err := забекапитьФайлСоВременем(INIT_TEST_GO_FNAME)
 	if err != nil {
 		return errors.Join(err, errors.New("cant backup orig file"))
 	}
@@ -266,7 +268,7 @@ func (a *ASTer) перезаписатьInitФайл() error {
 	return format.Node(f, a.init_test_fset, a.init_test_node)
 }
 func (a *ASTer) перезаписатьФайл() error {
-	err := переименоватьФайлСоВременем(a.pkg_test_go_fname)
+	err := забекапитьФайлСоВременем(a.pkg_test_go_fname)
 	if err != nil {
 		return errors.Join(err, errors.New("cant backup orig file"))
 	}

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"
@@ -12,10 +13,14 @@ import (
 	"git.golang1.ru/softonik/godog/pkg/lib"
 )
 
-const (
+var (
-	BACKUP_DIR = ".back"
+	BACKUP_DIR = "backup"
 )
 
+func init() {
+	BACKUP_DIR = os.TempDir() + "/godog/" + BACKUP_DIR
+}
+
 func получитьИмяGoПакетаВЭтойДире() (pkg_name string, err_ret error) {
 	err_ret = errors.New("not found")
 
@@ -41,12 +46,39 @@ func получитьИмяGoПакетаВЭтойДире() (pkg_name string,
 	return
 }
 
-func переименоватьФайлСоВременем(fname string) error {
+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()
 	timestamp := currentTime.Format("20060102_150405")
 	new_fname := fmt.Sprintf(BACKUP_DIR+"/%v_%v", fname, timestamp)
-	return os.Rename(fname, new_fname)
+	return lib.CopyFile(fname, new_fname)
 }
 
 func являетсяЛиШагомсТакойЖеФункцией(stmt ast.Stmt, step *ast.ExprStmt) bool {

pkg/lib/lib.go

@@ -44,3 +44,12 @@ func CopyFile(src, dst string) (err error) {
 
 	return
 }
+
+func MoveFile(src, dst string) (err error) {
+	err = CopyFile(src, dst)
+	if err != nil {
+		return
+	}
+
+	return os.Remove(src)
+}