Merge branch 'master' into 4523-search-index

[arvados.git] / services / api / test / unit / arvados_model_test.rb

diff --git a/services/api/test/unit/arvados_model_test.rb b/services/api/test/unit/arvados_model_test.rb
index e47aa3b8921fc067a8ea5d24c0b7f59eaa231853..ca9198fd98978eefddee78811ac09835364ca250 100644 (file)
--- a/services/api/test/unit/arvados_model_test.rb
+++ b/services/api/test/unit/arvados_model_test.rb
@@ -86,4 +86,62 @@ class ArvadosModelTest < ActiveSupport::TestCase
                    properties: {'foo' => 'bar'}.with_indifferent_access)
     end
   end
+
+  test "unique uuid index exists on all models with the column uuid" do
+    tables = ActiveRecord::Base.connection.tables
+    tables.each do |table|
+      columns = ActiveRecord::Base.connection.columns(table)
+
+      uuid_column = columns.select do |column|
+        column.name == 'uuid'
+      end
+
+      if !uuid_column.empty?
+        indexes = ActiveRecord::Base.connection.indexes(table)
+        uuid_index = indexes.select do |index|
+          index.columns == ['uuid'] and index.unique == true
+        end
+
+        assert !uuid_index.empty?, "#{table} does not have unique uuid index"
+      end
+    end
+  end
+
+  test "owner uuid index exists on all models with the owner_uuid column" do
+    all_tables = ActiveRecord::Base.connection.tables
+
+    all_tables.each do |table|
+      columns = ActiveRecord::Base.connection.columns(table)
+
+      uuid_column = columns.select do |column|
+        column.name == 'owner_uuid'
+      end
+
+      if !uuid_column.empty?
+        indexes = ActiveRecord::Base.connection.indexes(table)
+        owner_uuid_index = indexes.select do |index|
+          index.columns == ['owner_uuid']
+        end
+        assert !owner_uuid_index.empty?, "#{table} does not have owner_uuid index"
+      end
+    end
+  end
+
+  test "search index exists on models that go into projects" do
+    all_tables =  ActiveRecord::Base.connection.tables
+    all_tables.delete 'schema_migrations'
+
+    all_tables.each do |table|
+      table_class = table.classify.constantize
+      if table_class.respond_to?('searchable_columns')
+        search_index_columns = table_class.searchable_columns('ilike')
+
+        indexes = ActiveRecord::Base.connection.indexes(table)
+        search_index = indexes.select do |index|
+          index.columns == search_index_columns
+        end
+        assert !search_index.empty?, "#{table} does not have search index with all searchable columns #{search_index_columns}"
+      end
+    end
+  end
 end