spec/support/models.rb in calculated_attributes-0.1.1 vs spec/support/models.rb in calculated_attributes-0.1.2

@@ -1,15 +1,26 @@
 class Post < ActiveRecord::Base
-  calculated :comments, -> { 'select count(*) from comments where comments.post_id = posts.id' }
+  has_many :comments
+  belongs_to :user
+
+  calculated :comments_count, -> { 'select count(*) from comments where comments.post_id = posts.id' }
   calculated :comments_two, -> { 'select count(*) from comments where comments.post_id = posts.id' }
   calculated :comments_arel, -> { Comment.where(Comment.arel_table[:post_id].eq(Post.arel_table[:id])).select(Arel.sql('count(*)')) }
+  calculated :comments_to_sql, -> { Comment.where('comments.post_id = posts.id').select('count(*)') }
 end
 
 class Tutorial < Post
 end
 
 class Article < Post
   calculated :sub_comments, -> { 'select count(*) from comments where comments.post_id = posts.id' }
 end
 
 class Comment < ActiveRecord::Base
+  belongs_to :post
+  belongs_to :user
+end
+
+class User < ActiveRecord::Base
+  has_many :comments
+  has_many :posts
 end