Add tests for exercise file generation during publishing.

Only automatically write attached files to perseus file for raw perseus questions.

Author: rtibbles

contentcuration/contentcuration/tests/testdata.py

@@ -11,7 +11,9 @@
 
 import pytest
 from django.core.files.storage import default_storage
+from le_utils.constants import exercises
 from le_utils.constants import format_presets
+from PIL import Image
 
 from contentcuration import models as cc
 from contentcuration.tests.utils import mixer
@@ -385,27 +387,31 @@ def create_test_file(filebytes, ext="pdf"):
 ]
 
 
-def fileobj_exercise_image():
+def fileobj_exercise_image(size=(100, 100), color="red"):
     """
     Create a generic exercise image file in storage and return a File model pointing to it.
     """
-    filecontents = "".join(random.sample(string.printable, 20))
+    image = Image.new("RGB", size, color=color)
+    buffer = BytesIO()
+    image.save(buffer, "JPEG")
     temp_file_dict = create_studio_file(
-        filecontents, preset=format_presets.EXERCISE_IMAGE, ext="jpg"
+        buffer.getvalue(), preset=format_presets.EXERCISE_IMAGE, ext="jpg"
     )
     return temp_file_dict["db_file"]
 
 
-def fileobj_exercise_graphie():
+def fileobj_exercise_graphie(original_filename=None):
     """
     Create an graphi exercise image file in storage and return a File model pointing to it.
     """
-    filecontents = "".join(random.sample(string.printable, 20))
+    svg_content = f"<svg><circle cx='50' cy='50' r='40' />{original_filename or ''.join(random.sample(string.printable, 20))}</svg>"
+    json_content = '{"version": {"major": 0, "minor": 0}}'
+    filecontents = svg_content + exercises.GRAPHIE_DELIMITER + json_content
     temp_file_dict = create_studio_file(
         filecontents,
         preset=format_presets.EXERCISE_GRAPHIE,
         ext="graphie",
-        original_filename="theoriginalfilename",
+        original_filename=original_filename or "theoriginalfilename",
     )
     return temp_file_dict["db_file"]
 

contentcuration/contentcuration/tests/utils/perseus_question_new_bar_graphs.json

@@ -0,0 +1,114 @@
+{
+  "answerArea": {
+    "calculator": false,
+    "chi2Table": false,
+    "periodicTable": false,
+    "tTable": false,
+    "zTable": false
+  },
+  "hints": [
+    {
+      "content": "The bottom bar lines up to $\\purpleD{6}$.   \n\n![](web+graphie://cdn.kastatic.org/ka-perseus-graphie/d855aefe9a722f9a794b0883ebcdb8c37b4ba0c7)\n\nWhich type of fruit has $\\purpleD{6}$ in Luigi's home?",
+      "images": {
+        "web+graphie://cdn.kastatic.org/ka-perseus-graphie/d855aefe9a722f9a794b0883ebcdb8c37b4ba0c7": {
+          "height": 330,
+          "width": 404
+        }
+      },
+      "replace": false,
+      "widgets": {}
+    },
+    {
+      "content": "Kind of fruit | Number\n:- | :-: \nOranges | $\\purpleD{6}$ \n\nLuigi has $\\purpleD{6}$ oranges.  So, the bottom bar should be labeled $\\purpleD{\\text{Oranges}}$.",
+      "images": {},
+      "replace": false,
+      "widgets": {}
+    },
+    {
+      "content": "Now let's label the other bars to match the table.",
+      "images": {},
+      "replace": false,
+      "widgets": {}
+    },
+    {
+      "content": "Here is the completed graph:\n\n![](web+graphie://cdn.kastatic.org/ka-perseus-graphie/95262ebaf42bdd1929e5d6d1e2853d3eb0a5cc74)",
+      "images": {
+        "web+graphie://cdn.kastatic.org/ka-perseus-graphie/95262ebaf42bdd1929e5d6d1e2853d3eb0a5cc74": {
+          "height": 330,
+          "width": 404
+        }
+      },
+      "replace": false,
+      "widgets": {}
+    }
+  ],
+  "itemDataVersion": {
+    "major": 0,
+    "minor": 1
+  },
+  "question": {
+    "content": "Luigi created a chart and a bar graph to show how many of each type of fruit were in his home.\n\nKind of fruit | Number \n:- | :-: \nApple | $7$ \nStrawberries | $3$ \nOranges | $6$ \nBananas| $2$ \n\n**Label each bar on the bar graph.**\n\n[[☃ label-image 1]]\n",
+    "images": {},
+    "widgets": {
+      "label-image 1": {
+        "alignment": "default",
+        "graded": true,
+        "options": {
+          "choices": [
+            "Apple",
+            "Strawberries",
+            "Oranges",
+            "Bananas"
+          ],
+          "hideChoicesFromInstructions": true,
+          "imageAlt": "",
+          "imageHeight": 330,
+          "imageUrl": "web+graphie://cdn.kastatic.org/ka-perseus-graphie/ab207c6f38c887130b68c078e6158a87aab60c45",
+          "imageWidth": 404,
+          "markers": [
+            {
+              "answers": [
+                "Strawberries"
+              ],
+              "label": "",
+              "x": 24.1,
+              "y": 17.7
+            },
+            {
+              "answers": [
+                "Bananas"
+              ],
+              "label": "",
+              "x": 24.4,
+              "y": 35.7
+            },
+            {
+              "answers": [
+                "Apple"
+              ],
+              "label": "",
+              "x": 23.8,
+              "y": 52.9
+            },
+            {
+              "answers": [
+                "Oranges"
+              ],
+              "label": "",
+              "x": 24.1,
+              "y": 70.9
+            }
+          ],
+          "multipleAnswers": false,
+          "static": false
+        },
+        "static": false,
+        "type": "label-image",
+        "version": {
+          "major": 0,
+          "minor": 0
+        }
+      }
+    }
+  }
+}