Goit

// Copyright (C) 2023, Jakob Wakeling

// All rights reserved.

package cron_test

import (

	"testing"

	"time"

	"github.com/Jamozed/Goit/src/cron"

)

func TestNext(t *testing.T) {

	t.Run("Month", func(t *testing.T) {

		schedule := cron.Schedule{6, -1, -1, -1, -1, -1}

		baseTime := time.Date(1970, 1, 1, 0, 0, 0, 0, time.UTC)

		expected := time.Date(1970, 6, 1, 0, 0, 0, 0, time.UTC)

		r := schedule.Next(baseTime)

		if r != expected {

			t.Error("Expected", expected, "got", r)

		}

	})

	t.Run("Month with Wrap", func(t *testing.T) {

		schedule := cron.Schedule{6, -1, -1, -1, -1, -1}

		baseTime := time.Date(1970, 8, 1, 0, 0, 0, 0, time.UTC)

		expected := time.Date(1971, 6, 1, 0, 0, 0, 0, time.UTC)

		r := schedule.Next(baseTime)

		if r != expected {

			t.Error("Expected", expected, "got", r)

		}

	})

	t.Run("Day", func(t *testing.T) {

		schedule := cron.Schedule{-1, 12, -1, -1, -1, -1}

		baseTime := time.Date(1970, 1, 1, 0, 0, 0, 0, time.UTC)

		expected := time.Date(1970, 1, 12, 0, 0, 0, 0, time.UTC)

		r := schedule.Next(baseTime)

		if r != expected {

			t.Error("Expected", expected, "got", r)

		}

	})

	t.Run("Day with Wrap", func(t *testing.T) {

		schedule := cron.Schedule{-1, 12, -1, -1, -1, -1}

		baseTime := time.Date(1970, 1, 24, 0, 0, 0, 0, time.UTC)

		expected := time.Date(1970, 2, 12, 0, 0, 0, 0, time.UTC)

		r := schedule.Next(baseTime)

		if r != expected {

			t.Error("Expected", expected, "got", r)

		}

	})

	t.Run("Weekday", func(t *testing.T) {

		schedule := cron.Schedule{-1, -1, 3, -1, -1, -1}

		baseTime := time.Date(1970, 1, 1, 0, 0, 0, 0, time.UTC)

		expected := time.Date(1970, 1, 7, 0, 0, 0, 0, time.UTC)

		r := schedule.Next(baseTime)

		if r != expected {

			t.Error("Expected", expected, "got", r)

		}

	})

	t.Run("Day and weekday", func(t *testing.T) {

		schedule := cron.Schedule{-1, 12, 3, -1, -1, -1}

		baseTime := time.Date(1970, 1, 1, 0, 0, 0, 0, time.UTC)

		expected := time.Date(1970, 8, 12, 0, 0, 0, 0, time.UTC)

		r := schedule.Next(baseTime)

		if r != expected {

			t.Error("Expected", expected, "got", r)

		}

	})

	t.Run("Hour", func(t *testing.T) {

		schedule := cron.Schedule{-1, -1, -1, 18, -1, -1}

		baseTime := time.Date(1970, 1, 1, 0, 0, 0, 0, time.UTC)

		expected := time.Date(1970, 1, 1, 18, 0, 0, 0, time.UTC)

		r := schedule.Next(baseTime)

		if r != expected {

			t.Error("Expected", expected, "got", r)

		}

	})

	t.Run("Minute", func(t *testing.T) {

		schedule := cron.Schedule{-1, -1, -1, -1, 30, -1}

		baseTime := time.Date(1970, 1, 1, 0, 0, 0, 0, time.UTC)

		expected := time.Date(1970, 1, 1, 0, 30, 0, 0, time.UTC)

		r := schedule.Next(baseTime)

		if r != expected {

			t.Error("Expected", expected, "got", r)

		}

	})

	t.Run("Second", func(t *testing.T) {

		schedule := cron.Schedule{-1, -1, -1, -1, -1, 30}

		baseTime := time.Date(1970, 1, 1, 0, 0, 0, 0, time.UTC)

		expected := time.Date(1970, 1, 1, 0, 0, 30, 0, time.UTC)

		r := schedule.Next(baseTime)

		if r != expected {

			t.Error("Expected", expected, "got", r)

		}

	})

	t.Run("All", func(t *testing.T) {

		schedule := cron.Schedule{3, 6, 2, 6, 45, 15}

		baseTime := time.Date(1970, 1, 1, 0, 0, 0, 0, time.UTC)

		expected := time.Date(1973, 3, 6, 6, 45, 15, 0, time.UTC)

		r := schedule.Next(baseTime)

		if r != expected {

			t.Error("Expected", expected, "got", r)

		}

	})

	t.Run("Immediate", func(t *testing.T) {

		schedule := cron.Schedule{-1, -1, -1, -1, -1, -1}

		baseTime := time.Date(1970, 1, 1, 0, 0, 0, 0, time.UTC)

		expected := time.Date(1970, 1, 1, 0, 0, 1, 0, time.UTC)

		r := schedule.Next(baseTime)

		if r != expected {

			t.Error("Expected", expected, "got", r)

		}

	})

}