spec/integration/repository_spec.rb in rom-repository-0.2.0 vs spec/integration/repository_spec.rb in rom-repository-0.3.0

@@ -6,16 +6,28 @@
 
   it 'loads a single relation' do
     expect(repo.all_users.to_a).to match_array([jane, joe])
   end
 
+  it 'can be used with a custom mapper' do
+    expect(repo.all_users_as_users.to_a).to match_array([
+      Test::Models::User.new(jane),
+      Test::Models::User.new(joe)
+    ])
+  end
+
   it 'loads a relation by an association' do
     expect(repo.tasks_for_users(repo.all_users)).to match_array([jane_task, joe_task])
   end
 
   it 'loads a combine relation with one parent' do
-    expect(repo.task_with_user.first).to eql(task_with_user)
+    task = repo.task_with_user.first
+
+    expect(task.id).to eql(task_with_user.id)
+    expect(task.title).to eql(task_with_user.title)
+    expect(task.user.id).to eql(task_with_user.user.id)
+    expect(task.user.name).to eql(task_with_user.user.name)
   end
 
   it 'loads a combine relation with one parent with custom tuple key' do
     expect(repo.task_with_owner.first).to eql(task_with_owner)
   end
@@ -33,12 +45,95 @@
       jane_without_task, joe_with_task
     ])
   end
 
   it 'loads nested combined relations' do
-    expect(repo.users_with_tasks_and_tags.first).to eql(user_with_task_and_tags)
+    expect(repo.users_with_tasks_and_tags.first.to_h).to eql(user_with_task_and_tags.to_h)
   end
 
   it 'loads a wrapped relation' do
     expect(repo.tag_with_wrapped_task.first).to eql(tag_with_task)
+  end
+
+  it 'loads an aggregate via custom fks' do
+    jane = repo.aggregate(many: repo.posts).where(name: 'Jane').one
+
+    expect(jane.posts.size).to be(1)
+    expect(jane.posts.first.title).to eql('Hello From Jane')
+  end
+
+  it 'loads an aggregate via assoc name' do
+    jane = repo.aggregate(:posts).where(name: 'Jane').one
+
+    expect(jane.posts.size).to be(1)
+    expect(jane.posts.first.title).to eql('Hello From Jane')
+  end
+
+  it 'loads an aggregate with multiple associations' do
+    jane = repo.aggregate(:posts, :labels).where(name: 'Jane').one
+
+    expect(jane.posts.size).to be(1)
+    expect(jane.posts.first.title).to eql('Hello From Jane')
+
+    expect(jane.labels.size).to be(2)
+    expect(jane.labels[0].name).to eql('red')
+    expect(jane.labels[1].name).to eql('blue')
+  end
+
+  it 'loads a parent via custom fks' do
+    post = repo.posts.combine(:author).where(title: 'Hello From Jane').one
+
+    expect(post.title).to eql('Hello From Jane')
+    expect(post.author.name).to eql('Jane')
+  end
+
+  it 'loads aggregate through many-to-many via custom options' do
+    post = repo.posts
+      .combine_children(many: repo.labels)
+      .where(title: 'Hello From Jane')
+      .one
+
+    expect(post.title).to eql('Hello From Jane')
+    expect(post.labels.size).to be(2)
+    expect(post.labels.map(&:name)).to eql(%w(red blue))
+  end
+
+  it 'loads aggregate through many-to-many association' do
+    post = repo.posts.combine(:labels).where(title: 'Hello From Jane').one
+
+    expect(post.title).to eql('Hello From Jane')
+    expect(post.labels.size).to be(2)
+    expect(post.labels.map(&:name)).to eql(%w(red blue))
+  end
+
+  context 'not common naming conventions' do
+    it 'still loads nested relations' do
+      comments = comments_repo.comments_with_likes.to_a
+
+      expect(comments.size).to be(2)
+      expect(comments[0].author).to eql('Jane')
+      expect(comments[0].likes[0].author).to eql('Joe')
+      expect(comments[0].likes[1].author).to eql('Anonymous')
+      expect(comments[1].author).to eql('Joe')
+      expect(comments[1].likes[0].author).to eql('Jane')
+    end
+
+    it 'loads nested relations by association name' do
+      comments = comments_repo.comments_with_emotions.to_a
+
+      expect(comments.size).to be(2)
+      expect(comments[0].emotions[0].author).to eql('Joe')
+      expect(comments[0].emotions[1].author).to eql('Anonymous')
+    end
+  end
+
+  context 'with a table without columns' do
+    before { conn.create_table(:dummy) unless conn.table_exists?(:dummy) }
+
+    it 'does not fail with a weird error when a relation does not have attributes' do
+      configuration.relation(:dummy)
+
+      repo = Class.new(ROM::Repository[:dummy]).new(rom)
+      expect(repo.dummy.to_a).to eql([])
+    end
   end
 end