Testing new SB properties (Azure#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 <80794152+gavin-aguiar@users.noreply.github.com>

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