upgrade_test.go

package commands_test

import (
	"context"
	"testing"

	"github.com/loilo-inc/canarycage/v6/cli/cage/cageapp"
	"github.com/loilo-inc/canarycage/v6/cli/cage/commands"
	"github.com/loilo-inc/canarycage/v6/mocks/mock_types"
	"github.com/loilo-inc/canarycage/v6/types"
	"github.com/stretchr/testify/assert"
	"github.com/urfave/cli/v3"
	"go.uber.org/mock/gomock"
)

func TestUpgrade(t *testing.T) {
	t.Run("Upgrade", func(t *testing.T) {
		app := &cli.Command{}
		ctrl := gomock.NewController(t)
		u := mock_types.NewMockUpgrade(ctrl)
		u.EXPECT().Upgrade(gomock.Any()).Return(nil)
		app.Commands = []*cli.Command{
			commands.Upgrade(&cageapp.UpgradeCmdInput{}, func(_Ctx context.Context, _Input *cageapp.UpgradeCmdInput) (types.Upgrade, error) {
				return u, nil
			}),
		}
		err := app.Run(context.Background(), []string{"cage", "upgrade"})
		assert.NoError(t, err)
	})
	t.Run("Upgrade with pre-release", func(t *testing.T) {
		app := &cli.Command{}
		ctrl := gomock.NewController(t)
		u := mock_types.NewMockUpgrade(ctrl)
		u.EXPECT().Upgrade(gomock.Any()).Return(nil)
		app.Commands = []*cli.Command{
			commands.Upgrade(&cageapp.UpgradeCmdInput{PreRelease: true}, func(_Ctx context.Context, _Input *cageapp.UpgradeCmdInput) (types.Upgrade, error) {
				return u, nil
			}),
		}
		err := app.Run(context.Background(), []string{"cage", "upgrade", "--pre-release"})
		assert.NoError(t, err)
	})
	t.Run("should return error when provider fails", func(t *testing.T) {
		app := &cli.Command{}
		app.Commands = []*cli.Command{
			commands.Upgrade(&cageapp.UpgradeCmdInput{}, func(_Ctx context.Context, _Input *cageapp.UpgradeCmdInput) (types.Upgrade, error) {
				return nil, assert.AnError
			}),
		}
		err := app.Run(context.Background(), []string{"cage", "upgrade"})
		assert.Equal(t, assert.AnError, err)
	})
}