1 files changed, 14 insertions, 9 deletions
diff --git a/activerecord/test/cases/adapters/postgresql/ltree_test.rb b/activerecord/test/cases/adapters/postgresql/ltree_test.rb
index 889e369bd6..56516c82b4 100644
--- a/activerecord/test/cases/adapters/postgresql/ltree_test.rb
+++ b/activerecord/test/cases/adapters/postgresql/ltree_test.rb
@@ -1,7 +1,8 @@
-# encoding: utf-8
 require "cases/helper"
+require 'support/schema_dumping_helper'
 
-class PostgresqlLtreeTest < ActiveRecord::TestCase
+class PostgresqlLtreeTest < ActiveRecord::PostgreSQLTestCase
+  include SchemaDumpingHelper
   class Ltree < ActiveRecord::Base
     self.table_name = 'ltrees'
   end
@@ -9,9 +10,7 @@ class PostgresqlLtreeTest < ActiveRecord::TestCase
   def setup
     @connection = ActiveRecord::Base.connection
 
-    unless @connection.extension_enabled?('ltree')
-      @connection.enable_extension 'ltree'
-    end
+    enable_extension!('ltree', @connection)
 
     @connection.transaction do
       @connection.create_table('ltrees') do |t|
@@ -23,16 +22,17 @@ class PostgresqlLtreeTest < ActiveRecord::TestCase
   end
 
   teardown do
-    @connection.execute 'drop table if exists ltrees'
+    @connection.drop_table 'ltrees', if_exists: true
   end
 
   def test_column
     column = Ltree.columns_hash['path']
     assert_equal :ltree, column.type
     assert_equal "ltree", column.sql_type
-    assert_not column.number?
-    assert_not column.binary?
-    assert_not column.array
+    assert_not column.array?
+
+    type = Ltree.type_for_attribute('path')
+    assert_not type.binary?
   end
 
   def test_write
@@ -45,4 +45,9 @@ class PostgresqlLtreeTest < ActiveRecord::TestCase
     ltree = Ltree.first
     assert_equal '1.2.3', ltree.path
   end
+
+  def test_schema_dump_with_shorthand
+    output = dump_table_schema("ltrees")
+    assert_match %r[t\.ltree "path"], output
+  end
 end