availability_test.go

package hour_test

import (
	"testing"

	"github.com/ThreeDotsLabs/wild-workouts-go-ddd-example/internal/trainer/domain/hour"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

func TestHour_MakeNotAvailable(t *testing.T) {
	t.Parallel()
	h, err := testHourFactory.NewAvailableHour(validTrainingHour())
	require.NoError(t, err)

	require.NoError(t, h.MakeNotAvailable())
	assert.False(t, h.IsAvailable())
}

func TestHour_MakeNotAvailable_with_scheduled_training(t *testing.T) {
	t.Parallel()
	h := newHourWithScheduledTraining(t)

	assert.Equal(t, hour.ErrTrainingScheduled, h.MakeNotAvailable())
}

func TestHour_MakeAvailable(t *testing.T) {
	t.Parallel()
	h, err := testHourFactory.NewAvailableHour(validTrainingHour())
	require.NoError(t, err)

	require.NoError(t, h.MakeNotAvailable())

	require.NoError(t, h.MakeAvailable())
	assert.True(t, h.IsAvailable())
}

func TestHour_MakeAvailable_with_scheduled_training(t *testing.T) {
	t.Parallel()
	h := newHourWithScheduledTraining(t)

	assert.Equal(t, hour.ErrTrainingScheduled, h.MakeAvailable())
}

func TestHour_ScheduleTraining(t *testing.T) {
	t.Parallel()
	h, err := testHourFactory.NewAvailableHour(validTrainingHour())
	require.NoError(t, err)

	require.NoError(t, h.ScheduleTraining())

	assert.True(t, h.HasTrainingScheduled())
	assert.False(t, h.IsAvailable())
}

func TestHour_ScheduleTraining_with_not_available(t *testing.T) {
	t.Parallel()
	h := newNotAvailableHour(t)
	assert.Equal(t, hour.ErrHourNotAvailable, h.ScheduleTraining())
}

func TestHour_CancelTraining(t *testing.T) {
	t.Parallel()
	h := newHourWithScheduledTraining(t)

	require.NoError(t, h.CancelTraining())

	assert.False(t, h.HasTrainingScheduled())
	assert.True(t, h.IsAvailable())
}

func TestHour_CancelTraining_no_training_scheduled(t *testing.T) {
	t.Parallel()
	h, err := testHourFactory.NewAvailableHour(validTrainingHour())
	require.NoError(t, err)

	assert.Equal(t, hour.ErrNoTrainingScheduled, h.CancelTraining())
}

func TestNewAvailabilityFromString(t *testing.T) {
	t.Parallel()
	testCases := []hour.Availability{
		hour.Available,
		hour.NotAvailable,
		hour.TrainingScheduled,
	}

	for _, tc := range testCases {
		expectedAvailability := tc
		t.Run(expectedAvailability.String(), func(t *testing.T) {
			t.Parallel()
			availability, err := hour.NewAvailabilityFromString(expectedAvailability.String())
			require.NoError(t, err)

			assert.Equal(t, expectedAvailability, availability)
		})
	}
}

func TestNewAvailabilityFromString_invalid(t *testing.T) {
	t.Parallel()
	_, err := hour.NewAvailabilityFromString("invalid_value")
	assert.Error(t, err)

	_, err = hour.NewAvailabilityFromString("")
	assert.Error(t, err)
}

func newHourWithScheduledTraining(t *testing.T) *hour.Hour {
	h, err := testHourFactory.NewAvailableHour(validTrainingHour())
	require.NoError(t, err)

	require.NoError(t, h.ScheduleTraining())

	return h
}

func newNotAvailableHour(t *testing.T) *hour.Hour {
	h, err := testHourFactory.NewAvailableHour(validTrainingHour())
	require.NoError(t, err)

	require.NoError(t, h.MakeNotAvailable())

	return h
}