@@ -23,8 +23,14 @@ func TestEnvironmentVariables(t *testing.T) {
2323 for _ , tt := range tests {
2424 t .Run (tt .name , func (t * testing.T ) {
2525 // Set environment variable
26- os .Setenv (tt .envKey , tt .envValue )
27- defer os .Unsetenv (tt .envKey )
26+ if err := os .Setenv (tt .envKey , tt .envValue ); err != nil {
27+ t .Fatalf ("Failed to set env var: %v" , err )
28+ }
29+ defer func () {
30+ if err := os .Unsetenv (tt .envKey ); err != nil {
31+ t .Errorf ("Failed to unset env var: %v" , err )
32+ }
33+ }()
2834
2935 // Test string env vars
3036 getEnvOrDefault := func (key , defaultValue string ) string {
@@ -58,8 +64,14 @@ func TestEnvironmentVariablesInt(t *testing.T) {
5864 for _ , tt := range tests {
5965 t .Run (tt .name , func (t * testing.T ) {
6066 // Set environment variable
61- os .Setenv (tt .envKey , tt .envValue )
62- defer os .Unsetenv (tt .envKey )
67+ if err := os .Setenv (tt .envKey , tt .envValue ); err != nil {
68+ t .Fatalf ("Failed to set env var: %v" , err )
69+ }
70+ defer func () {
71+ if err := os .Unsetenv (tt .envKey ); err != nil {
72+ t .Errorf ("Failed to unset env var: %v" , err )
73+ }
74+ }()
6375
6476 // Test int env vars
6577 getEnvOrDefaultInt := func (key string , defaultValue int ) int {
@@ -95,8 +107,14 @@ func TestEnvironmentVariablesInt64(t *testing.T) {
95107 for _ , tt := range tests {
96108 t .Run (tt .name , func (t * testing.T ) {
97109 // Set environment variable
98- os .Setenv (tt .envKey , tt .envValue )
99- defer os .Unsetenv (tt .envKey )
110+ if err := os .Setenv (tt .envKey , tt .envValue ); err != nil {
111+ t .Fatalf ("Failed to set env var: %v" , err )
112+ }
113+ defer func () {
114+ if err := os .Unsetenv (tt .envKey ); err != nil {
115+ t .Errorf ("Failed to unset env var: %v" , err )
116+ }
117+ }()
100118
101119 // Test int64 env vars
102120 getEnvOrDefaultInt64 := func (key string , defaultValue int64 ) int64 {
0 commit comments