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(
- '',
render('{youtube ABCDEF}')
)
@@ -68,9 +70,11 @@ class YouTubeTest(AbstractPillarTest):
from pillar.shortcodes import render
self.assertEqual(
- '',
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"}')
)