Fix timestream endpoint tests (#1781)

Author: LeonLuttenberger

awswrangler/_config.py

@@ -408,7 +408,12 @@ def secretsmanager_endpoint_url(self, value: Optional[str]) -> None:
 
     @property
     def timestream_query_endpoint_url(self) -> Optional[str]:
-        """Property timestream_query_endpoint_url."""
+        """
+        Property timestream_query_endpoint_url.
+
+        Before setting this endpoint, consult the documentation for DescribeEndpoints.
+        https://docs.aws.amazon.com/timestream/latest/developerguide/API_DescribeEndpoints.html
+        """
         return cast(Optional[str], self["timestream_query_endpoint_url"])
 
     @timestream_query_endpoint_url.setter
@@ -417,7 +422,12 @@ def timestream_query_endpoint_url(self, value: Optional[str]) -> None:
 
     @property
     def timestream_write_endpoint_url(self) -> Optional[str]:
-        """Property timestream_write_endpoint_url."""
+        """
+        Property timestream_write_endpoint_url.
+
+        Before setting this endpoint, consult the documentation for DescribeEndpoints.
+        https://docs.aws.amazon.com/timestream/latest/developerguide/API_DescribeEndpoints.html
+        """
         return cast(Optional[str], self["timestream_write_endpoint_url"])
 
     @timestream_write_endpoint_url.setter

tests/unit/test_config.py

@@ -119,16 +119,12 @@ def test_basics(path, glue_database, glue_table, workgroup0, workgroup1):
     wr.config.athena_endpoint_url = f"https://athena.{region}.amazonaws.com"
     wr.config.glue_endpoint_url = f"https://glue.{region}.amazonaws.com"
     wr.config.secretsmanager_endpoint_url = f"https://secretsmanager.{region}.amazonaws.com"
-    wr.config.timestream_write_endpoint_url = f"https://ingest.timestream.{region}.amazonaws.com"
-    wr.config.timestream_query_endpoint_url = f"https://query.timestream.{region}.amazonaws.com"
     _urls_test(glue_database)
     os.environ["WR_STS_ENDPOINT_URL"] = f"https://sts.{region}.amazonaws.com"
     os.environ["WR_S3_ENDPOINT_URL"] = f"https://s3.{region}.amazonaws.com"
     os.environ["WR_ATHENA_ENDPOINT_URL"] = f"https://athena.{region}.amazonaws.com"
     os.environ["WR_GLUE_ENDPOINT_URL"] = f"https://glue.{region}.amazonaws.com"
     os.environ["WR_SECRETSMANAGER_ENDPOINT_URL"] = f"https://secretsmanager.{region}.amazonaws.com"
-    os.environ["WR_TIMESTREAM_WRITE_ENDPOINT_URL"] = f"https://ingest.timestream.{region}.amazonaws.com"
-    os.environ["WR_TIMESTREAM_QUERY_ENDPOINT_URL"] = f"https://query.timestream.{region}.amazonaws.com"
     wr.config.reset()
     _urls_test(glue_database)