Add support for news articles to GraphQL endpoint

app/graphql/types/edition_type.rb

@@ -23,6 +23,7 @@ class WithdrawnNotice < Types::BaseObject
     field :rendering_app, String
     field :scheduled_publishing_delay_seconds, Int
     field :schema_name, String
+    field :state, String
     field :title, String, null: false
     field :updated_at, GraphQL::Types::ISO8601DateTime
     field :withdrawn_notice, WithdrawnNotice
@@ -42,6 +43,79 @@ def not_stored_in_publishing_api
     alias_method :publishing_scheduled_at, :not_stored_in_publishing_api
     alias_method :scheduled_publishing_delay_seconds, :not_stored_in_publishing_api
 
+    class Translation < Types::BaseObject
+      field :locale, String
+      field :base_path, String
+    end
+
+    class GovernmentDetails < Types::BaseObject
+      field :current, Boolean
+    end
+
+    class GovernmentLink < Types::BaseObject
+      field :base_path, String
+      field :content_id, String
+      field :details, GovernmentDetails
+      field :title, String
+    end
+
+    class OrganisationLink < Types::BaseObject
+      field :base_path, String
+      field :content_id, String
+      field :title, String
+    end
+
+    class PersonLink < Types::BaseObject
+      field :base_path, String
+      field :content_id, String
+      field :title, String
+    end
+
+    class Taxon < Types::BaseObject
+      field :base_path, String
+      field :content_id, String
+      field :document_type, String
+      field :phase, String
+      field :title, String
+    end
+
+    class TaxonLink < Taxon
+      class TaxonLinks < Types::BaseObject
+        links_field :parent_taxons, [Taxon]
+      end
+
+      field :links, TaxonLinks, method: :itself
+    end
+
+    class TopicalEventLink < Types::BaseObject
+      field :base_path, String
+      field :content_id, String
+      field :title, String
+    end
+
+    class WorldLocationLink < Types::BaseObject
+      field :base_path, String
+      field :content_id, String
+      field :title, String
+    end
+
+    class EditionLinks < Types::BaseObject
+      field :available_translations, [Translation]
+      links_field :government, [GovernmentLink]
+      links_field :organisations, [OrganisationLink]
+      links_field :people, [PersonLink]
+      links_field :taxons, [TaxonLink]
+      links_field :topical_events, [TopicalEventLink]
+      links_field :world_locations, [WorldLocationLink]
+
+      def available_translations
+        Presenters::Queries::AvailableTranslations.by_edition(object)
+          .translations.fetch(:available_translations, [])
+      end
+    end
+
+    field :links, EditionLinks, method: :itself
+
   private
 
     def presented_edition

app/graphql/types/ministers_index_type.rb

@@ -2,7 +2,7 @@
 
 module Types
   class MinistersIndexType < Types::EditionType
-    def self.document_type = "ministers_index"
+    def self.document_types = %w[ministers_index]
 
     class MinistersIndexPerson < Types::BaseObject
       class MinistersIndexRoleAppointment < Types::BaseObject

app/graphql/types/news_article_type.rb

@@ -0,0 +1,12 @@
+# frozen_string_literal: true
+
+module Types
+  class NewsArticleType < Types::EditionType
+    def self.document_types = %w[
+      government_response
+      news_story
+      press_release
+      world_news_story
+    ]
+  end
+end

app/graphql/types/query_type.rb

@@ -4,10 +4,11 @@ module Types
   class QueryType < Types::BaseObject
     field :edition, EditionTypeOrSubtype, description: "An edition or one of its subtypes" do
       argument :base_path, String
+      argument :content_store, String, required: false, default_value: "live"
     end
 
-    def edition(base_path:)
-      Edition.live.find_by(base_path:)
+    def edition(base_path:, content_store:)
+      Edition.where(content_store:).find_by(base_path:)
     end
   end
 end

app/graphql/types/query_type/edition_type_or_subtype.rb

@@ -3,7 +3,13 @@
 module Types
   class QueryType
     class EditionTypeOrSubtype < Types::BaseUnion
-      EDITION_TYPES = [Types::EditionType, Types::RoleType, Types::WorldIndexType, Types::MinistersIndexType].freeze
+      EDITION_TYPES = [
+        Types::EditionType,
+        Types::MinistersIndexType,
+        Types::NewsArticleType,
+        Types::RoleType,
+        Types::WorldIndexType,
+      ].freeze
 
       possible_types(*EDITION_TYPES)
 
@@ -12,7 +18,7 @@ def resolve_type(object, _context)
           document_type = object.document_type
 
           matching_edition_subtype = Types::EditionType.descendants.find do |edition_subtype|
-            document_type == edition_subtype.document_type
+            edition_subtype.document_types.include?(document_type)
           end
 
           matching_edition_subtype || Types::EditionType

app/graphql/types/role_type.rb

@@ -2,7 +2,7 @@
 
 module Types
   class RoleType < Types::EditionType
-    def self.document_type = "ministerial_role"
+    def self.document_types = %w[ministerial_role]
 
     class Organisation < Types::BaseObject
       field :base_path, String
@@ -56,11 +56,6 @@ def person
       field :links, RoleAppointmentLinks, method: :itself
     end
 
-    class Translation < Types::BaseObject
-      field :locale, String
-      field :base_path, String
-    end
-
     class RoleDetails < Types::BaseObject
       field :body, String
       field :supports_historical_accounts, Boolean
@@ -75,16 +70,10 @@ def body
       end
     end
 
-    class RoleLinks < Types::BaseObject
-      field :available_translations, [Translation]
+    class RoleLinks < EditionType::EditionLinks
       field :ordered_parent_organisations, [Organisation]
       field :role_appointments, [RoleAppointment]
 
-      def available_translations
-        Presenters::Queries::AvailableTranslations.by_edition(object)
-          .translations.fetch(:available_translations, [])
-      end
-
       def ordered_parent_organisations
         Edition
           .live

app/graphql/types/world_index_type.rb

@@ -2,7 +2,7 @@
 
 module Types
   class WorldIndexType < Types::EditionType
-    def self.document_type = "world_index"
+    def self.document_types = %w[world_index]
 
     class WorldLocation < Types::BaseObject
       field :active, Boolean, null: false

spec/graphql/types/news_article_type_spec.rb

@@ -0,0 +1,12 @@
+RSpec.describe Types::NewsArticleType do
+  describe ".document_types" do
+    it "defines the document types for .resolve_type to distinguish the type from the generic Edition type" do
+      expect(described_class.document_types).to eq(%w[
+        government_response
+        news_story
+        press_release
+        world_news_story
+      ])
+    end
+  end
+end

spec/graphql/types/query_type_spec.rb

@@ -0,0 +1,101 @@
+RSpec.describe Types::QueryType do
+  include GraphQL::Testing::Helpers.for(PublishingApiSchema)
+
+  describe "#edition" do
+    let(:query) do
+      <<~QUERY
+        query($basePath: String!, $contentStore: String!) {
+          edition(basePath: $basePath, contentStore: $contentStore) {
+            ... on Edition {
+              basePath
+              state
+            }
+          }
+        }
+      QUERY
+    end
+
+    context "when there is only a draft edition" do
+      let(:draft_edition) { create(:draft_edition) }
+
+      context "requesting the draft edition" do
+        it "returns the draft edition" do
+          expected_data = {
+            "basePath" => draft_edition.base_path,
+            "state" => "draft",
+          }
+
+          result = PublishingApiSchema.execute(query, variables: { basePath: draft_edition.base_path, contentStore: "draft" })
+          edition_data = result.dig("data", "edition")
+          expect(edition_data).to eq(expected_data)
+        end
+      end
+
+      context "requesting the live edition" do
+        it "returns no edition" do
+          result = PublishingApiSchema.execute(query, variables: { basePath: draft_edition.base_path, contentStore: "live" })
+          edition_data = result.dig("data", "edition")
+          expect(edition_data).to be_nil
+        end
+      end
+    end
+
+    context "when there is only a live edition" do
+      let(:live_edition) { create(:live_edition) }
+
+      context "requesting the draft edition" do
+        it "returns no edition" do
+          result = PublishingApiSchema.execute(query, variables: { basePath: live_edition.base_path, contentStore: "draft" })
+          edition_data = result.dig("data", "edition")
+          expect(edition_data).to be_nil
+        end
+      end
+
+      context "requesting the live edition" do
+        it "returns the live edition" do
+          expected_data = {
+            "basePath" => live_edition.base_path,
+            "state" => "published",
+          }
+
+          result = PublishingApiSchema.execute(query, variables: { basePath: live_edition.base_path, contentStore: "live" })
+          edition_data = result.dig("data", "edition")
+          expect(edition_data).to eq(expected_data)
+        end
+      end
+    end
+
+    context "when there is a published edition and a newer draft edition" do
+      let(:document) { create(:document) }
+      let(:base_path) { "/foo" }
+      let!(:live_edition) { create(:live_edition, document:, base_path:, user_facing_version: 1) }
+      let!(:draft_edition) { create(:draft_edition, document:, base_path:, user_facing_version: 2) }
+
+      context "requesting the draft edition" do
+        it "returns the draft edition" do
+          expected_data = {
+            "basePath" => draft_edition.base_path,
+            "state" => "draft",
+          }
+
+          result = PublishingApiSchema.execute(query, variables: { basePath: draft_edition.base_path, contentStore: "draft" })
+          edition_data = result.dig("data", "edition")
+          expect(edition_data).to eq(expected_data)
+        end
+      end
+
+      context "requesting the live edition" do
+        it "returns the live edition" do
+          expected_data = {
+            "basePath" => live_edition.base_path,
+            "state" => "published",
+          }
+
+          result = PublishingApiSchema.execute(query, variables: { basePath: live_edition.base_path, contentStore: "live" })
+          edition_data = result.dig("data", "edition")
+          expect(edition_data).to eq(expected_data)
+        end
+      end
+    end
+  end
+end

spec/graphql/types/world_index_type_spec.rb

@@ -1,7 +1,7 @@
 RSpec.describe "Types::WorldIndexType" do
-  describe ".document_type" do
-    it "defines a document type for .resolve_type to distinguish the type from the generic Edition type" do
-      expect(Types::WorldIndexType.document_type).to eq("world_index")
+  describe ".document_types" do
+    it "defines the document types for .resolve_type to distinguish the type from the generic Edition type" do
+      expect(Types::WorldIndexType.document_types).to eq(%w[world_index])
     end
   end
 end