Add Support for mutable_content in aioapns

push_notifications/apns_async.py

@@ -222,6 +222,7 @@ def apns_send_message(
 	loc_key: str = None,
 	priority: int = None,
 	collapse_id: str = None,
+	mutable_content: int = None,
 	err_func: ErrFunc = None,
 ):
 	"""
@@ -253,6 +254,7 @@ def apns_send_message(
 		loc_key=loc_key,
 		priority=priority,
 		collapse_id=collapse_id,
+		mutable_content=mutable_content,
 		err_func=err_func,
 	)
 
@@ -277,6 +279,7 @@ def apns_send_bulk_message(
 	loc_key: str = None,
 	priority: int = None,
 	collapse_id: str = None,
+	mutable_content: int = None,
 	err_func: ErrFunc = None,
 ):
 	"""
@@ -311,6 +314,7 @@ def apns_send_bulk_message(
 			loc_key=loc_key,
 			priority=priority,
 			collapse_id=collapse_id,
+			mutable_content=mutable_content,
 			err_func=err_func,
 		))
 
@@ -355,12 +359,17 @@ async def _send_bulk_request(
 	loc_key: str = None,
 	priority: int = None,
 	collapse_id: str = None,
+	mutable_content: str = None,
 	err_func: ErrFunc = None,
 ):
 	client = _create_client(
 		creds=creds, application_id=application_id, topic=topic, err_func=err_func
 	)
 
+	aps_kwargs = {}
+	if mutable_content:
+		aps_kwargs["mutable-content"] = mutable_content
+
 	requests = [_create_notification_request_from_args(
 		registration_id,
 		alert,
@@ -372,6 +381,7 @@ async def _send_bulk_request(
 		loc_key=loc_key,
 		priority=priority,
 		collapse_id=collapse_id,
+		aps_kwargs=aps_kwargs
 	) for registration_id in registration_ids]
 
 	send_requests = [_send_request(client, request) for request in requests]

tests/test_apns_async_push_payload.py

@@ -181,6 +181,25 @@ def test_aioapns_err_func(self, mock_cert_pool):
 			mock.ANY, result
 		)
 
+	@mock.patch("push_notifications.apns_async.APNs", autospec=True)
+	def test_push_payload_with_mutable_content(self, mock_apns):
+		apns_send_message(
+			"123",
+			"Hello world",
+			mutable_content=True,
+			creds=TokenCredentials(key="aaa", key_id="bbb", team_id="ccc"),
+			sound="chime",
+			extra={"custom_data": 12345},
+			expiration=int(time.time()) + 3,
+		)
+
+		args, kwargs = mock_apns.return_value.send_notification.call_args
+		req = args[0]
+
+		# Assertions
+		self.assertTrue("mutable-content" in req.message["aps"])
+		self.assertEqual(req.message["aps"]["mutable-content"], 1)  # APNs expects 1 for True
+
 	# def test_bad_priority(self):
 	# 	with mock.patch("apns2.credentials.init_context"):
 	# 		with mock.patch("apns2.client.APNsClient.connect"):