SimpleDB: implement domain_metadata

Signed-off-by: Yves Bastide <stid@acm.org>

Author: ybastide

moto/sdb/models.py

@@ -99,5 +99,30 @@ def put_attributes(
         domain = self._get_domain(domain_name)
         domain.put(item_name, attributes)
 
+    def domain_metadata(self, domain_name: str) -> Dict[str, Any]:
+        self._validate_domain_name(domain_name)
+        domain = self._get_domain(domain_name)
+        item_names_size_bytes = 0
+        attribute_name_count = 0
+        attribute_names_size_bytes = 0
+        attribute_value_count = 0
+        attribute_values_size_bytes = 0
+        for name, item in domain.items.items():
+            item_names_size_bytes += len(name)
+            with item.lock:
+                for attr in item.attributes:
+                    attribute_name_count += 1
+                    attribute_names_size_bytes += len(attr["name"])
+                    attribute_value_count += 1
+                    attribute_values_size_bytes += len(attr["value"])
+        return {
+            "item_count": len(domain.items),
+            "item_names_size_bytes": item_names_size_bytes,
+            "attribute_name_count": attribute_name_count,
+            "attribute_names_size_bytes": attribute_names_size_bytes,
+            "attribute_value_count": attribute_value_count,
+            "attribute_values_size_bytes": attribute_values_size_bytes,
+        }
+
 
 sdb_backends = BackendDict(SimpleDBBackend, "sdb")

moto/sdb/responses.py

@@ -50,6 +50,12 @@ def put_attributes(self) -> str:
         template = self.response_template(PUT_ATTRIBUTES_TEMPLATE)
         return template.render()
 
+    def domain_metadata(self) -> str:
+        domain_name = self._get_param("DomainName")
+        metadata = self.sdb_backend.domain_metadata(domain_name=domain_name)
+        template = self.response_template(DOMAIN_METADATA_TEMPLATE)
+        return template.render(metadata=metadata)
+
 
 CREATE_DOMAIN_TEMPLATE = """<?xml version="1.0" encoding="UTF-8"?>
 <CreateDomainResult  xmlns="http://sdb.amazonaws.com/doc/2009-04-15/"></CreateDomainResult>
@@ -92,3 +98,20 @@ def put_attributes(self) -> str:
 {% endfor %}
   </GetAttributesResult>
 </GetAttributesResponse>"""
+
+
+DOMAIN_METADATA_TEMPLATE = """<DomainMetadataResponse xmlns="http://sdb.amazonaws.com/doc/2009-04-15/">
+  <DomainMetadataResult>
+    <ItemCount>{{ metadata.item_count }}</ItemCount>
+    <ItemNamesSizeBytes>{{ metadata.item_names_size_bytes }}</ItemNamesSizeBytes>
+    <AttributeNameCount >{{ metadata.attribute_name_count }}</AttributeNameCount >
+    <AttributeNamesSizeBytes>{{ metadata.attribute_names_size_bytes }}</AttributeNamesSizeBytes>
+    <AttributeValueCount>{{ metadata.attribute_value_count }}</AttributeValueCount>
+    <AttributeValuesSizeBytes>{{ metadata.attribute_values_size_bytes }}</AttributeValuesSizeBytes>
+    <Timestamp>1225486466</Timestamp>
+  </DomainMetadataResult>
+  <ResponseMetadata>
+    	<RequestId>b1e8f1f7-42e9-494c-ad09-2674e557526d</RequestId>
+    	<BoxUsage>0.0000219907</BoxUsage>
+  </ResponseMetadata>
+</DomainMetadataResponse>"""

tests/test_sdb/test_sdb_domains.py

@@ -1,3 +1,5 @@
+from unittest.mock import ANY
+
 import boto3
 import pytest
 from botocore.exceptions import ClientError
@@ -65,3 +67,25 @@ def test_delete_domain_invalid():
     assert err["Code"] == "InvalidParameterValue"
     assert err["Message"] == "Value (a) for parameter DomainName is invalid. "
     assert "BoxUsage" in err
+
+
+@mock_aws
+def test_domain_metadata():
+    name = "mydomain"
+    sdb = boto3.client("sdb", region_name="eu-west-1")
+    sdb.create_domain(DomainName=name)
+    sdb.put_attributes(
+        DomainName=name, ItemName="asdf", Attributes=[{"Name": "abc", "Value": "de"}]
+    )
+    metadata = sdb.domain_metadata(DomainName=name)
+    expected = {
+        "ItemCount": 1,
+        "ItemNamesSizeBytes": 4,
+        "AttributeNameCount": 1,
+        "AttributeNamesSizeBytes": 3,
+        "AttributeValueCount": 1,
+        "AttributeValuesSizeBytes": 2,
+        "Timestamp": ANY,
+        "ResponseMetadata": ANY,
+    }
+    assert metadata == expected