Add test for STI referencing

spec/fixtures/models/article_page.rb

@@ -0,0 +1,4 @@
+require_relative './page'
+
+class ArticlePage < Page
+end

spec/fixtures/models/page.rb

@@ -0,0 +1,3 @@
+class Page < ActiveRecord::Base
+  validates :title, :presence => true
+end

spec/fixtures/seeds/test/article_pages.yml

@@ -0,0 +1,5 @@
+records:
+  - sprig_id: 1
+    title: "<%= sprig_record(Page, 1).title %>"
+  - sprig_id: 2
+    title: Second Title

spec/fixtures/seeds/test/pages.yml

@@ -0,0 +1,5 @@
+records:
+  - sprig_id: 1
+    title: First Title
+  - sprig_id: 2
+    title: "<%= sprig_record(ArticlePage, 2).title %>"

spec/spec_helper.rb

@@ -64,6 +64,9 @@
 Tag.connection.execute "DROP TABLE IF EXISTS posts_tags;"
 Tag.connection.execute "CREATE TABLE posts_tags (id INTEGER PRIMARY KEY , post_id INTEGER, tag_id INTEGER);"
 
+Page.connection.execute "DROP TABLE IF EXISTS pages;"
+Page.connection.execute "CREATE TABLE pages (id INTEGER PRIMARY KEY , title VARCHAR(255), type VARCHAR(255));"
+
 # Helpers
 #
 # Setup fake `Rails.root`

spec/sprig_spec.rb

@@ -253,6 +253,23 @@
     end
   end
 
+  context "with STI records" do
+    around do |example|
+      load_seeds('article_pages.yml', 'pages.yml', &example)
+    end
+
+    it "allows cross-referencing of STI records" do
+      sprig [
+        ArticlePage,
+        Page
+      ]
+
+      Page.all.map(&:title).should == [
+        'First Title', 'First Title', 'Second Title', 'Second Title'
+      ]
+    end
+  end
+
   context "with cyclic dependencies" do
     around do |example|
       load_seeds('comments.yml', 'posts_with_cyclic_dependencies.yml', &example)