From 09a2d1d58ca13358812f529b4937686388964b3e Mon Sep 17 00:00:00 2001
From: Angela Nguyen <3279569+angethuy@users.noreply.github.com>
Date: Wed, 12 Jul 2023 14:23:17 -0700
Subject: [PATCH] [backend] add account field to aws mock for ddb table (#2728)

---
 backend/mock/service/awsmock/awsmock.go | 6 ++++--
 backend/module/dynamodb/aws_test.go     | 1 +
 2 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/backend/mock/service/awsmock/awsmock.go b/backend/mock/service/awsmock/awsmock.go
index 29a667de7a..18962fdadf 100644
--- a/backend/mock/service/awsmock/awsmock.go
+++ b/backend/mock/service/awsmock/awsmock.go
@@ -128,8 +128,9 @@ func (s *svc) S3GetBucketPolicy(ctx context.Context, account, region, bucket, ac
 
 func (s *svc) DescribeTable(ctx context.Context, account, region, tableName string) (*dynamodbv1.Table, error) {
 	ret := &dynamodbv1.Table{
-		Name:   tableName,
-		Region: region,
+		Name:    tableName,
+		Region:  region,
+		Account: "default",
 		ProvisionedThroughput: &dynamodbv1.Throughput{
 			ReadCapacityUnits:  100,
 			WriteCapacityUnits: 200,
@@ -194,6 +195,7 @@ func (s *svc) UpdateCapacity(ctx context.Context, account, region, tableName str
 	ret := &dynamodbv1.Table{
 		Name:                   tableName,
 		Region:                 region,
+		Account:                "default",
 		ProvisionedThroughput:  currentThroughput,
 		Status:                 dynamodbv1.Table_Status(3),
 		GlobalSecondaryIndexes: gsis,
diff --git a/backend/module/dynamodb/aws_test.go b/backend/module/dynamodb/aws_test.go
index 4a10c92c70..4785128c98 100644
--- a/backend/module/dynamodb/aws_test.go
+++ b/backend/module/dynamodb/aws_test.go
@@ -51,6 +51,7 @@ var g = []*dynamodbv1.GlobalSecondaryIndex{
 var testTable = &dynamodbv1.Table{
 	Name:                   "",
 	Region:                 "",
+	Account:                "default",
 	ProvisionedThroughput:  pt,
 	GlobalSecondaryIndexes: g,
 	Status:                 dynamodbv1.Table_Status(3),