Updating tests

Author: ZachOrr

firebase_admin/messaging.py

@@ -379,6 +379,8 @@ def send_all(self, messages, dry_run=False):
         """Sends the given messages to FCM via the batch API."""
         if not isinstance(messages, list):
             raise ValueError('Messages must be an list of messaging.Message instances.')
+        if len(messages) > 100:
+            raise ValueError('send_all messages must not contain more than 100 messages.')
 
         responses = []
 

tests/test_messaging.py

@@ -53,8 +53,15 @@ def test_invalid_tokens_type(self, tokens):
             expected = 'MulticastMessage.tokens must be a list of strings.'
             assert str(excinfo.value) == expected
 
+    def test_tokens_over_one_hundred(self):
+        with pytest.raises(ValueError) as excinfo:
+            messaging.MulticastMessage(tokens=['token' for i in xrange(0, 101)])
+        expected = 'MulticastMessage.tokens must not contain more than 100 tokens.'
+        assert str(excinfo.value) == expected
+
     def test_tokens_type(self):
         messaging.MulticastMessage(tokens=['token'])
+        messaging.MulticastMessage(tokens=['token' for i in xrange(0, 100)])
 
 
 class TestMessageEncoder(object):
@@ -1385,6 +1392,13 @@ def test_invalid_send_all(self, msg):
             expected = 'Messages must be an list of messaging.Message instances.'
             assert str(excinfo.value) == expected
 
+    def test_invalid_over_one_hundred(self):
+        msg = messaging.Message(topic='foo')
+        with pytest.raises(ValueError) as excinfo:
+            messaging.send_all([msg for i in xrange(0, 101)])
+        expected = 'send_all messages must not contain more than 100 messages.'
+        assert str(excinfo.value) == expected
+
     def test_send_all(self):
         payload = json.dumps({'name': 'message-id'})
         _ = self._instrument_batch_messaging_service(