From 7882505f24e76f09b7c0384ec292cca0e01cc5d1 Mon Sep 17 00:00:00 2001
From: Michael Sanders <m.r.sanders+github@gmail.com>
Date: Tue, 14 Jun 2022 15:23:55 +0100
Subject: [PATCH] Databrew: support for returning ResourceArn in databrew
 datasets (#5226)

---
 moto/databrew/models.py                       | 10 ++++++++++
 tests/test_databrew/test_databrew_datasets.py |  8 ++++++++
 2 files changed, 18 insertions(+)

diff --git a/moto/databrew/models.py b/moto/databrew/models.py
index 379091923..60976f199 100644
--- a/moto/databrew/models.py
+++ b/moto/databrew/models.py
@@ -240,6 +240,7 @@ class DataBrewBackend(BaseBackend):
 
         dataset = FakeDataset(
             self.region_name,
+            self.account_id,
             dataset_name,
             dataset_format,
             dataset_format_options,
@@ -432,6 +433,7 @@ class FakeDataset(BaseModel):
     def __init__(
         self,
         region_name,
+        account_id,
         dataset_name,
         dataset_format,
         dataset_format_options,
@@ -440,6 +442,7 @@ class FakeDataset(BaseModel):
         tags,
     ):
         self.region_name = region_name
+        self.account_id = account_id
         self.name = dataset_name
         self.format = dataset_format
         self.format_options = dataset_format_options
@@ -448,6 +451,12 @@ class FakeDataset(BaseModel):
         self.created_time = datetime.now()
         self.tags = tags
 
+    @property
+    def resource_arn(self):
+        return (
+            f"arn:aws:databrew:{self.region_name}:{self.account_id}:dataset/{self.name}"
+        )
+
     def as_dict(self):
         return {
             "Name": self.name,
@@ -457,6 +466,7 @@ class FakeDataset(BaseModel):
             "PathOptions": self.path_options,
             "CreateTime": self.created_time.isoformat(),
             "Tags": self.tags or dict(),
+            "ResourceArn": self.resource_arn,
         }
 
 
diff --git a/tests/test_databrew/test_databrew_datasets.py b/tests/test_databrew/test_databrew_datasets.py
index f2c48880a..81722324f 100644
--- a/tests/test_databrew/test_databrew_datasets.py
+++ b/tests/test_databrew/test_databrew_datasets.py
@@ -5,6 +5,7 @@ import pytest
 from botocore.exceptions import ClientError
 
 from moto import mock_databrew
+from moto.core import ACCOUNT_ID
 
 
 def _create_databrew_client():
@@ -109,6 +110,7 @@ def test_list_datasets_with_max_results():
     _create_test_datasets(client, 4)
     response = client.list_datasets(MaxResults=2)
     response["Datasets"].should.have.length_of(2)
+    response["Datasets"][0].should.have.key("ResourceArn")
     response.should.have.key("NextToken")
 
 
@@ -137,6 +139,9 @@ def test_describe_dataset():
     response = _create_test_dataset(client)
     dataset = client.describe_dataset(Name=response["Name"])
     dataset["Name"].should.equal(response["Name"])
+    dataset.should.have.key("ResourceArn").equal(
+        f"arn:aws:databrew:us-west-1:{ACCOUNT_ID}:dataset/{response['Name']}"
+    )
     # endregion
 
     # region JSON test
@@ -232,6 +237,9 @@ def test_update_dataset():
     dataset = client.describe_dataset(Name=response["Name"])
     dataset["Name"].should.equal(response["Name"])
     dataset["Format"].should.equal("TEST")
+    dataset.should.have.key("ResourceArn").equal(
+        f"arn:aws:databrew:us-west-1:{ACCOUNT_ID}:dataset/{response['Name']}"
+    )
 
 
 @mock_databrew