lib/pupper/api_associations.rb in pupper-0.1.6 vs lib/pupper/api_associations.rb in pupper-0.1.7

@@ -5,16 +5,16 @@
     class_methods do
       def associations
         @associations ||= { has_one: [], has_many: [] }
       end
 
-      def has_one(assoc)
-        associations[:has_one] << assoc
+      def has_one(*assocs)
+        associations[:has_one].concat(assocs)
       end
 
-      def has_many(assoc)
-        associations[:has_many] << assoc
+      def has_many(*assocs)
+        associations[:has_many].concat(assocs)
       end
     end
 
     included do
       private
@@ -32,14 +32,11 @@
         end
       end
 
       def create_attribute(name, foreign_key = nil, value = nil)
         self.class.attr_accessor(name)
-
-        return unless foreign_key.present?
-
         excluded_attrs << name if foreign_key.present?
-        send("#{name}#{foreign_key.prepend('_')}=", value)
+        send("#{name}=", value)
       end
 
       def find_assoc_type(name)
         associations
           .find { |(_, assoc)| assoc.include?(name) }&.first