Add Video Grant

jingming · jingming · commit 380e6242bab9 · 2016-09-23T14:10:23.000+01:00
diff --git a/tests/unit/jwt/test_access_token.py b/tests/unit/jwt/test_access_token.py
@@ -4,7 +4,7 @@
 from datetime import datetime
 from nose.tools import assert_equal
 from twilio.jwt import decode
-from twilio.jwt.access_token import AccessToken, ConversationsGrant, IpMessagingGrant, SyncGrant, VoiceGrant
+from twilio.jwt.access_token import AccessToken, ConversationsGrant, IpMessagingGrant, SyncGrant, VoiceGrant, VideoGrant
 
 ACCOUNT_SID = 'AC123'
 SIGNING_KEY_SID = 'SK123'
@@ -79,6 +79,19 @@ def test_conversations_grant(self):
             'configuration_profile_sid': 'CP123'
         }, payload['grants']['rtc'])
 
+    def test_video_grant(self):
+        scat = AccessToken(ACCOUNT_SID, SIGNING_KEY_SID, 'secret')
+        scat.add_grant(VideoGrant(configuration_profile_sid='CP123'))
+
+        token = str(scat)
+        assert_is_not_none(token)
+        payload = decode(token, 'secret')
+        self._validate_claims(payload)
+        assert_equal(1, len(payload['grants']))
+        assert_equal({
+            'configuration_profile_sid': 'CP123'
+        }, payload['grants']['video'])
+
     def test_ip_messaging_grant(self):
         scat = AccessToken(ACCOUNT_SID, SIGNING_KEY_SID, 'secret')
         scat.add_grant(IpMessagingGrant(service_sid='IS123', push_credential_sid='CR123'))
diff --git a/twilio/jwt/access_token.py b/twilio/jwt/access_token.py
@@ -105,6 +105,23 @@ def to_payload(self):
         return grant
 
 
+class VideoGrant(object):
+    """ Grant to access Twilio Video """
+    def __init__(self, configuration_profile_sid=None):
+        self.configuration_profile_sid = configuration_profile_sid
+
+    @property
+    def key(self):
+        return "video"
+
+    def to_payload(self):
+        grant = {}
+        if self.configuration_profile_sid:
+            grant['configuration_profile_sid'] = self.configuration_profile_sid
+
+        return grant
+
+
 class AccessToken(object):
     """ Access Token used to access Twilio Resources """
     def __init__(self, account_sid, signing_key_sid, secret,
