diff --git a/docs/docs/services/timestream-write.rst b/docs/docs/services/timestream-write.rst index 242230a2d..7c18901ca 100644 --- a/docs/docs/services/timestream-write.rst +++ b/docs/docs/services/timestream-write.rst @@ -12,6 +12,8 @@ timestream-write ================ +.. autoclass:: moto.timestreamwrite.models.TimestreamWriteBackend + |start-h3| Example usage |end-h3| .. sourcecode:: python diff --git a/moto/timestreamwrite/models.py b/moto/timestreamwrite/models.py index 1aa8f5dde..c14b8b46e 100644 --- a/moto/timestreamwrite/models.py +++ b/moto/timestreamwrite/models.py @@ -104,6 +104,19 @@ class TimestreamDatabase(BaseModel): class TimestreamWriteBackend(BaseBackend): + """ + When using the decorators, you can use the following internal API to verify records have arrived: + + .. sourcecode:: python + + from moto.core import DEFAULT_ACCOUNT_ID as ACCOUNT_ID + from moto.timestreamwrite.models import timestreamwrite_backends + + backend = timestreamwrite_backends[ACCOUNT_ID]["us-east-1"] + records = backend.databases["mydatabase"].tables["mytable"].records + + """ + def __init__(self, region_name, account_id): super().__init__(region_name, account_id) self.databases = dict()