chore(fcm): Add integration tests for AndroidConfigV2

Author: jonathanedey

integration/test_messaging.py

@@ -71,6 +71,139 @@ def test_send():
     msg_id = messaging.send(msg, dry_run=True)
     assert re.match('^projects/.*/messages/.*$', msg_id)
 
+
+def _full_android_notification_v2():
+    return messaging.AndroidNotificationV2(
+        title='test.title',
+        body='test.body',
+        icon='test.icon',
+        color='#AABBCC',
+        sound='test.sound',
+        tag='test.tag',
+        id=42,
+        click_action='test.click.action',
+        body_loc_key='test.body.loc.key',
+        body_loc_args=['body.arg1', 'body.arg2'],
+        title_loc_key='test.title.loc.key',
+        title_loc_args=['title.arg1', 'title.arg2'],
+        channel_id='test.channel.id',
+        image='https://example.com/image.png',
+        ticker='test.ticker',
+        sticky=True,
+        event_time=datetime.now(),
+        local_only=True,
+        notification_priority='high',
+        vibrate_timings_millis=[100, 50, 250],
+        default_vibrate_timings=False,
+        default_sound=True,
+        light_settings=messaging.LightSettings(
+            color='#AABBCC',
+            light_on_duration_millis=200,
+            light_off_duration_millis=300,
+        ),
+        default_light_settings=False,
+        visibility='private',
+        notification_count=1,
+    )
+
+
+def _full_android_config_v2_base(**kwargs):
+    return messaging.AndroidConfigV2(
+        collapse_key='test-key',
+        ttl=5,
+        restricted_package_name='com.google.firebase.testing',
+        data={'androidFoo': 'androidBar'},
+        fcm_options=messaging.AndroidFCMOptions(analytics_label='test-analytics'),
+        direct_boot_ok=True,
+        bandwidth_constrained_ok=True,
+        restricted_satellite_ok=True,
+        **kwargs,
+    )
+
+
+def _android_v2_remote_notification_message():
+    return messaging.Message(
+        topic='foo-bar',
+        notification=messaging.Notification('Title', 'Body'),
+        android_v2=_full_android_config_v2_base(
+            remote_notification=messaging.AndroidRemoteNotification(
+                mutable_content=True,
+                use_as_v1_data_message=True,
+                notification=_full_android_notification_v2(),
+            ),
+        ),
+    )
+
+
+def _android_v2_minimal_remote_notification_message():
+    return messaging.Message(
+        topic='foo-bar',
+        android_v2=messaging.AndroidConfigV2(
+            remote_notification=messaging.AndroidRemoteNotification(
+                notification=messaging.AndroidNotificationV2(
+                    title='test.title',
+                    body='test.body',
+                ),
+            ),
+        ),
+    )
+
+
+def _android_v2_background_sync_message():
+    return messaging.Message(
+        topic='foo-bar',
+        android_v2=_full_android_config_v2_base(
+            background_sync=messaging.AndroidBackgroundSyncMessage(),
+        ),
+    )
+
+
+def _android_v2_minimal_background_sync_message():
+    return messaging.Message(
+        topic='foo-bar',
+        android_v2=messaging.AndroidConfigV2(
+            background_sync=messaging.AndroidBackgroundSyncMessage(),
+        ),
+    )
+
+
+def test_send_android_v2_remote_notification():
+    msg_id = messaging.send(_android_v2_remote_notification_message(), dry_run=True)
+    assert re.match('^projects/.*/messages/.*$', msg_id)
+
+
+def test_send_android_v2_minimal_remote_notification():
+    msg_id = messaging.send(_android_v2_minimal_remote_notification_message(), dry_run=True)
+    assert re.match('^projects/.*/messages/.*$', msg_id)
+
+
+def test_send_android_v2_background_sync():
+    msg_id = messaging.send(_android_v2_background_sync_message(), dry_run=True)
+    assert re.match('^projects/.*/messages/.*$', msg_id)
+
+
+def test_send_android_v2_minimal_background_sync():
+    msg_id = messaging.send(_android_v2_minimal_background_sync_message(), dry_run=True)
+    assert re.match('^projects/.*/messages/.*$', msg_id)
+
+
+def test_send_each_android_v2():
+    messages = [
+        _android_v2_remote_notification_message(),
+        _android_v2_minimal_remote_notification_message(),
+        _android_v2_background_sync_message(),
+        _android_v2_minimal_background_sync_message(),
+    ]
+    batch_response = messaging.send_each(messages, dry_run=True)
+    assert batch_response.success_count == 4
+    assert batch_response.failure_count == 0
+    assert len(batch_response.responses) == 4
+    for response in batch_response.responses:
+        assert response.success is True
+        assert response.exception is None
+        assert re.match('^projects/.*/messages/.*$', response.message_id)
+
+
 def test_send_invalid_token():
     msg = messaging.Message(
         token=_REGISTRATION_TOKEN,
@@ -244,3 +377,21 @@ async def test_send_each_for_multicast_async():
         assert response.success is False
         assert response.exception is not None
         assert response.message_id is None
+
+
+@pytest.mark.asyncio(loop_scope="session")
+async def test_send_each_android_v2_async():
+    messages = [
+        _android_v2_remote_notification_message(),
+        _android_v2_minimal_remote_notification_message(),
+        _android_v2_background_sync_message(),
+        _android_v2_minimal_background_sync_message(),
+    ]
+    batch_response = await messaging.send_each_async(messages, dry_run=True)
+    assert batch_response.success_count == 4
+    assert batch_response.failure_count == 0
+    assert len(batch_response.responses) == 4
+    for response in batch_response.responses:
+        assert response.success is True
+        assert response.exception is None
+        assert re.match('^projects/.*/messages/.*$', response.message_id)