Pormenjen naicn rendovanja podatak iz template, popunjeni ceo dokument

backend/core/tests/test_views.py

@@ -53,13 +53,16 @@ class DocumentViewTest(TestCase):
 
         template_content = """
         - segment_type: "h1"
-          content: "{{ document.organization.name }} - Risk Report"
+          content: 
+            title: "{{ document.organization.name }} - Risk Report"
         - segment_type: "p"
-          content: "Created at: {{ document.created_at|date:'Y-m-d' }}"
+          content:
+            descripton: |
+              "Created at: {{ document.created_at|date:'Y-m-d' }}"
         - segment_type: "h2"
           content: "Top 10 Risk Identified"
         - segment_type: "table"
-          content: |
+          html: |
             <table>
               <tr>
                 <th>Risk ID</th>
@@ -81,11 +84,13 @@ class DocumentViewTest(TestCase):
               {% endfor %}
             </table>
         - segment_type: "image"
-          content: "data:image/png;base64,{{ graph }}"
+          content:
+            image: "data:image/png;base64,{{ graph }}"
         - segment_type: "h2"
-          content: "Risks with Controls"
+          content:
+            title: "Risks with Controls"
         - segment_type: "body"
-          content: |
+          html: |
             {% for item in risks_with_controls %}
             <div class="risk">
               <h3>Risk: {{ item.risk.name }}</h3>
@@ -107,7 +112,8 @@ class DocumentViewTest(TestCase):
         response = self.client.get(reverse('core:document', kwargs={'document_id': self.document.id}))
         self.assertEqual(response.status_code, 200)
         self.assertTemplateUsed(response, 'document.html')
-        self.assertContains(response, self.organization.name) 
+        self.assertContains(response, self.organization.name)  
+        self.assertContains(response, "Risk Report")  
 
     def test_index_view(self):
         response = self.client.get(reverse('core:index'))