default_test.go

package helium

import (
	"context"
	"testing"

	"github.com/spf13/viper"
	"github.com/stretchr/testify/require"
	"go.uber.org/atomic"
	"go.uber.org/dig"
	"go.uber.org/zap"
	"go.uber.org/zap/zaptest"

	"github.com/im-kulikov/helium/module"
	"github.com/im-kulikov/helium/service"
)

type errService struct {
	start bool
	stop  bool

	*atomic.Error
}

func (e *errService) Start(_ context.Context) error {
	if !e.start {
		return nil
	}

	return errTest
}

func (e *errService) Stop(context.Context) {
	if e.stop {
		e.Store(errTest)
	}
}

func (e errService) Name() string { return "errService" }

func TestDefaultApp(t *testing.T) {
	t.Run("create new helium with default application", func(t *testing.T) {
		ctx, cancel := context.WithCancel(context.Background())

		h, err := New(&Settings{},
			DefaultApp,
			module.New(viper.New),
			module.New(zap.NewNop),
			module.New(func() context.Context { return ctx }),
		)

		require.NotNil(t, h)
		require.NoError(t, err)

		cancel()

		require.NoError(t, h.Run())
	})

	t.Run("default application with start err", func(t *testing.T) {
		ctx, cancel := context.WithCancel(context.Background())

		h, err := New(&Settings{},
			DefaultApp,
			module.New(viper.New),
			module.New(zap.NewNop),
			module.New(func() context.Context { return ctx }),
			module.New(func() service.Service { return &errService{start: true} }, dig.Group("services")),
		)

		require.NotNil(t, h)
		require.NoError(t, err)

		require.EqualError(t, h.Run(), errTest.Error())

		cancel()
	})

	t.Run("default application with stop err", func(t *testing.T) {
		ctx, cancel := context.WithCancel(context.Background())
		svc := &errService{stop: true, Error: atomic.NewError(nil)}

		h, err := New(&Settings{},
			DefaultApp,
			module.New(viper.New),
			module.New(func() context.Context { return ctx }),
			module.New(func() *zap.Logger { return zaptest.NewLogger(t) }),
			module.New(func() service.Service { return svc }, dig.Group("services")),
		)

		require.NotNil(t, h)
		require.NoError(t, err)

		cancel()
		require.NoError(t, h.Run())
		require.EqualError(t, svc.Load(), errTest.Error())
	})
}