Merge pull request #168 from jkeen/relationship-linkage-for-belongs-to

Use linkage: always: true on belongs_to relationships

Author: richmolj

lib/graphiti/sideload.rb

@@ -18,30 +18,32 @@ class Sideload
       :link_proc
 
     def initialize(name, opts)
-      @name = name
+      @name                          = name
       validate_options!(opts)
-      @parent_resource_class = opts[:parent_resource]
-      @resource_class        = opts[:resource]
-      @primary_key           = opts[:primary_key]
-      @foreign_key           = opts[:foreign_key]
-      @type                  = opts[:type]
-      @base_scope            = opts[:base_scope]
-      @readable              = opts[:readable]
-      @writable              = opts[:writable]
-      @as                    = opts[:as]
-      @link                  = opts[:link]
-      @single                = opts[:single]
-      @remote                = opts[:remote]
+      @parent_resource_class         = opts[:parent_resource]
+      @resource_class                = opts[:resource]
+      @primary_key                   = opts[:primary_key]
+      @foreign_key                   = opts[:foreign_key]
+      @type                          = opts[:type]
+      @base_scope                    = opts[:base_scope]
+      @readable                      = opts[:readable]
+      @writable                      = opts[:writable]
+      @as                            = opts[:as]
+      @link                          = opts[:link]
+      @single                        = opts[:single]
+      @remote                        = opts[:remote]
       apply_belongs_to_many_filter if type == :many_to_many
 
-      @description           = opts[:description]
+      @description                   = opts[:description]
 
       # polymorphic has_many
-      @polymorphic_as        = opts[:polymorphic_as]
+      @polymorphic_as                = opts[:polymorphic_as]
       # polymorphic_belongs_to-specific
-      @group_name            = opts[:group_name]
-      @polymorphic_child     = opts[:polymorphic_child]
-      @parent                = opts[:parent]
+      @group_name                    = opts[:group_name]
+      @polymorphic_child             = opts[:polymorphic_child]
+      @parent                        = opts[:parent]
+      @always_include_resource_ids   = opts[:always_include_resource_ids]
+
       if polymorphic_child?
         parent.resource.polymorphic << resource_class
       end
@@ -112,6 +114,10 @@ def polymorphic_has_many?
       !!@polymorphic_as
     end
 
+    def always_include_resource_ids?
+      !!@always_include_resource_ids
+    end
+
     def link?
       return true if link_proc
 

lib/graphiti/sideload/belongs_to.rb

@@ -1,4 +1,9 @@
 class Graphiti::Sideload::BelongsTo < Graphiti::Sideload
+  def initialize(name, opts)
+    opts = { always_include_resource_ids: true }.merge(opts)
+    super(name, opts)
+  end
+
   def type
     :belongs_to
   end

lib/graphiti/util/serializer_relationships.rb

@@ -53,6 +53,10 @@ def block
         proc do
           data { instance_eval(&data_proc_ref) }
 
+          # include relationship links for belongs_to relationships
+          # https://github.com/graphiti-api/graphiti/issues/167
+          linkage always: sideload_ref.always_include_resource_ids?
+
           if link_ref
             if @proxy.query.links?
               self_ref.send(:validate_link!) unless self_ref.send(:eagerly_validate_links?)

spec/relationship_identifier_spec.rb

@@ -0,0 +1,235 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+RSpec.describe 'relationship identifiers' do
+  include_context 'resource testing'
+
+  # let(:base_scope) { { type: :positions } }
+  let!(:employee) { PORO::Employee.create }
+  let!(:employee2) { PORO::Employee.create }
+  let!(:position1) do
+    PORO::Position.create employee_id: employee.id,
+                          department_id: department1.id
+  end
+  let!(:position2) do
+    PORO::Position.create employee_id: employee.id,
+                          department_id: department2.id
+  end
+  let!(:department1) { PORO::Department.create }
+  let!(:department2) { PORO::Department.create }
+  let!(:bio1) { PORO::Bio.create(employee_id: employee.id) }
+  let!(:bio2) { PORO::Bio.create(employee_id: employee.id) }
+  let!(:team1) do
+    PORO::Team.create team_memberships: [
+      PORO::TeamMembership.new(employee_id: employee.id, team_id: 1)
+    ]
+  end
+  let!(:team2) do
+    PORO::Team.create team_memberships: [
+      PORO::TeamMembership.new(employee_id: employee.id, team_id: 2)
+    ]
+  end
+
+  describe 'has_many' do
+    context 'with default' do
+      let(:resource) do
+        Class.new(PORO::TeamResource) do
+          def self.name
+            'PORO::TeamResource'
+          end
+
+          has_many :employees
+        end
+      end
+
+      before do
+        allow_any_instance_of(PORO::Team).to receive(:employees) { [employee, employee2] }
+        render
+      end
+
+      it 'does not include anything' do
+        expect do
+          included('employees')
+        end.to raise_error(GraphitiSpecHelpers::Errors::NoSideloads)
+      end
+
+      it 'specifies meta[:included] = false' do
+        jsonapi_data.each do |record|
+          expect(record.relationships['employees']['meta']['included']).to eq(false)
+        end
+      end
+
+      it 'does not includes relationship identifiers' do
+        jsonapi_data.each do |record|
+          data = record.relationships['employees']['data']
+          expect(data).to be_nil
+        end
+      end
+    end
+    context 'with include directive' do
+      let(:resource) do
+        Class.new(PORO::TeamResource) do
+          def self.name
+            'PORO::TeamResource'
+          end
+
+          has_many :employees do
+            scope do |employee_ids|
+              {
+                type: :employees,
+                conditions: { employee_id: employee_ids }
+              }
+            end
+          end
+        end
+      end
+
+      before do
+        params[:include] = 'employees'
+        allow_any_instance_of(PORO::Team).to receive(:employees) { [employee, employee2] }
+        render
+      end
+
+      it 'includes employees' do
+        expect(included('employees').map(&:id)).to eq([1, 2])
+      end
+
+      it 'includes relationship identifiers' do
+        jsonapi_data.each do |record|
+          data = record.relationships['employees']['data']
+          expect(data).to_not be_nil
+          expect(data.pluck(:type).uniq).to match_array(['employees'])
+          expect(data.pluck(:id).uniq).to match_array(%w[1 2])
+        end
+      end
+    end
+
+    context 'without include directive and always_include_resource_ids: true' do
+      let(:resource) do
+        Class.new(PORO::TeamResource) do
+          def self.name
+            'PORO::TeamResource'
+          end
+
+          has_many :employees, always_include_resource_ids: true do
+            scope do |employee_ids|
+              {
+                type: :employees,
+                conditions: { employee_id: employee_ids }
+              }
+            end
+          end
+        end
+      end
+
+      before do
+        allow_any_instance_of(PORO::Team).to receive(:employees) { [employee, employee2] }
+        render
+      end
+
+      it 'does not include anything' do
+        expect do
+          included('employees')
+        end.to raise_error(GraphitiSpecHelpers::Errors::NoSideloads)
+      end
+
+      it 'includes relationship identifiers' do
+        jsonapi_data.each do |record|
+          data = record.relationships['employees']['data']
+          expect(data).to_not be_nil
+          expect(data.pluck(:type).uniq).to match_array(['employees'])
+          expect(data.pluck(:id).uniq).to match_array(%w[1 2])
+        end
+      end
+    end
+  end
+
+  describe 'belongs_to' do
+    context 'with include directive' do
+      let(:resource) do
+        Class.new(PORO::PositionResource) do
+          def self.name
+            'PORO::PositionResource'
+          end
+
+          belongs_to :employee
+        end
+      end
+      before do
+        params[:include] = 'employee'
+        render
+      end
+
+      it 'works' do
+        expect(included('employees').map(&:id)).to eq([1])
+      end
+
+      it 'has relationship identifiers' do
+        jsonapi_data.each do |record|
+          data = record.relationships['employee']['data']
+
+          expect(data[:type]).to eq('employees')
+          expect(data[:id]).to eq('1')
+        end
+      end
+    end
+
+    context 'with defaults' do
+      let(:resource) do
+        Class.new(PORO::PositionResource) do
+          def self.name
+            'PORO::PositionResource'
+          end
+
+          belongs_to :employee
+        end
+      end
+
+      before do
+        allow_any_instance_of(PORO::Position).to receive(:employee) { employee }
+        render
+      end
+
+      it 'has relationship ids' do
+        jsonapi_data.each do |record|
+          data = record.relationships['employee']['data']
+
+          expect(data[:type]).to eq('employees')
+          expect(data[:id]).to eq('1')
+        end
+      end
+    end
+
+    context 'with always_include_resource_ids: false' do
+      let(:resource) do
+        Class.new(PORO::PositionResource) do
+          def self.name
+            'PORO::PositionResource'
+          end
+
+          belongs_to :employee, always_include_resource_ids: false do
+            scope do |employee_ids|
+              {
+                type: :employees,
+                conditions: { id: employee_ids }
+              }
+            end
+          end
+        end
+      end
+
+      before do
+        allow_any_instance_of(PORO::Position).to receive(:employee) { employee }
+        render
+      end
+
+      it 'has no relationship identifiers' do
+        jsonapi_data.each do |record|
+          data = record.relationships['employee']
+          expect(data.keys).to_not include('data')
+        end
+      end
+    end
+  end
+end