github.com/Prakhar-Agarwal-byte/moby@v0.0.0-20231027092010-a14e3e8ab87e/integration/plugin/logging/main_test.go (about) 1 package logging // import "github.com/Prakhar-Agarwal-byte/moby/integration/plugin/logging" 2 3 import ( 4 "context" 5 "os" 6 "testing" 7 8 "github.com/Prakhar-Agarwal-byte/moby/testutil" 9 "github.com/Prakhar-Agarwal-byte/moby/testutil/environment" 10 "go.opentelemetry.io/otel" 11 "go.opentelemetry.io/otel/codes" 12 ) 13 14 var ( 15 testEnv *environment.Execution 16 baseContext context.Context 17 ) 18 19 func TestMain(m *testing.M) { 20 shutdown := testutil.ConfigureTracing() 21 ctx, span := otel.Tracer("").Start(context.Background(), "integration/plugin/logging.TestMain") 22 baseContext = ctx 23 24 var err error 25 testEnv, err = environment.New(ctx) 26 if err != nil { 27 span.SetStatus(codes.Error, err.Error()) 28 span.End() 29 shutdown(ctx) 30 panic(err) 31 } 32 err = environment.EnsureFrozenImagesLinux(ctx, testEnv) 33 if err != nil { 34 span.SetStatus(codes.Error, err.Error()) 35 span.End() 36 shutdown(ctx) 37 panic(err) 38 } 39 40 testEnv.Print() 41 code := m.Run() 42 span.End() 43 if code != 0 { 44 span.SetStatus(codes.Error, "m.Run() exited with non-zero code") 45 } 46 shutdown(ctx) 47 os.Exit(code) 48 }