lib/dato/json_schema_relationships.rb in dato-0.6.14 vs lib/dato/json_schema_relationships.rb in dato-0.6.15

@@ -1,16 +1,20 @@
 module Dato
   class JsonSchemaRelationships
-    attr_reader :link
+    attr_reader :schema
 
-    def initialize(link)
-      @link = link
+    def initialize(schema)
+      @schema = schema
     end
 
     def relationships
-      return {} unless link
+      if !schema || !schema.properties['data'] || !schema.properties['data'].properties['relationships']
+        return {}
+      end
 
-      link.properties.each_with_object({}) do |(relationship, schema), acc|
+      relationships = schema.properties['data'].properties['relationships'].properties
+
+      relationships.each_with_object({}) do |(relationship, schema), acc|
         is_collection = schema.properties['data'].type.first == 'array'
 
         types = if is_collection
                        [type(schema.properties['data'].items)]
                      elsif schema.properties['data'].type.first == 'object'