Testing new SB properties (#1425)

* added extra sb properties to test

* json format datetime

* json decode all datetime props

* testing for missed properties

* fixing eg and eh tests

* fixing eg tests

* parsing datetime

* missing comma

* lint

* isoformat

* moved isoformat

* revert eg changes

* comma

* just sb

* flake

* checking properties

---------

Co-authored-by: gavin-aguiar <[email protected]>

Author: hallvictoria

tests/endtoend/servicebus_functions/servicebus_functions_stein/function_app.py

@@ -50,6 +50,24 @@ def servicebus_trigger(msg: func.ServiceBusMessage) -> str:
         'time_to_live': msg.time_to_live,
         'to': msg.to,
         'user_properties': msg.user_properties,
+
+        'application_properties': msg.application_properties,
+        'correlation_id': msg.correlation_id,
+        'dead_letter_error_description': msg.dead_letter_error_description,
+        'dead_letter_reason': msg.dead_letter_reason,
+        'dead_letter_source': msg.dead_letter_source,
+        'enqueued_sequence_number': msg.enqueued_sequence_number,
+        'enqueued_time_utc': (msg.enqueued_time_utc.isoformat() if
+                              msg.enqueued_time_utc else None),
+        'expires_at_utc': (msg.expires_at_utc.isoformat() if
+                           msg.expires_at_utc else None),
+        'locked_until': (msg.locked_until.isoformat() if
+                         msg.locked_until else None),
+        'lock_token': msg.lock_token,
+        'sequence_number': msg.sequence_number,
+        'state': msg.state,
+        'subject': msg.subject,
+        'transaction_partition_key': msg.transaction_partition_key
     })
 
     return result

tests/endtoend/servicebus_functions/servicebus_functions_stein/generic/function_app.py

@@ -58,6 +58,24 @@ def servicebus_trigger(msg: func.ServiceBusMessage) -> str:
         'time_to_live': msg.time_to_live,
         'to': msg.to,
         'user_properties': msg.user_properties,
+
+        'application_properties': msg.application_properties,
+        'correlation_id': msg.correlation_id,
+        'dead_letter_error_description': msg.dead_letter_error_description,
+        'dead_letter_reason': msg.dead_letter_reason,
+        'dead_letter_source': msg.dead_letter_source,
+        'enqueued_sequence_number': msg.enqueued_sequence_number,
+        'enqueued_time_utc': (msg.enqueued_time_utc.isoformat() if
+                              msg.enqueued_time_utc else None),
+        'expires_at_utc': (msg.expires_at_utc.isoformat() if
+                           msg.expires_at_utc else None),
+        'locked_until': (msg.locked_until.isoformat() if
+                         msg.locked_until else None),
+        'lock_token': msg.lock_token,
+        'sequence_number': msg.sequence_number,
+        'state': msg.state,
+        'subject': msg.subject,
+        'transaction_partition_key': msg.transaction_partition_key
     })
 
     return result

tests/endtoend/servicebus_functions/servicebus_trigger/__init__.py

@@ -23,6 +23,24 @@ def main(msg: azf.ServiceBusMessage) -> str:
         'time_to_live': msg.time_to_live,
         'to': msg.to,
         'user_properties': msg.user_properties,
+
+        'application_properties': msg.application_properties,
+        'correlation_id': msg.correlation_id,
+        'dead_letter_error_description': msg.dead_letter_error_description,
+        'dead_letter_reason': msg.dead_letter_reason,
+        'dead_letter_source': msg.dead_letter_source,
+        'enqueued_sequence_number': msg.enqueued_sequence_number,
+        'enqueued_time_utc': (msg.enqueued_time_utc.isoformat() if
+                              msg.enqueued_time_utc else None),
+        'expires_at_utc': (msg.expires_at_utc.isoformat() if
+                           msg.expires_at_utc else None),
+        'locked_until': (msg.locked_until.isoformat() if
+                         msg.locked_until else None),
+        'lock_token': msg.lock_token,
+        'sequence_number': msg.sequence_number,
+        'state': msg.state,
+        'subject': msg.subject,
+        'transaction_partition_key': msg.transaction_partition_key
     })
 
     return result

tests/endtoend/test_servicebus_functions.py

@@ -31,6 +31,17 @@ def test_servicebus_basic(self):
                 self.assertEqual(r.status_code, 200)
                 msg = r.json()
                 self.assertEqual(msg['body'], data)
+                for attr in {'message_id', 'body', 'content_type', 'delivery_count',
+                             'expiration_time', 'label', 'partition_key', 'reply_to',
+                             'reply_to_session_id', 'scheduled_enqueue_time',
+                             'session_id', 'time_to_live', 'to', 'user_properties',
+                             'application_properties', 'correlation_id',
+                             'dead_letter_error_description', 'dead_letter_reason',
+                             'dead_letter_source', 'enqueued_sequence_number',
+                             'enqueued_time_utc', 'expires_at_utc', 'locked_until',
+                             'lock_token', 'sequence_number', 'state', 'subject',
+                             'transaction_partition_key'}:
+                    self.assertIn(attr, msg)
             except (AssertionError, json.JSONDecodeError):
                 if try_no == max_retries - 1:
                     raise