From f49fd0b4d29567566347ac99e36a5809cfff07fe Mon Sep 17 00:00:00 2001
From: Nikitin Aleksandr <nikitinav@atomsbt.ru>
Date: Wed, 21 Feb 2024 14:49:17 +0300
Subject: [PATCH] =?UTF-8?q?=D1=81=D0=B4=D0=B5=D0=BB=D0=B0=D0=BB=20LoadEnvT?=
 =?UTF-8?q?est()?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 config_main/config_main.go      | 19 ++++++++++++++++++-
 config_main/config_main_test.go |  4 ++++
 2 files changed, 22 insertions(+), 1 deletion(-)

diff --git a/config_main/config_main.go b/config_main/config_main.go
index 1d303d14..904c36fb 100644
--- a/config_main/config_main.go
+++ b/config_main/config_main.go
@@ -6,6 +6,7 @@ import (
 	"github.com/ManyakRus/starter/logger"
 	"github.com/ManyakRus/starter/micro"
 	"github.com/joho/godotenv"
+	"os"
 	//log "github.com/sirupsen/logrus"
 	//log "github.com/sirupsen/logrus"
 	//"gitlab.aescorp.ru/dsp_dev/notifier/notifier_adp_eml/internal/v0/app/types"
@@ -17,12 +18,28 @@ var log = logger.GetLog()
 
 // LoadEnv - загружает из файла .env переменные в переменные окружения
 func LoadEnv() {
-
 	dir := micro.ProgramDir()
 	filename := dir + ".env"
 	LoadEnv_from_file(filename)
 }
 
+// LoadEnvTest - загружает из файла .env переменные в переменные окружения, кроме для STAGE=dev или prod
+// для модулей тестирования _test.go
+func LoadEnvTest() {
+	dir := micro.ProgramDir()
+	filename := dir + ".env"
+
+	//не загружаем для STAGE=dev, т.к. переменные окружения кубернетеса
+	stage := os.Getenv("STAGE")
+	if stage == "dev" || stage == "prod" {
+		log.Info("LoadEnv() ignore STAGE: dev, filename: ", filename)
+		return
+	}
+
+	//
+	LoadEnv_from_file(filename)
+}
+
 // LoadEnv - загружает из файла .env переменные в переменные окружения, возвращает ошибку
 func LoadEnv_err() error {
 	var err error
diff --git a/config_main/config_main_test.go b/config_main/config_main_test.go
index 32dea141..e8e7ce15 100644
--- a/config_main/config_main_test.go
+++ b/config_main/config_main_test.go
@@ -38,3 +38,7 @@ func TestLoadENV_or_SettingsTXT(t *testing.T) {
 
 	LoadENV_or_SettingsTXT()
 }
+
+func TestLoadEnvTest(t *testing.T) {
+	LoadEnvTest()
+}