lib/quick_travel/adapter.rb in quicktravel_client-2.3.1 vs lib/quick_travel/adapter.rb in quicktravel_client-2.4.0

@@ -44,9 +44,37 @@
           }
         end
       end
     end
 
+    def self.has_many(relation_name, options = {})
+      define_method relation_name do
+        instance_variable_get("@#{relation_name}") || instance_variable_set(
+          "@#{relation_name}",
+          begin
+            klass = QuickTravel.const_get(options[:class_name] || relation_name.to_s.singularize.classify)
+            instance_variable_get("@#{relation_name}_attributes").map { |attr|
+              klass.new(attr)
+            }
+          end
+        )
+      end
+    end
+
+    def self.belongs_to(relation_name, options = {})
+      define_method relation_name do
+        instance_variable_get("@#{relation_name}") || instance_variable_set(
+          "@#{relation_name}",
+          begin
+            attrs = instance_variable_get("@#{relation_name}_attributes")
+            return nil unless attrs
+            klass = QuickTravel.const_get(options[:class_name] || relation_name.to_s.singularize.classify)
+            klass.new(attrs)
+          end
+        )
+      end
+    end
+
     def self.find(id, opts = {})
       check_id!(id)
       if lookup
         all.detect { |o| o.id == id.to_i }
       else