TestMain final form

testcontainers_test.go

func TestMain(m *testing.M) {
	var code = 1
	defer func() { os.Exit(code) }()

	ctx := context.Background()
	mongoContainer, err := setup(ctx, mongoDB)
	if err != nil {
		log.Printf("Unexpected error, fallback to mem repository implementations.\nerror: %s.\n", err)
		allGames = mem.NewAllGames()
	} else {
		ctx, cancel := context.WithTimeout(ctx, 10*time.Second)
		defer cancel()
		client, err := mdriver.Connect(ctx, options.Client().ApplyURI(mongoContainer.URI))
		if err != nil {
			log.Printf("Could not connect to mongodb, fallback to mem repository implementations.\nerror: %s.\n", err)
			allGames = mem.NewAllGames()
		} else {
			allGames = mongo.NewAllGames(client)
		}
		defer func() {
			if err = client.Disconnect(ctx); err != nil {
				panic(err)
			}
		}()
	}

	code = m.Run()
}