auto commit

Author: rishav-karanjit

Examples/runtimes/go/migration/PlaintextToAWSDBE/awsdbe/migrationexamplestep2.go

@@ -50,9 +50,9 @@ func MigrationStep2(kmsKeyID, ddbTableName, partitionKeyValue, sortKeyValue stri
 	// you must use the plaintext override `FORBID_PLAINTEXT_WRITE_ALLOW_PLAINTEXT_READ`.
 	// This plaintext override means:
 	// 	- Write: Items are forbidden to be written as plaintext.
-	//  	Items will be written as encrypted items.
+	//  	     Items will be written as encrypted items.
 	//  - Read: Items are allowed to be read as plaintext.
-	//  	Items are allowed to be read as encrypted items.
+	//  	    Items are allowed to be read as encrypted items.
 
 	listOfTableConfigs := configureTable(kmsKeyID, ddbTableName, dbesdkdynamodbencryptiontypes.PlaintextOverrideForbidPlaintextWriteAllowPlaintextRead)
 

Examples/runtimes/go/migration/PlaintextToAWSDBE/awsdbe/migrationexamplestep2_test.go

@@ -0,0 +1,34 @@
+package awsdbe
+
+import (
+	"testing"
+
+	"github.com/aws/aws-database-encryption-sdk-dynamodb/releases/go/dynamodb-esdk/examples/migration/PlaintextToAWSDBE/plaintext"
+	"github.com/aws/aws-database-encryption-sdk-dynamodb/releases/go/dynamodb-esdk/examples/utils"
+	"github.com/google/uuid"
+)
+
+func TestMigrationStep2(t *testing.T) {
+	kmsKeyID := utils.KmsKeyID()
+	tableName := utils.DdbTableName()
+	partitionKey := uuid.New().String()
+	sortKeys := []string{"0", "1", "2"}
+
+	// Successfully executes Step 2
+	MigrationStep2(kmsKeyID, tableName, partitionKey, sortKeys[2])
+
+	// Given: Step 0 has succeeded
+	plaintext.MigrationStep0(tableName, partitionKey, sortKeys[0])
+	// When: Execute Step 2 with sortReadValue=0, Then: Success (i.e. can read plaintext values)
+	MigrationStep2(kmsKeyID, tableName, partitionKey, sortKeys[0])
+
+	// Given: Step 1 has succeeded
+	MigrationStep1(kmsKeyID, tableName, partitionKey, sortKeys[1])
+	// When: Execute Step 2 with sortReadValue=1, Then: Success (i.e. can read encrypted values)
+	MigrationStep2(kmsKeyID, tableName, partitionKey, sortKeys[1])
+
+	// Cleanup
+	for _, sortKey := range sortKeys {
+		utils.DeleteItem(tableName, "partition_key", partitionKey, "sort_key", sortKey)
+	}
+}