inferrer_spec.rb in rom-sql-0.9.1

- old
+ new

@@ -1,18 +1,26 @@
 RSpec.describe 'ROM::SQL::Schema::PostgresInferrer', :postgres do
   include_context 'database setup'
 
+  colors = %w(red orange yellow green blue purple)
+
   before do
+    conn.extension :pg_enum
+
     conn.drop_table?(:test_inferrence)
+    conn.drop_enum(:rainbow, if_exists: true)
 
+    conn.create_enum(:rainbow, colors)
+
     conn.create_table :test_inferrence do
       primary_key :id, :uuid
       Json :json_data
       Jsonb :jsonb_data
       Decimal :money, null: false
       column :tags, "text[]"
       column :tag_ids, "bigint[]"
+      rainbow :color
     end
   end
 
   let(:dataset) { :test_inferrence }
 
@@ -31,10 +39,11 @@
         id: ROM::SQL::Types::PG::UUID.meta(name: :id, primary_key: true),
         json_data: ROM::SQL::Types::PG::JSON.optional.meta(name: :json_data),
         jsonb_data: ROM::SQL::Types::PG::JSONB.optional.meta(name: :jsonb_data),
         money: ROM::SQL::Types::Decimal.meta(name: :money),
         tags: ROM::SQL::Types::PG::Array('text').optional.meta(name: :tags),
-        tag_ids: ROM::SQL::Types::PG::Array('biging').optional.meta(name: :tag_ids)
+        tag_ids: ROM::SQL::Types::PG::Array('biging').optional.meta(name: :tag_ids),
+        color: ROM::SQL::Types::String.enum(*colors).optional.meta(name: :color)
       )
     end
   end
 end