From 277de772565536b8c37b92206f2a60dc2b05be0e Mon Sep 17 00:00:00 2001
From: Paul Seiffert <paul.seiffert@jimdo.com>
Date: Sat, 19 Dec 2015 17:34:36 +0100
Subject: [PATCH] Add tests for DynamoDB backend

---
 physical/dynamodb_test.go | 113 ++++++++++++++++++++++++++++++++++++++
 1 file changed, 113 insertions(+)
 create mode 100644 physical/dynamodb_test.go

diff --git a/physical/dynamodb_test.go b/physical/dynamodb_test.go
new file mode 100644
index 000000000..5f6632450
--- /dev/null
+++ b/physical/dynamodb_test.go
@@ -0,0 +1,113 @@
+package physical
+
+import (
+	"fmt"
+	"math/rand"
+	"os"
+	"testing"
+	"time"
+
+	"github.com/aws/aws-sdk-go/aws"
+	"github.com/aws/aws-sdk-go/aws/credentials"
+	"github.com/aws/aws-sdk-go/aws/session"
+	"github.com/aws/aws-sdk-go/service/dynamodb"
+)
+
+func TestDynamoDBBackend(t *testing.T) {
+	if os.Getenv("AWS_ACCESS_KEY_ID") == "" || os.Getenv("AWS_SECRET_ACCESS_KEY") == "" {
+		t.SkipNow()
+	}
+
+	creds, err := credentials.NewEnvCredentials().Get()
+	if err != nil {
+		t.Fatalf("err: %v", err)
+	}
+
+	// If the variable is empty or doesn't exist, the default
+	// AWS endpoints will be used
+	endpoint := os.Getenv("AWS_DYNAMODB_ENDPOINT")
+
+	region := os.Getenv("AWS_DEFAULT_REGION")
+	if region == "" {
+		region = "us-east-1"
+	}
+
+	conn := dynamodb.New(session.New(&aws.Config{
+		Credentials: credentials.NewEnvCredentials(),
+		Endpoint:    aws.String(endpoint),
+		Region:      aws.String(region),
+	}))
+
+	var randInt = rand.New(rand.NewSource(time.Now().UnixNano())).Int()
+	table := fmt.Sprintf("vault-dynamodb-testacc-%d", randInt)
+
+	defer func() {
+		conn.DeleteTable(&dynamodb.DeleteTableInput{
+			TableName: aws.String(table),
+		})
+	}()
+
+	b, err := NewBackend("dynamodb", map[string]string{
+		"access_key":    creds.AccessKeyID,
+		"secret_key":    creds.SecretAccessKey,
+		"session_token": creds.SessionToken,
+		"table":         table,
+	})
+	if err != nil {
+		t.Fatalf("err: %s", err)
+	}
+
+	testBackend(t, b)
+	testBackend_ListPrefix(t, b)
+}
+
+func TestDynamoDBHABackend(t *testing.T) {
+	if os.Getenv("AWS_ACCESS_KEY_ID") == "" || os.Getenv("AWS_SECRET_ACCESS_KEY") == "" {
+		t.SkipNow()
+	}
+
+	creds, err := credentials.NewEnvCredentials().Get()
+	if err != nil {
+		t.Fatalf("err: %v", err)
+	}
+
+	// If the variable is empty or doesn't exist, the default
+	// AWS endpoints will be used
+	endpoint := os.Getenv("AWS_DYNAMODB_ENDPOINT")
+
+	region := os.Getenv("AWS_DEFAULT_REGION")
+	if region == "" {
+		region = "us-east-1"
+	}
+
+	conn := dynamodb.New(session.New(&aws.Config{
+		Credentials: credentials.NewEnvCredentials(),
+		Endpoint:    aws.String(endpoint),
+		Region:      aws.String(region),
+	}))
+
+	var randInt = rand.New(rand.NewSource(time.Now().UnixNano())).Int()
+	table := fmt.Sprintf("vault-dynamodb-testacc-%d", randInt)
+
+	defer func() {
+		conn.DeleteTable(&dynamodb.DeleteTableInput{
+			TableName: aws.String(table),
+		})
+	}()
+
+	b, err := NewBackend("dynamodb", map[string]string{
+		"access_key":    creds.AccessKeyID,
+		"secret_key":    creds.SecretAccessKey,
+		"session_token": creds.SessionToken,
+		"table":         table,
+	})
+	if err != nil {
+		t.Fatalf("err: %s", err)
+	}
+
+	ha, ok := b.(HABackend)
+	if !ok {
+		t.Fatalf("dynamodb does not implement HABackend")
+	}
+	testHABackend(t, ha, ha)
+}