78 lines
1.8 KiB
Go
78 lines
1.8 KiB
Go
|
package kinesis
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/influxdata/telegraf/testutil"
|
||
|
|
uuid "github.com/satori/go.uuid"
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestPartitionKey(t *testing.T) {
|
||
|
|
|
||
|
|
assert := assert.New(t)
|
||
|
|
testPoint := testutil.TestMetric(1)
|
||
|
|
|
||
|
|
k := KinesisOutput{
|
||
|
|
Partition: &Partition{
|
||
|
|
Method: "static",
|
||
|
|
Key: "-",
|
||
|
|
},
|
||
|
|
}
|
||
|
|
assert.Equal("-", k.getPartitionKey(testPoint), "PartitionKey should be '-'")
|
||
|
|
|
||
|
|
k = KinesisOutput{
|
||
|
|
Partition: &Partition{
|
||
|
|
Method: "tag",
|
||
|
|
Key: "tag1",
|
||
|
|
},
|
||
|
|
}
|
||
|
|
assert.Equal(testPoint.Tags()["tag1"], k.getPartitionKey(testPoint), "PartitionKey should be value of 'tag1'")
|
||
|
|
|
||
|
|
k = KinesisOutput{
|
||
|
|
Partition: &Partition{
|
||
|
|
Method: "tag",
|
||
|
|
Key: "doesnotexist",
|
||
|
|
},
|
||
|
|
}
|
||
|
|
assert.Equal("", k.getPartitionKey(testPoint), "PartitionKey should be value of ''")
|
||
|
|
|
||
|
|
k = KinesisOutput{
|
||
|
|
Partition: &Partition{
|
||
|
|
Method: "not supported",
|
||
|
|
},
|
||
|
|
}
|
||
|
|
assert.Equal("", k.getPartitionKey(testPoint), "PartitionKey should be value of ''")
|
||
|
|
|
||
|
|
k = KinesisOutput{
|
||
|
|
Partition: &Partition{
|
||
|
|
Method: "measurement",
|
||
|
|
},
|
||
|
|
}
|
||
|
|
assert.Equal(testPoint.Name(), k.getPartitionKey(testPoint), "PartitionKey should be value of measurement name")
|
||
|
|
|
||
|
|
k = KinesisOutput{
|
||
|
|
Partition: &Partition{
|
||
|
|
Method: "random",
|
||
|
|
},
|
||
|
|
}
|
||
|
|
partitionKey := k.getPartitionKey(testPoint)
|
||
|
|
u, err := uuid.FromString(partitionKey)
|
||
|
|
assert.Nil(err, "Issue parsing UUID")
|
||
|
|
assert.Equal(uint(4), u.Version(), "PartitionKey should be UUIDv4")
|
||
|
|
|
||
|
|
k = KinesisOutput{
|
||
|
|
PartitionKey: "-",
|
||
|
|
}
|
||
|
|
assert.Equal("-", k.getPartitionKey(testPoint), "PartitionKey should be '-'")
|
||
|
|
|
||
|
|
k = KinesisOutput{
|
||
|
|
RandomPartitionKey: true,
|
||
|
|
}
|
||
|
|
partitionKey = k.getPartitionKey(testPoint)
|
||
|
|
u, err = uuid.FromString(partitionKey)
|
||
|
|
assert.Nil(err, "Issue parsing UUID")
|
||
|
|
assert.Equal(uint(4), u.Version(), "PartitionKey should be UUIDv4")
|
||
|
|
|
||
|
|
}
|