RSpec.describe 'Plugins / :auto_wrap' do
  with_adapters do
    include_context 'users and tasks'

    describe '#for_wrap' do
      shared_context 'joined tuple' do
        let(:name) do
          users.name.relation
        end

        it 'returns joined tuples' do
          task_with_user = tasks
            .for_wrap({ id: :user_id }, name)
            .where(tasks__id: 2)
            .one

          expect(task_with_user).to eql(
            id: 2, user_id: 1, title: "Jane's task", users_name: "Jane", users_id: 1
          )
        end

        it 'works with by_pk' do
          task_with_user = tasks
                             .for_wrap({ id: :user_id }, users.name.relation)
                             .by_pk(1)
                             .one

          expect(task_with_user).
            to eql(id: 1, user_id: 2, title: "Joe's task", users_name: "Joe", users_id: 2)
        end
      end

      context 'when parent relation is registered under dataset name' do
        subject(:tasks) { relations[:tasks] }

        let(:users) { relations[:users] }

        before do
          conf.relation(:tasks) { schema(infer: true) }
          conf.relation(:users) { schema(infer: true) }
        end

        include_context 'joined tuple'
      end

      context 'when parent relation is registered under a custom name' do
        subject(:tasks) { relations[:tasks] }

        let(:users) { relations[:authors] }

        before do
          conf.relation(:tasks) { schema(infer: true) }
          conf.relation(:authors) { schema(:users, infer: true) }
        end

        include_context 'joined tuple'
      end

      context 'using association' do
        subject(:tasks) { relations[:tasks] }

        let(:users) { relations[:users] }

        before do
          conf.relation(:tasks) {
            schema(infer: true) { associations { belongs_to :users, as: :assignee } }
          }

          conf.relation(:users) { schema(infer: true) }
        end

        include_context 'joined tuple' do
          let(:name) { :assignee }
        end
      end
    end
  end
end