Update embedded content helper to add rollup data

alphagov · Nov 28, 2024 · e6c398e · e6c398e
1 parent 21a90bf
commit e6c398e
Showing 1 changed file with 28 additions and 4 deletions.
diff --git a/lib/gds_api/test_helpers/publishing_api.rb b/lib/gds_api/test_helpers/publishing_api.rb
@@ -371,12 +371,27 @@ def stub_publishing_api_has_content(items, params = {})
       #         "base_path" => "/organisation/bar",
       #       },
       #     }] # an array of content items that embed the target content_id
+      #     rollup: {
+      #       "views" => 1
+      #       "locations" => 1
+      #       "instances" => 1
+      #       "organisations" => 1
+      #     } # metadata with the total counts across all pages
       #   )
-      # @param content_id [UUID, Mocha::ParameterMatchers::Anything]
+      # @param content_id [UUID]
       # @param total Integer
       # @param total_pages Integer
-      # @param results [Hash]
-      def stub_publishing_api_has_embedded_content(content_id:, total: 0, total_pages: 0, results: [], page_number: nil, order: nil)
+      # @param results [Array]
+      # @param rollup [Hash]
+      def stub_publishing_api_has_embedded_content(
+        content_id:,
+        total: 0,
+        total_pages: 0,
+        results: [],
+        page_number: nil,
+        order: nil,
+        rollup: {}
+      )
         url = "#{PUBLISHING_API_V2_ENDPOINT}/content/#{content_id}/embedded"
 
         query = {
@@ -390,11 +405,19 @@ def stub_publishing_api_has_embedded_content(content_id:, total: 0, total_pages:
             "content_id" => content_id,
             "total" => total,
             "total_pages" => total_pages,
+            "rollup" => rollup,
             "results" => results,
           }.to_json)
       end
 
-      def stub_publishing_api_has_embedded_content_for_any_content_id(total: 0, total_pages: 0, results: [], page_number: nil, order: nil)
+      def stub_publishing_api_has_embedded_content_for_any_content_id(
+        total: 0,
+        total_pages: 0,
+        results: [],
+        page_number: nil,
+        order: nil,
+        rollup: {}
+      )
         url = %r{\A#{PUBLISHING_API_V2_ENDPOINT}/content/[0-9a-fA-F-]{36}/embedded}
 
         query = {
@@ -408,6 +431,7 @@ def stub_publishing_api_has_embedded_content_for_any_content_id(total: 0, total_
             "content_id" => SecureRandom.uuid,
             "total" => total,
             "total_pages" => total_pages,
+            "rollup" => rollup,
             "results" => results,
           }.to_json)
       end