Add Mongoid::Document.primary_key (with spec)

activeadmin · Oct 30, 2012 · e9b7edf · e9b7edf
1 parent d449bc8
commit e9b7edf
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 2 deletions.
diff --git a/lib/active_admin/mongoid/document.rb b/lib/active_admin/mongoid/document.rb
@@ -2,7 +2,7 @@
 
 module ActiveAdmin::Mongoid::Document
   extend ActiveSupport::Concern
-  
+
   module ClassMethods
     def content_columns
       @content_columns ||= fields.map(&:second).select {|f| f.name !~ /(^_|^(created|updated)_at)/}
@@ -11,10 +11,14 @@ def content_columns
     def columns
       @columns ||= fields.map(&:second)
     end
-    
+
     def reorder *args
       scoped
     end
+
+    def primary_key
+      :_id
+    end
   end
 end
 

diff --git a/spec/active_admin/mongoid/document_spec.rb b/spec/active_admin/mongoid/document_spec.rb
@@ -0,0 +1,12 @@
+require 'spec_helper'
+require 'active_admin/mongoid/document'
+
+describe ActiveAdmin::Mongoid::Document do
+  # let(:resource_class) { Class.new.tap { include Mongoid::Document } }
+  describe '.primary_key' do
+    it 'responds' do
+      Post.primary_key.should eq(:_id)
+      # resource_class.human_attribute_name(resource_class.primary_key)
+    end
+  end
+end