github.com/observiq/carbon@v0.9.11-0.20200820160507-1b872e368a5e/operator/helper/time_test.go (about)

     1  package helper
     2  
     3  import (
     4  	"testing"
     5  	"time"
     6  
     7  	"github.com/stretchr/testify/require"
     8  )
     9  
    10  func Test_setTimestampYear(t *testing.T) {
    11  	t.Run("Normal", func(t *testing.T) {
    12  		now = func() time.Time {
    13  			return time.Date(2020, 06, 16, 3, 31, 34, 525, time.UTC)
    14  		}
    15  
    16  		noYear := time.Date(0, 06, 16, 3, 31, 34, 525, time.UTC)
    17  		yearAdded := setTimestampYear(noYear)
    18  		expected := time.Date(2020, 06, 16, 3, 31, 34, 525, time.UTC)
    19  		require.Equal(t, expected, yearAdded)
    20  	})
    21  
    22  	t.Run("FutureOneDay", func(t *testing.T) {
    23  		now = func() time.Time {
    24  			return time.Date(2020, 01, 16, 3, 31, 34, 525, time.UTC)
    25  		}
    26  
    27  		noYear := time.Date(0, 01, 17, 3, 31, 34, 525, time.UTC)
    28  		yearAdded := setTimestampYear(noYear)
    29  		expected := time.Date(2020, 01, 17, 3, 31, 34, 525, time.UTC)
    30  		require.Equal(t, expected, yearAdded)
    31  	})
    32  
    33  	t.Run("FutureEightDays", func(t *testing.T) {
    34  		now = func() time.Time {
    35  			return time.Date(2020, 01, 16, 3, 31, 34, 525, time.UTC)
    36  		}
    37  
    38  		noYear := time.Date(0, 01, 24, 3, 31, 34, 525, time.UTC)
    39  		yearAdded := setTimestampYear(noYear)
    40  		expected := time.Date(2019, 01, 24, 3, 31, 34, 525, time.UTC)
    41  		require.Equal(t, expected, yearAdded)
    42  	})
    43  
    44  	t.Run("RolloverYear", func(t *testing.T) {
    45  		now = func() time.Time {
    46  			return time.Date(2020, 01, 01, 3, 31, 34, 525, time.UTC)
    47  		}
    48  
    49  		noYear := time.Date(0, 12, 31, 3, 31, 34, 525, time.UTC)
    50  		yearAdded := setTimestampYear(noYear)
    51  		expected := time.Date(2019, 12, 31, 3, 31, 34, 525, time.UTC)
    52  		require.Equal(t, expected, yearAdded)
    53  	})
    54  }