fix SQS API spec rename of Integer to NullableInteger

alexrashed · alexrashed · commit 49b84a7079e3 · 2023-11-13T09:29:01.000+01:00
diff --git a/localstack/services/sqs/provider.py b/localstack/services/sqs/provider.py
@@ -30,7 +30,6 @@
     EmptyBatchRequest,
     GetQueueAttributesResult,
     GetQueueUrlResult,
-    Integer,
     InvalidAttributeName,
     InvalidBatchEntryId,
     InvalidMessageContents,
@@ -42,6 +41,7 @@
     MessageBodyAttributeMap,
     MessageBodySystemAttributeMap,
     MessageSystemAttributeName,
+    NullableInteger,
     PurgeQueueInProgress,
     QueueAttributeMap,
     QueueAttributeName,
@@ -762,7 +762,7 @@ def change_message_visibility(
         context: RequestContext,
         queue_url: String,
         receipt_handle: String,
-        visibility_timeout: Integer,
+        visibility_timeout: NullableInteger,
     ) -> None:
         queue = self._resolve_queue(context, queue_url=queue_url)
         queue.update_visibility_timeout(receipt_handle, visibility_timeout)
@@ -859,7 +859,7 @@ def send_message(
         context: RequestContext,
         queue_url: String,
         message_body: String,
-        delay_seconds: Integer = None,
+        delay_seconds: NullableInteger = None,
         message_attributes: MessageBodyAttributeMap = None,
         message_system_attributes: MessageBodySystemAttributeMap = None,
         message_deduplication_id: String = None,
@@ -955,7 +955,7 @@ def _put_message(
         queue: SqsQueue,
         context: RequestContext,
         message_body: String,
-        delay_seconds: Integer = None,
+        delay_seconds: NullableInteger = None,
         message_attributes: MessageBodyAttributeMap = None,
         message_system_attributes: MessageBodySystemAttributeMap = None,
         message_deduplication_id: String = None,
@@ -993,9 +993,9 @@ def receive_message(
         queue_url: String,
         attribute_names: AttributeNameList = None,
         message_attribute_names: MessageAttributeNameList = None,
-        max_number_of_messages: Integer = None,
-        visibility_timeout: Integer = None,
-        wait_time_seconds: Integer = None,
+        max_number_of_messages: NullableInteger = None,
+        visibility_timeout: NullableInteger = None,
+        wait_time_seconds: NullableInteger = None,
         receive_request_attempt_id: String = None,
     ) -> ReceiveMessageResult:
         queue = self._resolve_queue(context, queue_url=queue_url)
