lib/railroady/models_diagram.rb in railroady-1.3.1 vs lib/railroady/models_diagram.rb in railroady-1.4.0

@@ -106,11 +106,11 @@
         content_columns = current_class.columns
       end
 
       content_columns.each do |a|
         content_column = a.name
-        content_column += ' :' + a.type.to_s unless @options.hide_types
+        content_column += ' :' + a.sql_type.to_s unless @options.hide_types
         node_attribs << content_column
       end
     end
     @graph.add_node [node_type, current_class.name, node_attribs]
 
@@ -282,9 +282,11 @@
     if %w(has_one references_one embeds_one).include?(macro)
       assoc_type = 'one-one'
     elsif macro == 'has_many' && (!assoc.options[:through]) ||
           %w(references_many embeds_many).include?(macro)
       assoc_type = 'one-many'
+    elsif macro == 'belongs_to'
+      assoc_type = 'belongs-to'
     else # habtm or has_many, :through
       # Add FAKE associations too in order to understand mistakes
       return if @habtm.include? [assoc_class_name, class_name, assoc_name]
       assoc_type = 'many-many'
       @habtm << [class_name, assoc_class_name, assoc_name]