From 82dc18b24ff383d214eaa0ccbd5c51d81a1611b2 Mon Sep 17 00:00:00 2001
From: Matt Westcott <matt@west.co.tt>
Date: Wed, 13 Jan 2021 23:43:55 +0000
Subject: [PATCH] add unit tests for telepath

---
 wagtail/core/tests/test_telepath.py | 105 ++++++++++++++++++++++++++++
 1 file changed, 105 insertions(+)
 create mode 100644 wagtail/core/tests/test_telepath.py

diff --git a/wagtail/core/tests/test_telepath.py b/wagtail/core/tests/test_telepath.py
new file mode 100644
index 0000000000..1beffa0f8d
--- /dev/null
+++ b/wagtail/core/tests/test_telepath.py
@@ -0,0 +1,105 @@
+from django.test import TestCase
+
+from wagtail.core.telepath import Adapter, JSContext, register
+
+
+class Artist:
+    def __init__(self, name):
+        self.name = name
+
+
+class Album:
+    def __init__(self, title, artists):
+        self.title = title
+        self.artists = artists
+
+
+class ArtistAdapter(Adapter):
+    js_constructor = 'music.Artist'
+
+    def js_args(self, obj):
+        return [obj.name]
+
+
+register(ArtistAdapter(), Artist)
+
+
+class AlbumAdapter(Adapter):
+    js_constructor = 'music.Album'
+
+    def js_args(self, obj):
+        return [obj.title, obj.artists]
+
+    class Media:
+        js = ['music_player.js']
+
+
+register(AlbumAdapter(), Album)
+
+
+class TestPacking(TestCase):
+    def test_pack_object(self):
+        beyonce = Artist("Beyoncé")
+        ctx = JSContext()
+        result = ctx.pack(beyonce)
+
+        self.assertEqual(result, {'_type': 'music.Artist', '_args': ["Beyoncé"]})
+
+    def test_pack_list(self):
+        destinys_child = [
+            Artist("Beyoncé"), Artist("Kelly Rowland"), Artist("Michelle Williams")
+        ]
+        ctx = JSContext()
+        result = ctx.pack(destinys_child)
+
+        self.assertEqual(result, [
+            {'_type': 'music.Artist', '_args': ["Beyoncé"]},
+            {'_type': 'music.Artist', '_args': ["Kelly Rowland"]},
+            {'_type': 'music.Artist', '_args': ["Michelle Williams"]},
+        ])
+
+    def test_pack_dict(self):
+        glastonbury = {
+            'pyramid_stage': Artist("Beyoncé"),
+            'acoustic_stage': Artist("Ed Sheeran"),
+        }
+        ctx = JSContext()
+        result = ctx.pack(glastonbury)
+        self.assertEqual(result, {
+            'pyramid_stage': {'_type': 'music.Artist', '_args': ["Beyoncé"]},
+            'acoustic_stage': {'_type': 'music.Artist', '_args': ["Ed Sheeran"]},
+        })
+
+    def test_dict_reserved_words(self):
+        profile = {
+            '_artist': Artist("Beyoncé"),
+            '_type': 'R&B',
+        }
+        ctx = JSContext()
+        result = ctx.pack(profile)
+
+        self.assertEqual(result, {
+            '_dict': {
+                '_artist': {'_type': 'music.Artist', '_args': ["Beyoncé"]},
+                '_type': 'R&B',
+            }
+        })
+
+    def test_recursive_arg_packing(self):
+        dangerously_in_love = Album("Dangerously in Love", [
+            Artist("Beyoncé"),
+        ])
+        ctx = JSContext()
+        result = ctx.pack(dangerously_in_love)
+
+        self.assertEqual(result, {
+            '_type': 'music.Album',
+            '_args': [
+                "Dangerously in Love",
+                [
+                    {'_type': 'music.Artist', '_args': ["Beyoncé"]},
+                ]
+            ]
+        })
+
+        self.assertIn('music_player.js', str(ctx.media))