update tests to work properly

This commit is contained in:
James Lamb 2015-12-10 20:54:30 +11:00
parent 6e3ad123b2
commit 2cb66ad02c
2 changed files with 15 additions and 4 deletions

View File

@ -41,4 +41,3 @@ kinesis throughput using a different partition key on different hosts or host gr
## todo ## todo
* Check if the stream exists so that we have a graceful exit. * Check if the stream exists so that we have a graceful exit.
* Better client smarts to ensure that the plugin operates within the Amazon service limits.

View File

@ -1,9 +1,15 @@
package kinesis_output package kinesis_output
import ( import (
"fmt"
"testing" "testing"
"github.com/influxdb/telegraf/testutil" "github.com/aws/aws-sdk-go/aws"
"github.com/aws/aws-sdk-go/aws/credentials"
"github.com/aws/aws-sdk-go/aws/credentials/ec2rolecreds"
"github.com/aws/aws-sdk-go/aws/ec2metadata"
"github.com/aws/aws-sdk-go/aws/session"
"github.com/aws/aws-sdk-go/service/kinesis"
"github.com/stretchr/testify/require" "github.com/stretchr/testify/require"
) )
@ -15,7 +21,7 @@ func TestConnectAndWrite(t *testing.T) {
// Verify that we can connect kinesis endpoint. This test allows for a chain of credential // Verify that we can connect kinesis endpoint. This test allows for a chain of credential
// so that various authentication methods can pass depending on the system that executes. // so that various authentication methods can pass depending on the system that executes.
Config := &aws.Config{ Config := &aws.Config{
Region: aws.String(k.Region), Region: aws.String("us-west-1"),
Credentials: credentials.NewChainCredentials( Credentials: credentials.NewChainCredentials(
[]credentials.Provider{ []credentials.Provider{
&ec2rolecreds.EC2RoleProvider{Client: ec2metadata.New(session.New())}, &ec2rolecreds.EC2RoleProvider{Client: ec2metadata.New(session.New())},
@ -23,7 +29,13 @@ func TestConnectAndWrite(t *testing.T) {
&credentials.SharedCredentialsProvider{}, &credentials.SharedCredentialsProvider{},
}), }),
} }
err := kinesis.New(session.New(Config)) svc := kinesis.New(session.New(Config))
KinesisParams := &kinesis.ListStreamsInput{
Limit: aws.Int64(1)}
resp, err := svc.ListStreams(KinesisParams)
fmt.Println(resp)
require.NoError(t, err) require.NoError(t, err)
} }