diff --git a/tests/test_shortcodes.py b/tests/test_shortcodes.py index 2e90cf28..d402ae7a 100644 --- a/tests/test_shortcodes.py +++ b/tests/test_shortcodes.py @@ -58,9 +58,11 @@ class YouTubeTest(AbstractPillarTest): from pillar.shortcodes import render self.assertEqual( - '', + 'allow="autoplay; encrypted-media" allowfullscreen>' + '', render('{youtube ABCDEF}') ) @@ -68,9 +70,11 @@ class YouTubeTest(AbstractPillarTest): from pillar.shortcodes import render self.assertEqual( - '', + 'allow="autoplay; encrypted-media" allowfullscreen>' + '', render('{youtube http://youtube.com/embed/ABCDEF}') ) @@ -78,9 +82,11 @@ class YouTubeTest(AbstractPillarTest): from pillar.shortcodes import render self.assertEqual( - '', + 'allow="autoplay; encrypted-media" allowfullscreen>' + '', render('{youtube https://youtu.be/NwVGvcIrNWA}') ) @@ -88,9 +94,11 @@ class YouTubeTest(AbstractPillarTest): from pillar.shortcodes import render self.assertEqual( - '', + 'allow="autoplay; encrypted-media" allowfullscreen>' + '', render('{youtube "https://www.youtube.com/watch?v=NwVGvcIrNWA"}') ) @@ -98,9 +106,11 @@ class YouTubeTest(AbstractPillarTest): from pillar.shortcodes import render self.assertEqual( - '', + 'allow="autoplay; encrypted-media" allowfullscreen>' + '', render('{youtube "https://www.youtube.com/watch?v=NwVGvcIrNWA" width=5 height="3"}') )