sexy_pg_constraints 0.1.2 → 0.2.0

data/Manifest

@@ -11,4 +11,8 @@ lib/sexy_pg_constraints.rb
 sexy_pg_constraints.gemspec
 test/postgresql_adapter.rb
 test/sexy_pg_constraints_test.rb
+test/support/assert_prohibits_allows.rb
+test/support/database.yml
+test/support/database.yml.example
+test/support/models.rb
 test/test_helper.rb

data/README.rdoc

@@ -106,6 +106,8 @@ Below is the list of constraints available and tested so far.
 * even
 * odd
 * format
+* lowercase
+* xor
 
 == Extensibility
 
@@ -113,8 +115,12 @@ All constraints are located in the lib/constraints.rb. Extending this module wit
 
 == TODO
 
+* Add support for Rails schema.rb
 * Create better API for adding constraints
 
+== Contributors
+* Empact[http://github.com/Empact] (Big thanks for lots of work. Better flexibility, more tests, organizing code, bug fixes.)
+
 == License
 
 Copyright (c) 2008 Maxim Chernyak

data/Rakefile

@@ -2,7 +2,7 @@ require 'rubygems'
 require 'rake'
 require 'echoe'
 
-Echoe.new('sexy_pg_constraints', '0.1.2') do |p|
+Echoe.new('sexy_pg_constraints', '0.2.0') do |p|
   p.description     = "Use migrations and simple syntax to manage constraints in PostgreSQL DB."
   p.summary         = "If you're on PostgreSQL and see the importance of data-layer constraints - this gem/plugin is for you. It integrates constraints into PostgreSQL adapter so you can add/remove them in your migrations. You get two simple methods for adding/removing constraints, as well as a pack of pre-made constraints."
   p.url             = "http://github.com/maxim/sexy_pg_constraints"

data/lib/constrainer.rb

@@ -24,7 +24,7 @@ module SexyPgConstraints
       def add_constraints(table, column, constraints)
         constraints.each_pair do |type, options|
           sql = "alter table #{table} add constraint #{make_title(table, column, type)} " + 
-            SexyPgConstraints::Constraints.send(type, column, options) + ';'
+            SexyPgConstraints::Constraints.send(type, table, column, options) + ';'
           
           execute sql
         end

data/lib/constraints.rb

@@ -1,38 +1,38 @@
 module SexyPgConstraints
   module Constraints
     module_function
-    
+
     ##
     # Only allow listed values.
     #
-    # Example: 
+    # Example:
     #   constrain :books, :variation, :whitelist => %w(hardcover softcover)
     #
-    def whitelist(column, options)
-      "check (#{column} in (#{ options.collect{|v| "'#{v}'"}.join(',')  }))"
+    def whitelist(table, column, options)
+      "check (#{table}.#{column} in (#{ options.collect{|v| "'#{v}'"}.join(',')  }))"
     end
-  
-    ## 
+
+    ##
     # Prohibit listed values.
     #
-    # Example: 
+    # Example:
     #   constrain :books, :isbn, :blacklist => %w(invalid_isbn1 invalid_isbn2)
     #
-    def blacklist(column, options)
-      "check (#{column} not in (#{ options.collect{|v| "'#{v}'"}.join(',') }))"
+    def blacklist(table, column, options)
+      "check (#{table}.#{column} not in (#{ options.collect{|v| "'#{v}'"}.join(',') }))"
     end
-  
-    ## 
+
+    ##
     # The value must have at least 1 non-space character.
     #
     # Example:
     #   constrain :books, :title, :not_blank => true
     #
-    def not_blank(column, options)
-      "check ( length(trim(both from #{column})) > 0 )"
+    def not_blank(table, column, options)
+      "check ( length(trim(both from #{table}.#{column})) > 0 )"
     end
 
-    ## 
+    ##
     # The numeric value must be within given range.
     #
     # Example:
@@ -40,115 +40,140 @@ module SexyPgConstraints
     #   constrain :books, :year, :within => 1980...2009
     # (the two lines above do the same thing)
     #
-    def within(column, options)
-      "check (#{column} >= #{options.begin} and #{column} #{options.exclude_end? ? ' < ' : ' <= '} #{options.end})"
+    def within(table, column, options)
+      column_ref = column.to_s.include?('.') ? column : "#{table}.#{column}"
+      "check (#{column_ref} >= #{options.begin} and #{column_ref} #{options.exclude_end? ? ' < ' : ' <= '} #{options.end})"
     end
 
-    ## 
+    ##
     # Check the length of strings/text to be within the range.
     #
     # Example:
     #   constrain :books, :author, :length_within => 4..50
     #
-    def length_within(column, options)
-      within("length(#{column})", options)
+    def length_within(table, column, options)
+      within(table, "length(#{table}.#{column})", options)
     end
-    
-    ## 
+
+    ##
     # Allow only valid email format.
     #
     # Example:
     #   constrain :books, :author, :email => true
     #
-    def email(column, options)
-      "check (((#{column})::text ~ E'^([-a-z0-9]+)@([-a-z0-9]+[.]+[a-z]{2,4})$'::text))"
+    def email(table, column, options)
+      "check (((#{table}.#{column})::text ~ E'^([-a-z0-9]+)@([-a-z0-9]+[.]+[a-z]{2,4})$'::text))"
     end
 
-    ## 
+    ##
     # Allow only alphanumeric values.
     #
     # Example:
     #   constrain :books, :author, :alphanumeric => true
     #
-    def alphanumeric(column, options)
-      "check (((#{column})::text ~* '^[a-z0-9]+$'::text))"
+    def alphanumeric(table, column, options)
+      "check (((#{table}.#{column})::text ~* '^[a-z0-9]+$'::text))"
+    end
+
+    ##
+    # Allow only lower case values.
+    #
+    # Example:
+    #   constrain :books, :author, :lowercase => true
+    #
+    def lowercase(table, column, options)
+      "check (#{table}.#{column} = lower(#{table}.#{column}))"
     end
-    
-    ## 
+
+    ##
     # Allow only positive values.
     #
     # Example:
     #   constrain :books, :quantity, :positive => true
     #
-    def positive(column, options)
-      "check (#{column} >= 0)"
+    def positive(table, column, options)
+      "check (#{table}.#{column} >= 0)"
     end
-    
-    ## 
+
+    ##
     # Allow only odd values.
     #
     # Example:
     #   constrain :books, :quantity, :odd => true
     #
-    def odd(column, options)
-      "check (mod(#{column}, 2) != 0)"
+    def odd(table, column, options)
+      "check (mod(#{table}.#{column}, 2) != 0)"
     end
 
-    ## 
+    ##
     # Allow only even values.
     #
     # Example:
     #   constrain :books, :quantity, :even => true
     #
-    def even(column, options)
-      "check (mod(#{column}, 2) = 0)"
+    def even(table, column, options)
+      "check (mod(#{table}.#{column}, 2) = 0)"
     end
 
-    ## 
+    ##
     # Make sure every entry in the column is unique.
     #
     # Example:
     #   constrain :books, :isbn, :unique => true
     #
-    def unique(column, options)
-      column = column.join(', ') if column.respond_to?(:join)
+    def unique(table, column, options)
+      column = Array(column).map {|c| %{"#{c}"} }.join(', ')
       "unique (#{column})"
     end
-    
-    ## 
+
+    ##
+    # Allow only one of the values in the given columns to be true.
+    # Only reasonable with more than one column.
+    # See Enterprise Rails, Chapter 10 for details.
+    #
+    # Example:
+    #   constrain :books, [], :xor => true
+    #
+    def xor(table, column, options)
+      addition = Array(column).map {|c| %{("#{c}" is not null)::integer} }.join(' + ')
+
+      "check (#{addition} = 1)"
+    end
+
+    ##
     # Allow only text/strings of the exact length specified, no more, no less.
     #
     # Example:
     #   constrain :books, :hash, :exact_length => 32
     #
-    def exact_length(column, options)
-      "check ( length(trim(both from #{column})) = #{options} )"
+    def exact_length(table, column, options)
+      "check ( length(trim(both from #{table}.#{column})) = #{options} )"
     end
-    
-    ## 
+
+    ##
     # Allow only values that match the regular expression.
     #
     # Example:
     #   constrain :orders, :visa, :format => /^([4]{1})([0-9]{12,15})$/
     #
-    def format(column, options)
-      "check (((#{column})::text #{options.casefold? ? '~*' : '~'}  E'#{options.source}'::text ))"
+    def format(table, column, options)
+      "check (((#{table}.#{column})::text #{options.casefold? ? '~*' : '~'}  E'#{options.source}'::text ))"
     end
-    
-    ## 
+
+    ##
     # Add foreign key constraint.
     #
     # Example:
     #   constrain :books, :author_id, :reference => {:authors => :id, :on_delete => :cascade}
     #
-    def reference(column, options)
+    def reference(table, column, options)
       on_delete = options.delete(:on_delete)
       fk_table = options.keys.first
       fk_column = options[fk_table]
-      
+
       on_delete = "on delete #{on_delete}" if on_delete
-      
-      "foreign key (#{column}) references #{fk_table} (#{fk_column}) #{on_delete}"
+
+      %{foreign key ("#{column}") references #{fk_table} (#{fk_column}) #{on_delete}}
     end
   end
-end
+end

data/sexy_pg_constraints.gemspec

@@ -2,15 +2,15 @@
 
 Gem::Specification.new do |s|
   s.name = %q{sexy_pg_constraints}
-  s.version = "0.1.2"
+  s.version = "0.2.0"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 1.2") if s.respond_to? :required_rubygems_version=
   s.authors = ["Maxim Chernyak"]
-  s.date = %q{2009-10-04}
+  s.date = %q{2010-02-01}
   s.description = %q{Use migrations and simple syntax to manage constraints in PostgreSQL DB.}
   s.email = %q{max@bitsonnet.com}
   s.extra_rdoc_files = ["CHANGELOG.rdoc", "README.rdoc", "lib/constrainer.rb", "lib/constraints.rb", "lib/deconstrainer.rb", "lib/helpers.rb", "lib/sexy_pg_constraints.rb"]
-  s.files = ["CHANGELOG.rdoc", "Manifest", "README.rdoc", "Rakefile", "init.rb", "lib/constrainer.rb", "lib/constraints.rb", "lib/deconstrainer.rb", "lib/helpers.rb", "lib/sexy_pg_constraints.rb", "sexy_pg_constraints.gemspec", "test/postgresql_adapter.rb", "test/sexy_pg_constraints_test.rb", "test/test_helper.rb"]
+  s.files = ["CHANGELOG.rdoc", "Manifest", "README.rdoc", "Rakefile", "init.rb", "lib/constrainer.rb", "lib/constraints.rb", "lib/deconstrainer.rb", "lib/helpers.rb", "lib/sexy_pg_constraints.rb", "sexy_pg_constraints.gemspec", "test/postgresql_adapter.rb", "test/sexy_pg_constraints_test.rb", "test/support/assert_prohibits_allows.rb", "test/support/database.yml", "test/support/database.yml.example", "test/support/models.rb", "test/test_helper.rb"]
   s.homepage = %q{http://github.com/maxim/sexy_pg_constraints}
   s.rdoc_options = ["--line-numbers", "--inline-source", "--title", "Sexy_pg_constraints", "--main", "README.rdoc"]
   s.require_paths = ["lib"]

data/test/sexy_pg_constraints_test.rb

@@ -1,572 +1,899 @@
 require File.dirname(__FILE__) + '/test_helper.rb'
 
-# Database spc_test should be created manually.
-ActiveRecord::Base.establish_connection(:adapter => "postgresql", :database => "spc_test")
-
-# Setting up sample migrations
-class CreateBooks < ActiveRecord::Migration
-  def self.up
-    create_table :books do |t|
-      t.string  :title
-      t.string  :author
-      t.integer :author_id
-      t.integer :quantity
-      t.string  :isbn
-    end
-  end
-  
-  def self.down
-    drop_table :books
-  end
-end
-
-class CreateAuthors < ActiveRecord::Migration
-  def self.up
-    create_table :authors do |t|
-      t.string :name
-      t.string :bio
-    end
-  end
-  
-  def self.down
-    drop_table :authors
-  end
-end
-
-class Book < ActiveRecord::Base; end
-class Author < ActiveRecord::Base; end
-
 class SexyPgConstraintsTest < Test::Unit::TestCase
   def setup
     CreateBooks.up
     CreateAuthors.up
   end
-  
+
   def teardown
     CreateBooks.down
     CreateAuthors.down
   end
-  
+
   def test_should_create_book
     Book.create
     assert_equal 1, Book.count
   end
-  
+
   def test_whitelist
     ActiveRecord::Migration.constrain :books, :author, :whitelist => %w(whitelisted1 whitelisted2 whitelisted3)
-    
-    assert_prohibits :author, :whitelist do |book|
+
+    assert_prohibits Book, :author, :whitelist do |book|
       book.author = 'not_whitelisted'
     end
-    
-    assert_allows do |book|
+
+    assert_allows Book do |book|
       book.author = 'whitelisted2'
     end
-    
+
     ActiveRecord::Migration.deconstrain :books, :author, :whitelist
-    
-    assert_allows do |book|
+
+    assert_allows Book do |book|
       book.author = 'not_whitelisted'
     end
   end
-  
+
+  def test_whitelist_on_a_column_whose_name_is_a_sql_keyword
+    ActiveRecord::Migration.constrain :books, :as, :whitelist => %w(whitelisted1 whitelisted2 whitelisted3)
+
+    assert_prohibits Book, :as, :whitelist do |book|
+      book.as = 'not_whitelisted'
+    end
+
+    assert_allows Book do |book|
+      book.as = 'whitelisted2'
+    end
+
+    ActiveRecord::Migration.deconstrain :books, :as, :whitelist
+
+    assert_allows Book do |book|
+      book.as = 'not_whitelisted'
+    end
+  end
+
   def test_blacklist
     ActiveRecord::Migration.constrain :books, :author, :blacklist => %w(blacklisted1 blacklisted2 blacklisted3)
-    
-    assert_prohibits :author, :blacklist do |book|
+
+    assert_prohibits Book, :author, :blacklist do |book|
       book.author = 'blacklisted2'
     end
-    
-    assert_allows do |book|
+
+    assert_allows Book do |book|
       book.author = 'not_blacklisted'
     end
-    
+
     ActiveRecord::Migration.deconstrain :books, :author, :blacklist
-    
-    assert_allows do |book|
+
+    assert_allows Book do |book|
       book.author = 'blacklisted2'
     end
   end
-  
+
+  def test_blacklist_on_a_column_whose_name_is_a_sql_keyword
+    ActiveRecord::Migration.constrain :books, :as, :blacklist => %w(blacklisted1 blacklisted2 blacklisted3)
+
+    assert_prohibits Book, :as, :blacklist do |book|
+      book.as = 'blacklisted2'
+    end
+
+    assert_allows Book do |book|
+      book.as = 'not_blacklisted'
+    end
+
+    ActiveRecord::Migration.deconstrain :books, :as, :blacklist
+
+    assert_allows Book do |book|
+      book.as = 'blacklisted2'
+    end
+  end
+
   def test_not_blank
     ActiveRecord::Migration.constrain :books, :author, :not_blank => true
-    
-    assert_prohibits :author, :not_blank do |book|
+
+    assert_prohibits Book, :author, :not_blank do |book|
       book.author = ' '
     end
-    
-    assert_allows do |book|
+
+    assert_allows Book do |book|
       book.author = 'foo'
     end
-    
+
     ActiveRecord::Migration.deconstrain :books, :author, :not_blank
-    
-    assert_allows do |book|
+
+    assert_allows Book do |book|
       book.author = ' '
     end
   end
-  
+
+  def test_not_blank_on_a_column_whose_name_is_a_sql_keyword
+    ActiveRecord::Migration.constrain :books, :as, :not_blank => true
+
+    assert_prohibits Book, :as, :not_blank do |book|
+      book.as = ' '
+    end
+
+    assert_allows Book do |book|
+      book.as = 'foo'
+    end
+
+    ActiveRecord::Migration.deconstrain :books, :as, :not_blank
+
+    assert_allows Book do |book|
+      book.as = ' '
+    end
+  end
+
   def test_within_inclusive
     ActiveRecord::Migration.constrain :books, :quantity, :within => 5..11
-    
-    assert_prohibits :quantity, :within do |book|
+
+    assert_prohibits Book, :quantity, :within do |book|
       book.quantity = 12
     end
-    
-    assert_prohibits :quantity, :within do |book|
+
+    assert_prohibits Book, :quantity, :within do |book|
       book.quantity = 4
     end
-    
-    assert_allows do |book|
+
+    assert_allows Book do |book|
       book.quantity = 7
     end
-    
+
     ActiveRecord::Migration.deconstrain :books, :quantity, :within
-    
-    assert_allows do |book|
+
+    assert_allows Book do |book|
       book.quantity = 12
     end
   end
-  
+
+  def test_within_inclusive_on_a_column_whose_name_is_a_sql_keyword
+    ActiveRecord::Migration.constrain :books, :from, :within => 5..11
+
+    assert_prohibits Book, :from, :within do |book|
+      book.from = 12
+    end
+
+    assert_prohibits Book, :from, :within do |book|
+      book.from = 4
+    end
+
+    assert_allows Book do |book|
+      book.from = 7
+    end
+
+    ActiveRecord::Migration.deconstrain :books, :from, :within
+
+    assert_allows Book do |book|
+      book.from = 12
+    end
+  end
+
   def test_within_non_inclusive
     ActiveRecord::Migration.constrain :books, :quantity, :within => 5...11
-    
-    assert_prohibits :quantity, :within do |book|
+
+    assert_prohibits Book, :quantity, :within do |book|
       book.quantity = 11
     end
-    
-    assert_prohibits :quantity, :within do |book|
+
+    assert_prohibits Book, :quantity, :within do |book|
       book.quantity = 4
     end
-    
-    assert_allows do |book|
+
+    assert_allows Book do |book|
       book.quantity = 10
     end
-    
+
     ActiveRecord::Migration.deconstrain :books, :quantity, :within
-    
-    assert_allows do |book|
+
+    assert_allows Book do |book|
       book.quantity = 11
     end
   end
-  
+
+  def test_within_non_inclusive_on_a_column_whose_name_is_a_sql_keyword
+    ActiveRecord::Migration.constrain :books, :from, :within => 5...11
+
+    assert_prohibits Book, :from, :within do |book|
+      book.from = 11
+    end
+
+    assert_prohibits Book, :from, :within do |book|
+      book.from = 4
+    end
+
+    assert_allows Book do |book|
+      book.from = 10
+    end
+
+    ActiveRecord::Migration.deconstrain :books, :from, :within
+
+    assert_allows Book do |book|
+      book.from = 11
+    end
+  end
+
   def test_length_within_inclusive
     ActiveRecord::Migration.constrain :books, :title, :length_within => 5..11
-    
-    assert_prohibits :title, :length_within do |book|
+
+    assert_prohibits Book, :title, :length_within do |book|
       book.title = 'abcdefghijkl'
     end
-    
-    assert_prohibits :title, :length_within do |book|
+
+    assert_prohibits Book, :title, :length_within do |book|
       book.title = 'abcd'
     end
-    
-    assert_allows do |book|
+
+    assert_allows Book do |book|
       book.title = 'abcdefg'
     end
-    
+
     ActiveRecord::Migration.deconstrain :books, :title, :length_within
-    
-    assert_allows do |book|
+
+    assert_allows Book do |book|
       book.title = 'abcdefghijkl'
     end
   end
-  
+
+  def test_length_within_inclusive_on_a_column_whose_name_is_a_sql_keyword
+    ActiveRecord::Migration.constrain :books, :as, :length_within => 5..11
+
+    assert_prohibits Book, :as, :length_within do |book|
+      book.as = 'abcdefghijkl'
+    end
+
+    assert_prohibits Book, :as, :length_within do |book|
+      book.as = 'abcd'
+    end
+
+    assert_allows Book do |book|
+      book.as = 'abcdefg'
+    end
+
+    ActiveRecord::Migration.deconstrain :books, :as, :length_within
+
+    assert_allows Book do |book|
+      book.as = 'abcdefghijkl'
+    end
+  end
+
   def test_length_within_non_inclusive
     ActiveRecord::Migration.constrain :books, :title, :length_within => 5...11
-    
-    assert_prohibits :title, :length_within do |book|
+
+    assert_prohibits Book, :title, :length_within do |book|
       book.title = 'abcdefghijk'
     end
-    
-    assert_prohibits :title, :length_within do |book|
+
+    assert_prohibits Book, :title, :length_within do |book|
       book.title = 'abcd'
     end
-    
-    assert_allows do |book|
+
+    assert_allows Book do |book|
       book.title = 'abcdefg'
     end
-    
+
     ActiveRecord::Migration.deconstrain :books, :title, :length_within
-    
-    assert_allows do |book|
+
+    assert_allows Book do |book|
       book.title = 'abcdefghijk'
     end
   end
-  
+
+  def test_length_within_non_inclusive_on_a_column_whose_name_is_a_sql_keyword
+    ActiveRecord::Migration.constrain :books, :as, :length_within => 5...11
+
+    assert_prohibits Book, :as, :length_within do |book|
+      book.as = 'abcdefghijk'
+    end
+
+    assert_prohibits Book, :as, :length_within do |book|
+      book.as = 'abcd'
+    end
+
+    assert_allows Book do |book|
+      book.as = 'abcdefg'
+    end
+
+    ActiveRecord::Migration.deconstrain :books, :as, :length_within
+
+    assert_allows Book do |book|
+      book.as = 'abcdefghijk'
+    end
+  end
+
   def test_email
     ActiveRecord::Migration.constrain :books, :author, :email => true
-    
-    assert_prohibits :author, :email do |book|
+
+    assert_prohibits Book, :author, :email do |book|
       book.author = 'blah@example'
     end
-    
-    assert_allows do |book|
+
+    assert_allows Book do |book|
       book.author = 'blah@example.com'
     end
-    
+
     ActiveRecord::Migration.deconstrain :books, :author, :email
-    
-    assert_allows do |book|
+
+    assert_allows Book do |book|
       book.author = 'blah@example'
     end
-  end 
-  
+  end
+
+  def test_email_on_a_column_whose_name_is_a_sql_keyword
+    ActiveRecord::Migration.constrain :books, :as, :email => true
+
+    assert_prohibits Book, :as, :email do |book|
+      book.as = 'blah@example'
+    end
+
+    assert_allows Book do |book|
+      book.as = 'blah@example.com'
+    end
+
+    ActiveRecord::Migration.deconstrain :books, :as, :email
+
+    assert_allows Book do |book|
+      book.as = 'blah@example'
+    end
+  end
+
   def test_alphanumeric
     ActiveRecord::Migration.constrain :books, :title, :alphanumeric => true
-    
-    assert_prohibits :title, :alphanumeric do |book|
+
+    assert_prohibits Book, :title, :alphanumeric do |book|
       book.title = 'asdf@asdf'
     end
-    
-    assert_allows do |book|
+
+    assert_allows Book do |book|
       book.title = 'asdf'
     end
-    
+
     ActiveRecord::Migration.deconstrain :books, :title, :alphanumeric
-    
-    assert_allows do |book|
+
+    assert_allows Book do |book|
       book.title = 'asdf@asdf'
     end
   end
-  
+
+  def test_alphanumeric_on_a_column_whose_name_is_a_sql_keyword
+    ActiveRecord::Migration.constrain :books, :as, :alphanumeric => true
+
+    assert_prohibits Book, :as, :alphanumeric do |book|
+      book.as = 'asdf@asdf'
+    end
+
+    assert_allows Book do |book|
+      book.as = 'asdf'
+    end
+
+    ActiveRecord::Migration.deconstrain :books, :as, :alphanumeric
+
+    assert_allows Book do |book|
+      book.as = 'asdf@asdf'
+    end
+  end
+
   def test_positive
     ActiveRecord::Migration.constrain :books, :quantity, :positive => true
-    
-    assert_prohibits :quantity, :positive do |book|
+
+    assert_prohibits Book, :quantity, :positive do |book|
       book.quantity = -1
     end
-    
-    assert_allows do |book|
+
+    assert_allows Book do |book|
       book.quantity = 0
     end
-    
-    assert_allows do |book|
+
+    assert_allows Book do |book|
       book.quantity = 1
     end
-    
+
     ActiveRecord::Migration.deconstrain :books, :quantity, :positive
-    
-    assert_allows do |book|
+
+    assert_allows Book do |book|
       book.quantity = -1
     end
   end
-  
+
+  def test_positive_on_a_column_whose_name_is_a_sql_keyword
+    ActiveRecord::Migration.constrain :books, :from, :positive => true
+
+    assert_prohibits Book, :from, :positive do |book|
+      book.from = -1
+    end
+
+    assert_allows Book do |book|
+      book.from = 0
+    end
+
+    assert_allows Book do |book|
+      book.from = 1
+    end
+
+    ActiveRecord::Migration.deconstrain :books, :from, :positive
+
+    assert_allows Book do |book|
+      book.from = -1
+    end
+  end
+
   def test_odd
     ActiveRecord::Migration.constrain :books, :quantity, :odd => true
-    
-    assert_prohibits :quantity, :odd do |book|
+
+    assert_prohibits Book, :quantity, :odd do |book|
       book.quantity = 2
     end
-    
-    assert_allows do |book|
+
+    assert_allows Book do |book|
       book.quantity = 1
     end
-    
+
     ActiveRecord::Migration.deconstrain :books, :quantity, :odd
-    
-    assert_allows do |book|
+
+    assert_allows Book do |book|
       book.quantity = 2
     end
   end
-  
+
+  def test_odd_on_a_column_whose_name_is_a_sql_keyword
+    ActiveRecord::Migration.constrain :books, :from, :odd => true
+
+    assert_prohibits Book, :from, :odd do |book|
+      book.from = 2
+    end
+
+    assert_allows Book do |book|
+      book.from = 1
+    end
+
+    ActiveRecord::Migration.deconstrain :books, :from, :odd
+
+    assert_allows Book do |book|
+      book.from = 2
+    end
+  end
+
   def test_even
     ActiveRecord::Migration.constrain :books, :quantity, :even => true
-    
-    assert_prohibits :quantity, :even do |book|
+
+    assert_prohibits Book, :quantity, :even do |book|
       book.quantity = 1
     end
-    
-    assert_allows do |book|
+
+    assert_allows Book do |book|
       book.quantity = 2
     end
-    
+
     ActiveRecord::Migration.deconstrain :books, :quantity, :even
-    
-    assert_allows do |book|
+
+    assert_allows Book do |book|
       book.quantity = 1
     end
   end
-  
+
+  def test_even_on_a_column_whose_name_is_a_sql_keyword
+    ActiveRecord::Migration.constrain :books, :from, :even => true
+
+    assert_prohibits Book, :from, :even do |book|
+      book.from = 1
+    end
+
+    assert_allows Book do |book|
+      book.from = 2
+    end
+
+    ActiveRecord::Migration.deconstrain :books, :from, :even
+
+    assert_allows Book do |book|
+      book.from = 1
+    end
+  end
+
   def test_unique
     ActiveRecord::Migration.constrain :books, :isbn, :unique => true
-    
-    assert_allows do |book|
+
+    assert_allows Book do |book|
       book.isbn = 'foo'
     end
-    
-    assert_prohibits :isbn, :unique, 'unique' do |book|
+
+    assert_prohibits Book, :isbn, :unique, 'unique' do |book|
       book.isbn = 'foo'
     end
-    
+
     ActiveRecord::Migration.deconstrain :books, :isbn, :unique
-    
-    assert_allows do |book|
+
+    assert_allows Book do |book|
       book.isbn = 'foo'
     end
   end
-  
+
+  def test_unique_on_a_column_whose_name_is_a_sql_keyword
+    ActiveRecord::Migration.constrain :books, :as, :unique => true
+
+    assert_allows Book do |book|
+      book.as = 'foo'
+    end
+
+    assert_prohibits Book, :as, :unique, 'unique' do |book|
+      book.as = 'foo'
+    end
+
+    ActiveRecord::Migration.deconstrain :books, :as, :unique
+
+    assert_allows Book do |book|
+      book.as = 'foo'
+    end
+  end
+
   def test_exact_length
     ActiveRecord::Migration.constrain :books, :isbn, :exact_length => 5
-    
-    assert_prohibits :isbn, :exact_length do |book|
+
+    assert_prohibits Book, :isbn, :exact_length do |book|
       book.isbn = '123456'
     end
-    
-    assert_prohibits :isbn, :exact_length do |book|
+
+    assert_prohibits Book, :isbn, :exact_length do |book|
       book.isbn = '1234'
     end
-    
-    assert_allows do |book|
+
+    assert_allows Book do |book|
       book.isbn = '12345'
     end
-    
+
     ActiveRecord::Migration.deconstrain :books, :isbn, :exact_length
-    
-    assert_allows do |book|
+
+    assert_allows Book do |book|
       book.isbn = '123456'
     end
   end
-  
+
+  def test_exact_length_on_a_column_whose_name_is_a_sql_keyword
+    ActiveRecord::Migration.constrain :books, :as, :exact_length => 5
+
+    assert_prohibits Book, :as, :exact_length do |book|
+      book.as = '123456'
+    end
+
+    assert_prohibits Book, :as, :exact_length do |book|
+      book.as = '1234'
+    end
+
+    assert_allows Book do |book|
+      book.as = '12345'
+    end
+
+    ActiveRecord::Migration.deconstrain :books, :as, :exact_length
+
+    assert_allows Book do |book|
+      book.as = '123456'
+    end
+  end
+
   def test_format_case_insensitive
     ActiveRecord::Migration.constrain :books, :title, :format => /^[a-z]+$/i
-    
-    assert_prohibits :title, :format do |book|
+
+    assert_prohibits Book, :title, :format do |book|
       book.title = 'abc3'
     end
-    
-    assert_prohibits :title, :format do |book|
+
+    assert_prohibits Book, :title, :format do |book|
       book.title = ''
     end
-    
-    assert_allows do |book|
+
+    assert_allows Book do |book|
       book.title = 'abc'
     end
-    
-    assert_allows do |book|
+
+    assert_allows Book do |book|
       book.title = 'ABc'
     end
-    
+
     ActiveRecord::Migration.deconstrain :books, :title, :format
-    
-    assert_allows do |book|
+
+    assert_allows Book do |book|
       book.title = 'abc3'
     end
   end
-  
+
+  def test_format_case_insensitive_on_a_column_whose_name_is_a_sql_keyword
+    ActiveRecord::Migration.constrain :books, :as, :format => /^[a-z]+$/i
+
+    assert_prohibits Book, :as, :format do |book|
+      book.as = 'abc3'
+    end
+
+    assert_prohibits Book, :as, :format do |book|
+      book.as = ''
+    end
+
+    assert_allows Book do |book|
+      book.as = 'abc'
+    end
+
+    assert_allows Book do |book|
+      book.as = 'ABc'
+    end
+
+    ActiveRecord::Migration.deconstrain :books, :as, :format
+
+    assert_allows Book do |book|
+      book.as = 'abc3'
+    end
+  end
+
   def test_format_case_sensitive
     ActiveRecord::Migration.constrain :books, :title, :format => /^[a-z]+$/
-    
-    assert_prohibits :title, :format do |book|
+
+    assert_prohibits Book, :title, :format do |book|
       book.title = 'aBc'
     end
-    
-    assert_allows do |book|
+
+    assert_allows Book do |book|
       book.title = 'abc'
     end
-    
+
     ActiveRecord::Migration.deconstrain :books, :title, :format
-    
-    assert_allows do |book|
+
+    assert_allows Book do |book|
       book.title = 'aBc'
     end
   end
-  
+
+  def test_format_case_sensitive_on_a_column_whose_name_is_a_sql_keyword
+    ActiveRecord::Migration.constrain :books, :as, :format => /^[a-z]+$/
+
+    assert_prohibits Book, :as, :format do |book|
+      book.as = 'aBc'
+    end
+
+    assert_allows Book do |book|
+      book.as = 'abc'
+    end
+
+    ActiveRecord::Migration.deconstrain :books, :as, :format
+
+    assert_allows Book do |book|
+      book.as = 'aBc'
+    end
+  end
+
   def test_reference
     ActiveRecord::Migration.constrain :books, :author_id, :reference => {:authors => :id}
-    
-    assert_prohibits :author_id, :reference, 'foreign key' do |book|
+
+    assert_prohibits Book, :author_id, :reference, 'foreign key' do |book|
       book.author_id = 1
     end
-    
+
     author = Author.new
     author.name = "Mark Twain"
     author.bio = "American writer"
     assert author.save
-    
+
     assert_equal 1, author.id
-    
-    assert_allows do |book|
+
+    assert_allows Book do |book|
       book.author_id = 1
     end
-    
+
     ActiveRecord::Migration.deconstrain :books, :author_id, :reference
-    
-    assert_allows do |book|
+
+    assert_allows Book do |book|
       book.author_id = 2
     end
   end
-  
+
+  def test_reference_on_a_column_whose_name_is_a_sql_keyword
+    ActiveRecord::Migration.constrain :books, :from, :reference => {:authors => :id}
+
+    assert_prohibits Book, :from, :reference, 'foreign key' do |book|
+      book.from = 1
+    end
+
+    author = Author.new
+    author.name = "Mark Twain"
+    author.bio = "American writer"
+    assert author.save
+
+    assert_equal 1, author.id
+
+    assert_allows Book do |book|
+      book.from = 1
+    end
+
+    ActiveRecord::Migration.deconstrain :books, :from, :reference
+
+    assert_allows Book do |book|
+      book.from = 2
+    end
+  end
+
   def test_reference_with_on_delete
     ActiveRecord::Migration.constrain :books, :author_id, :reference => {:authors => :id, :on_delete => :cascade}
-    
+
     author = Author.new
     author.name = "Mark Twain"
     author.bio = "American writer"
     assert author.save
-    
+
     assert_equal 1, Author.count
-    
-    assert_allows do |book|
+
+    assert_allows Book do |book|
       book.title = "The Adventures of Tom Sawyer"
       book.author_id = 1
     end
-    
-    assert_allows do |book|
+
+    assert_allows Book do |book|
       book.title = "The Adventures of Huckleberry Finn"
       book.author_id = 1
     end
-    
+
     author.destroy
-    
+
     assert_equal 0, Author.count
     assert_equal 0, Book.count
   end
-  
+
   def test_block_syntax
     ActiveRecord::Migration.constrain :books do |t|
       t.title :not_blank => true
       t.isbn :exact_length => 15
       t.author :alphanumeric => true
     end
-    
-    assert_prohibits :title, :not_blank do |book|
+
+    assert_prohibits Book, :title, :not_blank do |book|
       book.title = '  '
     end
-    
-    assert_prohibits :isbn, :exact_length do |book|
+
+    assert_prohibits Book, :isbn, :exact_length do |book|
       book.isbn = 'asdf'
     end
-    
-    assert_prohibits :author, :alphanumeric do |book|
+
+    assert_prohibits Book, :author, :alphanumeric do |book|
       book.author = 'foo#bar'
     end
-    
+
     ActiveRecord::Migration.deconstrain :books do |t|
       t.title :not_blank
       t.isbn :exact_length
       t.author :alphanumeric
     end
-    
-    assert_allows do |book|
+
+    assert_allows Book do |book|
       book.title  = '  '
       book.isbn   = 'asdf'
       book.author = 'foo#bar'
     end
   end
-  
+
   def test_multiple_constraints_per_line
     ActiveRecord::Migration.constrain :books do |t|
       t.title :not_blank => true, :alphanumeric => true, :blacklist => %w(foo bar)
     end
-    
-    assert_prohibits :title, :not_blank do |book|
+
+    assert_prohibits Book, :title, [:not_blank, :alphanumeric] do |book|
       book.title = ' '
     end
-    
-    assert_prohibits :title, :alphanumeric do |book|
+
+    assert_prohibits Book, :title, :alphanumeric do |book|
       book.title = 'asdf@asdf'
     end
-    
-    assert_prohibits :title, :blacklist do |book|
+
+    assert_prohibits Book, :title, :blacklist do |book|
       book.title = 'foo'
     end
-    
+
     ActiveRecord::Migration.deconstrain :books do |t|
       t.title :not_blank, :alphanumeric, :blacklist
     end
-    
-    assert_allows do |book|
+
+    assert_allows Book do |book|
       book.title = ' '
     end
-    
-    assert_allows do |book|
+
+    assert_allows Book do |book|
       book.title = 'asdf@asdf'
     end
-    
-    assert_allows do |book|
+
+    assert_allows Book do |book|
       book.title = 'foo'
     end
   end
-  
+
   def test_multicolumn_constraint
     ActiveRecord::Migration.constrain :books, [:title, :isbn], :unique => true
-    
-    assert_allows do |book|
+
+    assert_allows Book do |book|
       book.title = 'foo'
       book.isbn = 'bar'
     end
-    
-    assert_allows do |book|
+
+    assert_allows Book do |book|
       book.title = 'foo'
       book.isbn = 'foo'
     end
-    
-    assert_prohibits [:title, :isbn], :unique, 'unique' do |book|
+
+    assert_prohibits Book, [:title, :isbn], :unique, 'unique' do |book|
       book.title = 'foo'
       book.isbn = 'bar'
     end
-    
+
     ActiveRecord::Migration.deconstrain :books, [:title, :isbn], :unique
-    
-    assert_allows do |book|
+
+    assert_allows Book do |book|
       book.title = 'foo'
       book.isbn = 'bar'
     end
   end
-  
+
   def test_multicolumn_constraint_block_syntax
     ActiveRecord::Migration.constrain :books do |t|
       t[:title, :isbn].all :unique => true
     end
-    
-    assert_allows do |book|
+
+    assert_allows Book do |book|
       book.title = 'foo'
       book.isbn = 'bar'
     end
-    
-    assert_allows do |book|
+
+    assert_allows Book do |book|
       book.title = 'foo'
       book.isbn = 'foo'
     end
-    
-    assert_prohibits [:title, :isbn], :unique, 'unique' do |book|
+
+    assert_prohibits Book, [:title, :isbn], :unique, 'unique' do |book|
       book.title = 'foo'
       book.isbn = 'bar'
     end
-    
+
     ActiveRecord::Migration.deconstrain :books do |t|
       t[:title, :isbn].all :unique
     end
-    
-    assert_allows do |book|
+
+    assert_allows Book do |book|
       book.title = 'foo'
       book.isbn = 'bar'
     end
   end
-  
-  private
-  def assert_prohibits(column, constraint, constraint_type = 'check')
-    column = column.join('_') if column.respond_to?(:join)
-    
-    book = Book.new
-    yield(book)
-    assert book.valid?
-    error = assert_raise ActiveRecord::StatementInvalid do 
-      book.save 
-    end
-    assert_match /PGError/, error.message
-    assert_match /violates #{constraint_type} constraint "books_#{column}_#{constraint}"/, error.message
-  end
-  
-  def assert_allows
-    first_count = Book.count
-    book = Book.new
-    yield(book)
-    assert book.valid?
-    assert_nothing_raised do 
-      book.save
-    end
-    assert_equal first_count + 1, Book.count
+
+  def test_lowercase
+    ActiveRecord::Migration.constrain :books, :author, :lowercase => true
+
+    assert_prohibits Book, :author, :lowercase do |book|
+      book.author = 'UPPER'
+    end
+
+    assert_allows Book do |book|
+      book.author = 'lower with 1337'
+    end
+
+    ActiveRecord::Migration.deconstrain :books, :author, :lowercase
+
+    assert_allows Book do |book|
+      book.author = 'UPPER'
+    end
+
+  end
+
+  def test_xor
+    ActiveRecord::Migration.constrain :books, [:xor_col_1, :xor_col_2], :xor => true
+
+    assert_prohibits Book, [:xor_col_1, :xor_col_2], :xor do |book|
+      book.xor_col_1 = 123
+      book.xor_col_2 = 321
+    end
+
+    assert_allows Book do |book|
+      book.xor_col_1 = 123
+    end
+
+    assert_allows Book do |book|
+      book.xor_col_2 = 123
+    end
+
+    ActiveRecord::Migration.deconstrain :books, [:xor_col_1, :xor_col_2], :xor
+
+    assert_allows Book do |book|
+      book.xor_col_1 = 123
+      book.xor_col_2 = 123
+    end
   end
 end