openfoodnetwork/spec/models/enterprise_relationship_spec.rb

require 'spec_helper'

describe EnterpriseRelationship do
  describe "scopes" do
    let(:e1)  { create(:enterprise, name: 'A') }
    let(:e2)  { create(:enterprise, name: 'B') }
    let(:e3)  { create(:enterprise, name: 'C') }

    it "sorts by child, parent enterprise name" do
      er1 = create(:enterprise_relationship, parent: e3, child: e1)
      er2 = create(:enterprise_relationship, parent: e1, child: e2)
      er3 = create(:enterprise_relationship, parent: e2, child: e1)

      EnterpriseRelationship.by_name.should == [er3, er1, er2]
    end

    describe "finding relationships involving some enterprises" do
      let!(:er) { create(:enterprise_relationship, parent: e1, child: e2) }

      it "returns relationships where an enterprise is the parent" do
        EnterpriseRelationship.involving_enterprises([e1]).should == [er]
      end

      it "returns relationships where an enterprise is the child" do
        EnterpriseRelationship.involving_enterprises([e2]).should == [er]
      end

      it "does not return other relationships" do
        EnterpriseRelationship.involving_enterprises([e3]).should == []
      end
    end

    describe "creating with a permission list" do
      it "creates permissions with a list" do
        er = EnterpriseRelationship.create! parent: e1, child: e2, permissions_list: ['one', 'two']
        er.reload
        er.permissions.map(&:name).should match_array ['one', 'two']
      end

      it "does nothing when the list is nil" do
        er = EnterpriseRelationship.create! parent: e1, child: e2, permissions_list: nil
        er.reload
        er.permissions.should be_empty
      end
    end

    describe "finding by permission" do
      let!(:er1) { create(:enterprise_relationship, parent: e2, child: e1) }
      let!(:er2) { create(:enterprise_relationship, parent: e3, child: e2) }
      let!(:er3) { create(:enterprise_relationship, parent: e1, child: e3) }

      it "finds relationships that grant permissions to some enterprises" do
        EnterpriseRelationship.permitting([e1, e2]).should match_array [er1, er2]
      end

      it "finds relationships that are granted by particular enterprises" do
        EnterpriseRelationship.permitted_by([e1, e2]).should match_array [er1, er3]
      end
    end

    it "finds relationships that grant a particular permission" do
      er1 = create(:enterprise_relationship, parent: e1, child: e2,
                   permissions_list: ['one', 'two'])
      er2 = create(:enterprise_relationship, parent: e2, child: e3,
                   permissions_list: ['two', 'three'])
      er3 = create(:enterprise_relationship, parent: e3, child: e1,
                   permissions_list: ['three', 'four'])

      EnterpriseRelationship.with_permission('two').should match_array [er1, er2]
    end
  end

  describe "finding relatives" do
    let(:e1) { create(:supplier_enterprise) }
    let(:e2) { create(:distributor_enterprise) }
    let!(:er) { create(:enterprise_relationship, parent: e1, child: e2) }
    let(:er_reverse) { create(:enterprise_relationship, parent: e2, child: e1) }

    it "includes self where appropriate" do
      EnterpriseRelationship.relatives[e2.id][:distributors].should include e2.id
      EnterpriseRelationship.relatives[e2.id][:producers].should_not include e2.id
    end

    it "categorises enterprises into distributors and producers" do
      e2.update_attribute :is_primary_producer, true
      EnterpriseRelationship.relatives.should ==
        {e1.id => {distributors: Set.new([e2.id]), producers: Set.new([e1.id, e2.id])},
         e2.id => {distributors: Set.new([e2.id]), producers: Set.new([e2.id, e1.id])}}
    end

    it "finds inactive enterprises by default" do
      e1.update_attribute :confirmed_at, nil
      EnterpriseRelationship.relatives[e2.id][:producers].should == Set.new([e1.id])
    end

    it "does not find inactive enterprises when requested" do
      e1.update_attribute :confirmed_at, nil
      EnterpriseRelationship.relatives(true)[e2.id][:producers].should be_empty
    end

    it "does not show duplicates" do
      er_reverse
      EnterpriseRelationship.relatives[e2.id][:producers].should == Set.new([e1.id])
    end
  end
end