spec/integration/relations/reading_spec.rb in rom-0.5.0 vs spec/integration/relations/reading_spec.rb in rom-0.6.0.beta1

@@ -16,18 +16,23 @@
       end
     end
 
     setup.mappers do
       define(:users) do
-        model name: 'User'
+        model name: 'Test::User'
+
+        attribute :name
+        attribute :email
       end
     end
 
+    rom = setup.finalize
+
     users = rom.read(:users).sorted.by_name('Jane')
     user = users.first
 
-    expect(user).to be_an_instance_of(User)
+    expect(user).to be_an_instance_of(Test::User)
     expect(user.name).to eql 'Jane'
     expect(user.email).to eql '[e-mail]'
   end
 
   it 'maps grouped relations' do
@@ -43,40 +48,43 @@
       end
     end
 
     setup.mappers do
       define(:users) do
-        model name: 'User'
+        model name: 'Test::User'
+
+        attribute :name
+        attribute :email
       end
 
       define(:with_tasks, parent: :users) do
-        model name: 'UserWithTasks'
+        model name: 'Test::UserWithTasks'
 
         group tasks: [:title, :priority]
       end
     end
 
     rom = setup.finalize
 
-    User.send(:include, Equalizer.new(:name, :email))
-    UserWithTasks.send(:include, Equalizer.new(:name, :email, :tasks))
+    Test::User.send(:include, Equalizer.new(:name, :email))
+    Test::UserWithTasks.send(:include, Equalizer.new(:name, :email, :tasks))
 
     keys = rom.read(:users).with_tasks.header.keys
     expect(keys).to eql([:name, :email, :tasks])
 
     user = rom.read(:users).sorted.first
 
     expect(user).to eql(
-      User.new(name: "Jane", email: "[e-mail]")
+      Test::User.new(name: "Jane", email: "[e-mail]")
     )
 
     expect(rom.read(:users)).to_not respond_to(:join)
 
     user = rom.read(:users).with_tasks.sorted.first
 
     expect(user).to eql(
-      UserWithTasks.new(
+      Test::UserWithTasks.new(
         name: "Jane",
         email: "[e-mail]",
         tasks: [{ title: "be cool", priority: 2 }])
     )
   end
@@ -94,33 +102,36 @@
       end
     end
 
     setup.mappers do
       define(:users) do
-        model name: 'User'
+        model name: 'Test::User'
+
+        attribute :name
+        attribute :email
       end
 
       define(:with_task, parent: :users) do
-        model name: 'UserWithTask'
+        model name: 'Test::UserWithTask'
 
         wrap task: [:title, :priority]
       end
     end
 
     rom = setup.finalize
 
-    User.send(:include, Equalizer.new(:name, :email))
-    UserWithTask.send(:include, Equalizer.new(:name, :email, :task))
+    Test::User.send(:include, Equalizer.new(:name, :email))
+    Test::UserWithTask.send(:include, Equalizer.new(:name, :email, :task))
 
     keys = rom.read(:users).with_task.header.keys
     expect(keys).to eql([:name, :email, :task])
 
     user = rom.read(:users).sorted.with_task.first
 
     expect(user).to eql(
-      UserWithTask.new(name: "Jane", email: "[e-mail]",
-                       task: { title: "be cool", priority: 2 })
+      Test::UserWithTask.new(name: "Jane", email: "[e-mail]",
+                             task: { title: "be cool", priority: 2 })
     )
   end
 
   it 'maps hashes' do
     setup.relation(:users) do
@@ -136,7 +147,64 @@
     rom = setup.finalize
 
     user = rom.read(:users).by_name("Jane").first
 
     expect(user).to eql(name: "Jane", email: "[e-mail]")
+  end
+
+  it 'allows cherry-picking of a mapper' do
+    setup.relation(:users)
+
+    setup.mappers do
+      define(:users) do
+        attribute :name
+        attribute :email
+      end
+
+      define(:prefixer, parent: :users) do
+        attribute :user_name, from: :name
+        attribute :user_email, from: :email
+      end
+    end
+
+    rom = setup.finalize
+    user = rom.read(:users).map(:prefixer).first
+
+    expect(user).to eql(user_name: 'Joe', user_email: "[e-mail]")
+  end
+
+  it 'allows passing a block to retrieve relations for mapping' do
+    setup.relation(:users) do
+      def by_name(name)
+        restrict(name: name)
+      end
+    end
+
+    setup.mappers do
+      define(:users) do
+        attribute :name
+        attribute :email
+      end
+
+      define(:prefixer, parent: :users) do
+        attribute :user_name, from: :name
+        attribute :user_email, from: :email
+      end
+    end
+
+    rom = setup.finalize
+
+    expect {
+      rom.relation(:users) { |users| users.not_here }
+    }.to raise_error(NoMethodError, /not_here/)
+
+    expect {
+      rom.relation(:users) { |users| users.by_name('Joe') }.as(:not_here)
+    }.to raise_error(ROM::MapperMissingError, /not_here/)
+
+    user = rom.relation(:users) { |users|
+      users.by_name('Joe')
+    }.map_with(:prefixer).call.first
+
+    expect(user).to eql(user_name: 'Joe', user_email: "[e-mail]")
   end
 end