Merge pull request #86 from kuroda/without_primary_key

expect a model without primary key

Author: alexch

lib/annotate/annotate_models.rb

@@ -82,7 +82,7 @@ def get_schema_info(klass, header, options = {})
         attrs = []
         attrs << "default(#{quote(col.default)})" unless col.default.nil?
         attrs << "not null" unless col.null
-        attrs << "primary key" if col.name.to_sym == klass.primary_key.to_sym
+        attrs << "primary key" if klass.primary_key && col.name.to_sym == klass.primary_key.to_sym
 
         col_type = (col.type || col.sql_type).to_s
         if col_type == "decimal"

spec/annotate/annotate_models_spec.rb

@@ -8,7 +8,7 @@ def mock_class(table_name, primary_key, columns)
     options = {
       :connection   => mock("Conn", :indexes => []),
       :table_name   => table_name,
-      :primary_key  => primary_key.to_s,
+      :primary_key  => primary_key && primary_key.to_s,
       :column_names => columns.map { |col| col.name.to_s },
       :columns      => columns
     }
@@ -52,6 +52,24 @@ def mock_column(name, type, options={})
 #  name :string(50)       not null
 #
 
+EOS
+  end
+
+  it "should get schema info even if the primary key is not set" do
+    klass = mock_class(:users, nil, [
+                                     mock_column(:id, :integer),
+                                     mock_column(:name, :string, :limit => 50)
+                                    ])
+
+    AnnotateModels.get_schema_info(klass, "Schema Info").should eql(<<-EOS)
+# Schema Info
+#
+# Table name: users
+#
+#  id   :integer          not null
+#  name :string(50)       not null
+#
+
 EOS
   end