Added collection_fields to Collection

gabynaiman · Nov 29, 2019 · db299fe · db299fe
1 parent c3d5b96
commit db299fe
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/lib/rasti/db/collection.rb b/lib/rasti/db/collection.rb
@@ -16,6 +16,10 @@ def collection_name
           @collection_name ||= underscore(demodulize(name)).to_sym
         end
 
+        def collection_fields
+          @collection_fields ||= model.attributes - relations.keys
+        end
+
         def primary_key
           @primary_key ||= :id
         end

diff --git a/spec/collection_spec.rb b/spec/collection_spec.rb
@@ -6,13 +6,15 @@
 
     it 'Implicit' do
       Users.collection_name.must_equal :users
+      Users.collection_fields.must_equal [:id, :name]
       Users.model.must_equal User
       Users.primary_key.must_equal :id
       Users.foreign_key.must_equal :user_id
     end
 
     it 'Explicit' do
       People.collection_name.must_equal :people
+      People.collection_fields.must_equal [:document_number, :first_name, :last_name, :birth_date, :user_id]
       People.model.must_equal Person
       People.primary_key.must_equal :document_number
       People.foreign_key.must_equal :document_number