92 lines
3.8 KiB
Go
92 lines
3.8 KiB
Go
package timestream
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/aws/aws-sdk-go/aws"
|
|
"github.com/aws/aws-sdk-go/service/timestreamwrite"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestGetTimestreamTime(t *testing.T) {
|
|
assertions := assert.New(t)
|
|
|
|
tWithNanos := time.Date(2020, time.November, 10, 23, 44, 20, 123, time.UTC)
|
|
tWithMicros := time.Date(2020, time.November, 10, 23, 44, 20, 123000, time.UTC)
|
|
tWithMillis := time.Date(2020, time.November, 10, 23, 44, 20, 123000000, time.UTC)
|
|
tOnlySeconds := time.Date(2020, time.November, 10, 23, 44, 20, 0, time.UTC)
|
|
|
|
tUnitNanos, tValueNanos := getTimestreamTime(tWithNanos)
|
|
assertions.Equal("NANOSECONDS", tUnitNanos)
|
|
assertions.Equal("1605051860000000123", tValueNanos)
|
|
|
|
tUnitMicros, tValueMicros := getTimestreamTime(tWithMicros)
|
|
assertions.Equal("MICROSECONDS", tUnitMicros)
|
|
assertions.Equal("1605051860000123", tValueMicros)
|
|
|
|
tUnitMillis, tValueMillis := getTimestreamTime(tWithMillis)
|
|
assertions.Equal("MILLISECONDS", tUnitMillis)
|
|
assertions.Equal("1605051860123", tValueMillis)
|
|
|
|
tUnitSeconds, tValueSeconds := getTimestreamTime(tOnlySeconds)
|
|
assertions.Equal("SECONDS", tUnitSeconds)
|
|
assertions.Equal("1605051860", tValueSeconds)
|
|
}
|
|
|
|
func TestPartitionRecords(t *testing.T) {
|
|
assertions := assert.New(t)
|
|
|
|
testDatum := timestreamwrite.Record{
|
|
MeasureName: aws.String("Foo"),
|
|
MeasureValueType: aws.String("DOUBLE"),
|
|
MeasureValue: aws.String("123"),
|
|
}
|
|
|
|
var zeroDatum []*timestreamwrite.Record
|
|
oneDatum := []*timestreamwrite.Record{&testDatum}
|
|
twoDatum := []*timestreamwrite.Record{&testDatum, &testDatum}
|
|
threeDatum := []*timestreamwrite.Record{&testDatum, &testDatum, &testDatum}
|
|
|
|
assertions.Equal([][]*timestreamwrite.Record{}, partitionRecords(2, zeroDatum))
|
|
assertions.Equal([][]*timestreamwrite.Record{oneDatum}, partitionRecords(2, oneDatum))
|
|
assertions.Equal([][]*timestreamwrite.Record{oneDatum}, partitionRecords(2, oneDatum))
|
|
assertions.Equal([][]*timestreamwrite.Record{twoDatum}, partitionRecords(2, twoDatum))
|
|
assertions.Equal([][]*timestreamwrite.Record{twoDatum, oneDatum}, partitionRecords(2, threeDatum))
|
|
}
|
|
|
|
func TestConvertValueSupported(t *testing.T) {
|
|
intInputValues := []interface{}{-1, int8(-2), int16(-3), int32(-4), int64(-5)}
|
|
intOutputValues := []string{"-1", "-2", "-3", "-4", "-5"}
|
|
intOutputValueTypes := []string{"BIGINT", "BIGINT", "BIGINT", "BIGINT", "BIGINT"}
|
|
testConvertValueSupportedCases(t, intInputValues, intOutputValues, intOutputValueTypes)
|
|
|
|
uintInputValues := []interface{}{uint(1), uint8(2), uint16(3), uint32(4), uint64(5)}
|
|
uintOutputValues := []string{"1", "2", "3", "4", "5"}
|
|
uintOutputValueTypes := []string{"BIGINT", "BIGINT", "BIGINT", "BIGINT", "BIGINT"}
|
|
testConvertValueSupportedCases(t, uintInputValues, uintOutputValues, uintOutputValueTypes)
|
|
|
|
otherInputValues := []interface{}{"foo", float32(22.123), 22.1234, true}
|
|
otherOutputValues := []string{"foo", "22.123", "22.1234", "true"}
|
|
otherOutputValueTypes := []string{"VARCHAR", "DOUBLE", "DOUBLE", "BOOLEAN"}
|
|
testConvertValueSupportedCases(t, otherInputValues, otherOutputValues, otherOutputValueTypes)
|
|
}
|
|
|
|
func TestConvertValueUnsupported(t *testing.T) {
|
|
assertions := assert.New(t)
|
|
_, _, ok := convertValue(time.Date(2020, time.November, 10, 23, 44, 20, 0, time.UTC))
|
|
assertions.False(ok, "Expected unsuccessful conversion")
|
|
}
|
|
|
|
func testConvertValueSupportedCases(t *testing.T,
|
|
inputValues []interface{}, outputValues []string, outputValueTypes []string) {
|
|
assertions := assert.New(t)
|
|
for i, inputValue := range inputValues {
|
|
v, vt, ok := convertValue(inputValue)
|
|
assertions.Equal(true, ok, "Expected successful conversion")
|
|
assertions.Equal(outputValues[i], v, "Expected different string representation of converted value")
|
|
assertions.Equal(outputValueTypes[i], vt, "Expected different value type of converted value")
|
|
}
|
|
}
|