Added indexes test.

Author: ryanfox1985

spec/annotate/annotate_models_spec.rb

@@ -5,6 +5,14 @@
 require 'active_support/core_ext/string'
 
 describe AnnotateModels do
+  def mock_index(name, columns = [], unique = false)
+    double("IndexKeyDefinition",
+           :name          => name,
+           :columns       => columns,
+           :unique        => unique
+    )
+  end
+
   def mock_foreign_key(name, from_column, to_table, to_column = 'id', constraints = {})
     double("ForeignKeyDefinition",
            :name         => name,
@@ -24,9 +32,9 @@ def mock_connection(indexes = [], foreign_keys = [])
     )
   end
 
-  def mock_class(table_name, primary_key, columns, foreign_keys = [])
+  def mock_class(table_name, primary_key, columns, indexes = [], foreign_keys = [])
     options = {
-      :connection       => mock_connection([], foreign_keys),
+      :connection       => mock_connection(indexes, foreign_keys),
       :table_exists?    => true,
       :table_name       => table_name,
       :primary_key      => primary_key,
@@ -181,7 +189,7 @@ def mock_column(name, type, options={})
            klass = mock_class(:users, :id, [
               mock_column(:id, :integer),
               mock_column(:foreign_thing_id, :integer),
-            ],
+            ], [],
                               [
                                 mock_foreign_key(
                                   'fk_rails_cf2568e89e',
@@ -220,7 +228,7 @@ def mock_column(name, type, options={})
     klass = mock_class(:users, :id, [
        mock_column(:id, :integer),
        mock_column(:foreign_thing_id, :integer),
-     ],
+     ], [],
                        [
                          mock_foreign_key(
                            'fk_rails_02e851e3b7',
@@ -246,6 +254,28 @@ def mock_column(name, type, options={})
 EOS
   end
 
+  it "should get indexes keys" do
+    klass = mock_class(:users, :id, [
+       mock_column(:id, :integer),
+       mock_column(:foreign_thing_id, :integer),
+     ], [mock_index('index_rails_02e851e3b7', ['id']),
+         mock_index('index_rails_02e851e3b8', ['foreign_thing_id'])])
+    expect(AnnotateModels.get_schema_info(klass, "Schema Info", :show_indexes => true)).to eql(<<-EOS)
+# Schema Info
+#
+# Table name: users
+#
+#  id               :integer          not null, primary key
+#  foreign_thing_id :integer          not null
+#
+# Indexes
+#
+#  index_rails_02e851e3b7  (id)
+#  index_rails_02e851e3b8  (foreign_thing_id)
+#
+EOS
+  end
+
   it "should get schema info as RDoc" do
     klass = mock_class(:users, :id, [
                                      mock_column(:id, :integer),