package convert

import (
	"testing"
	"time"

	"github.com/stretchr/testify/assert"
)

func TestTimeToString(t *testing.T) {
	tz, err := time.LoadLocation("Asia/Shanghai")
	assert.NoError(t, err)

	someTime := time.Date(2016, 9, 1, 15, 30, 0, 500000000, tz)
	assert.Equal(t, "Thu, 01 Sep 2016 07:30:00 GMT", TimeToString(someTime, RFC822))
	assert.Equal(t, "2016-09-01T07:30:00Z", TimeToString(someTime, ISO8601))
	assert.Equal(t, "2016-09-01T07:30:00.500Z", TimeToString(someTime, ISO8601Milli))
	assert.Equal(t, "01/Sep/2016:15:30:00 +0800", TimeToString(someTime, NGINXTime))
	assert.Equal(t, "01/Sep/2016:07:30:00 +0000", TimeToString(someTime.UTC(), NGINXTime))
}

func TestStringToTime(t *testing.T) {
	tz, err := time.LoadLocation("Asia/Shanghai")
	assert.NoError(t, err)
	someTime := time.Date(2016, 9, 1, 15, 30, 0, 0, tz)

	parsedTime, err := StringToTime("Thu, 01 Sep 2016 07:30:00 GMT", RFC822)
	assert.NoError(t, err)
	assert.Equal(t, someTime.UTC(), parsedTime)

	parsedTime, err = StringToTime("2016-09-01T07:30:00Z", ISO8601)
	assert.NoError(t, err)
	assert.Equal(t, someTime.UTC(), parsedTime)

	parsedTime, err = StringToTime("1472715000", ISO8601)
	assert.Error(t, err)
	assert.Equal(t, time.Time{}, parsedTime)

	someTime = time.Date(2016, 9, 1, 15, 30, 0, 500000000, tz)
	parsedTime, err = StringToTime("2016-09-01T07:30:00.500Z", ISO8601Milli)
	assert.NoError(t, err)
	assert.Equal(t, someTime.UTC(), parsedTime)
	someTime = time.Date(2016, 9, 1, 15, 30, 0, 0, tz)

	parsedTime, err = StringToTime("01/Sep/2016:15:30:00 +0800", NGINXTime)
	assert.NoError(t, err)
	assert.Equal(t, someTime.UTC(), parsedTime.UTC())

	parsedTime, err = StringToTime("01/Sep/2016:07:30:00 +0000", NGINXTime)
	assert.NoError(t, err)
	assert.Equal(t, someTime.UTC(), parsedTime.UTC())
}

func TestStringToUnixString(t *testing.T) {
	assert.Equal(t, int64(1472715000), StringToUnixTimestamp("Thu, 01 Sep 2016 07:30:00 GMT", RFC822))
	assert.Equal(t, int64(-1), StringToUnixTimestamp("2016-09-01T07:30:00.000Z", RFC822))
	assert.Equal(t, int64(1472715000), StringToUnixTimestamp("2016-09-01T07:30:00Z", ISO8601))
	assert.Equal(t, int64(1472715000), StringToUnixTimestamp("2016-09-01T07:30:00.000Z", ISO8601Milli))
	assert.Equal(t, int64(1472715000), StringToUnixTimestamp("2016-09-01T07:30:00.500Z", ISO8601Milli))
	assert.Equal(t, int64(1472715000), StringToUnixTimestamp("01/Sep/2016:15:30:00 +0800", NGINXTime))
	assert.Equal(t, int64(1472715000), StringToUnixTimestamp("01/Sep/2016:07:30:00 +0000", NGINXTime))
}

func TestTimeToUnixInt(t *testing.T) {
	tz, err := time.LoadLocation("Asia/Shanghai")
	assert.NoError(t, err)
	someTime := time.Date(2016, 9, 1, 15, 30, 0, 0, tz)

	assert.Equal(t, int64(1472715000), TimeToTimestamp(someTime))
}

func TestUnixIntToTime(t *testing.T) {
	tz, err := time.LoadLocation("Asia/Shanghai")
	assert.NoError(t, err)
	someTime := time.Date(2016, 9, 1, 15, 30, 0, 0, tz)

	assert.Equal(t, someTime.UTC(), TimestampToTime(1472715000))
}