@@ -60,13 +60,19 @@ func TestMongoTestContainer(t *testing.T) {
6060 // save current MONGO_TEST value
6161 origEnv := os .Getenv ("MONGO_TEST" )
6262 testValue := "mongodb://original-value:27017"
63- os .Setenv ("MONGO_TEST" , testValue )
63+ if err := os .Setenv ("MONGO_TEST" , testValue ); err != nil {
64+ t .Fatalf ("Failed to set MONGO_TEST environment variable: %v" , err )
65+ }
6466 defer func () {
6567 // restore original value
6668 if origEnv == "" {
67- os .Unsetenv ("MONGO_TEST" )
69+ if err := os .Unsetenv ("MONGO_TEST" ); err != nil {
70+ t .Logf ("Warning: failed to unset MONGO_TEST environment variable: %v" , err )
71+ }
6872 } else {
69- os .Setenv ("MONGO_TEST" , origEnv )
73+ if err := os .Setenv ("MONGO_TEST" , origEnv ); err != nil {
74+ t .Logf ("Warning: failed to restore MONGO_TEST environment variable: %v" , err )
75+ }
7076 }
7177 }()
7278
@@ -82,11 +88,15 @@ func TestMongoTestContainer(t *testing.T) {
8288 t .Run ("close with no original environment variable" , func (t * testing.T ) {
8389 // save current MONGO_TEST value
8490 origEnv := os .Getenv ("MONGO_TEST" )
85- os .Unsetenv ("MONGO_TEST" )
91+ if err := os .Unsetenv ("MONGO_TEST" ); err != nil {
92+ t .Fatalf ("Failed to unset MONGO_TEST environment variable: %v" , err )
93+ }
8694 defer func () {
8795 // restore original value
8896 if origEnv != "" {
89- os .Setenv ("MONGO_TEST" , origEnv )
97+ if err := os .Setenv ("MONGO_TEST" , origEnv ); err != nil {
98+ t .Logf ("Warning: failed to restore MONGO_TEST environment variable: %v" , err )
99+ }
90100 }
91101 }()
92102
0 commit comments