tables can fetch attributes

1 files changed, 41 insertions, 0 deletions

diff --git a/spec/arel/attributes_spec.rb b/spec/arel/attributes_spec.rb
new file mode 100644
index 0000000000..8b437dff9b
--- /dev/null
+++ b/spec/arel/attributes_spec.rb
@@ -0,0 +1,41 @@
+require 'spec_helper'
+
+module Arel
+  describe 'Attributes' do
+    describe 'for' do
+      it 'returns the correct constant for strings' do
+        [:string, :text, :binary].each do |type|
+          column = Struct.new(:type).new type
+          Attributes.for(column).should == Attributes::String
+        end
+      end
+
+      it 'returns the correct constant for ints' do
+        column = Struct.new(:type).new :integer
+        Attributes.for(column).should == Attributes::Integer
+      end
+
+      it 'returns the correct constant for floats' do
+        column = Struct.new(:type).new :float
+        Attributes.for(column).should == Attributes::Float
+      end
+
+      it 'returns the correct constant for decimals' do
+        column = Struct.new(:type).new :decimal
+        Attributes.for(column).should == Attributes::Decimal
+      end
+
+      it 'returns the correct constant for boolean' do
+        column = Struct.new(:type).new :boolean
+        Attributes.for(column).should == Attributes::Boolean
+      end
+
+      it 'returns the correct constant for time' do
+        [:date, :datetime, :timestamp, :time].each do |type|
+          column = Struct.new(:type).new type
+          Attributes.for(column).should == Attributes::Time
+        end
+      end
+    end
+  end
+end