Merge pull request #317 from theonion/guide-to-fields

Guide To serializer updates
theonion · Sep 6, 2016 · bd1d788 · bd1d788
2 parents ca41de0 + 7e15753
commit bd1d788
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 2 deletions.
diff --git a/bulbs/super_features/data_serializers.py b/bulbs/super_features/data_serializers.py
@@ -2,14 +2,21 @@
 
 from djbetty.serializers import ImageFieldSerializer
 
-from bulbs.utils.data_serializers import EntrySerializer
+from bulbs.utils.fields import RichTextField
+from bulbs.utils.data_serializers import BaseEntrySerializer
+
+
+class GuideToEntrySerializer(BaseEntrySerializer):
+    copy = RichTextField(required=True, field_size="long")
+    image = ImageFieldSerializer(required=False, default=None, allow_null=True)
 
 
 class GuideToChildSerializer(serializers.Serializer):
-    entries = EntrySerializer(many=True, required=False, child_label="entry")
+    entries = GuideToEntrySerializer(many=True, required=False, child_label="entry")
 
 
 class GuideToParentSerializer(serializers.Serializer):
+    copy = RichTextField(required=False, field_size="long")
     sponsor_brand_messaging = serializers.CharField(required=False)
     sponsor_product_shot = ImageFieldSerializer(
         required=False,

diff --git a/tests/super_features/test_api.py b/tests/super_features/test_api.py
@@ -11,6 +11,8 @@
 
 class BaseSuperFeatureTestCase(BaseAPITestCase):
 
+    maxDiff = None
+
     def setUp(self):
         super(BaseSuperFeatureTestCase, self).setUp()
         self.doc_type = BaseSuperFeature.search_objects.mapping.doc_type
@@ -121,6 +123,12 @@ def test_options_guide_to(self):
         data_field = fields.get("data")
         self.assertEqual(data_field, {
             'fields': {
+                'copy': OrderedDict([
+                    ("field_size", "long"),
+                    ("read_only", False),
+                    ("required", False),
+                    ("type", "richtext")
+                ]),
                 'sponsor_brand_messaging': OrderedDict([
                     ('type', 'string'),
                     ('required', False),