spec/unit/relation/assoc_spec.rb in rom-sql-1.0.0.rc2 vs spec/unit/relation/assoc_spec.rb in rom-sql-1.0.0

@@ -1,38 +1,87 @@
 require 'spec_helper'
 
-RSpec.describe ROM::SQL::Relation, :sqlite do
+RSpec.describe ROM::SQL::Relation do
   include_context 'users and tasks'
 
-  let(:users) { relations[:users] }
-  let(:tasks) { relations[:tasks] }
+  context 'with has_many' do
+    subject(:users) { relations[:users] }
 
-  before do
-    conf.relation(:users) do
-      schema(infer: true) do
-        associations do
-          has_many :tasks
+    let(:tasks) { relations[:tasks] }
+
+    before do
+      conf.relation(:users) do
+        schema(infer: true) do
+          associations do
+            has_many :tasks
+          end
         end
       end
     end
 
-    conf.relation(:tasks) do
-      schema(infer: true) do
-        associations do
-          belongs_to :users, as: :user
+    with_adapters do
+      it 'returns child tuples for a relation' do
+        expect(users.assoc(:tasks).where(name: 'Jane').to_a).
+          to eql([{ id: 2, user_id: 1, title: "Jane's task" }])
+      end
+    end
+  end
+
+  context 'with has_many-through' do
+    subject(:tasks) { relations[:tasks] }
+
+    before do
+      conf.relation(:task_tags) do
+        schema(infer: true) do
+          associations do
+            belongs_to :tasks
+            belongs_to :tags
+          end
         end
       end
+
+      conf.relation(:tasks) do
+        schema(infer: true) do
+          associations do
+            has_many :tags, through: :task_tags
+          end
+        end
+      end
+
+      conn[:tags].insert id: 2, name: 'whatevah'
+      conn[:task_tags].insert(tag_id: 2, task_id: 2)
     end
+
+    with_adapters do
+      it 'returns child tuples for a relation' do
+        expect(tasks.assoc(:tags).to_a).
+          to eql([{ id: 1, name: 'important', task_id: 1 }, { id: 2, name: 'whatevah', task_id: 2 }])
+      end
+
+      it 'returns child tuples for a restricted relation' do
+        expect(tasks.assoc(:tags).where(title: "Jane's task").to_a).
+          to eql([{ id: 2, name: 'whatevah', task_id: 2 }])
+      end
+    end
   end
 
-  with_adapters(:sqlite) do
-    it 'returns child tuples for a relation' do
-      expect(users.assoc(:tasks).where(name: 'Jane').to_a).
-        to eql([{ id: 2, user_id: 1, title: "Jane's task" }])
+  context 'with belongs_to' do
+    subject(:tasks) { relations[:tasks] }
+
+    before do
+      conf.relation(:tasks) do
+        schema(infer: true) do
+          associations do
+            belongs_to :users, as: :user
+          end
+        end
+      end
     end
 
-    it 'returns parent tuples for a relation' do
-      expect(tasks.assoc(:user).where(title: "Jane's task").to_a).
-        to eql([{ id: 1, task_id: 2, name: 'Jane' }])
+    with_adapters do
+      it 'returns parent tuples for a relation' do
+        expect(tasks.assoc(:user).where(title: "Jane's task").to_a).
+          to eql([{ id: 1, task_id: 2, name: 'Jane' }])
+      end
     end
   end
 end