require File.join(File.dirname(__FILE__),'lib/boot') unless defined?(ActiveRecord)
require 'test/unit'
require 'shoulda'
require 'mocha'
require 'factory_girl'
require 'lib/test_case'
require 'grouped_scope'

class GroupedScope::TestCase
  
  def setup_environment(options={})
    options.reverse_merge! :group_column => :group_id
    setup_database(options)
    Department.create! :name => 'IT'
    Department.create! :name => 'Human Resources'
    Department.create! :name => 'Finance'
  end
  
  protected
  
  def setup_database(options)
    ActiveRecord::Base.class_eval do
      silence do
        connection.create_table :employees, :force => true do |t|
          t.column :name,         :string
          t.column :email,        :string
          t.column options[:group_column],  :integer
        end
        connection.create_table :reports, :force => true do |t|
          t.column :title,        :string
          t.column :body,         :string
          t.column :employee_id,  :integer
        end
        connection.create_table :departments, :force => true do |t|
          t.column :name,         :string
        end
        connection.create_table :department_memberships, :force => true do |t|
          t.column :employee_id,    :integer
          t.column :department_id,  :integer
          t.column :meta_info,      :string
        end
        connection.create_table :legacy_employees, :force => true, :id => false do |t|
          t.column :name,         :string
          t.column :email,        :string
          t.column options[:group_column],  :integer
        end
        connection.create_table :legacy_reports, :force => true do |t|
          t.column :title,        :string
          t.column :body,         :string
          t.column :email,        :string
        end
        connection.create_table :foo_bars, :force => true do |t|
          t.column :foo,          :string
          t.column :bar,          :string
        end
      end
    end
  end
  
end

class Employee < ActiveRecord::Base
  has_many :reports do ; def urgent ; find(:all,:conditions => {:title => 'URGENT'}) ; end ; end
  has_many :taxonomies, :as => :classable
  has_many :department_memberships
  has_many :departments, :through => :department_memberships
  grouped_scope :reports, :departments
end

class Report < ActiveRecord::Base
  named_scope :with_urgent_title, :conditions => {:title => 'URGENT'}
  named_scope :with_urgent_body, :conditions => "body LIKE '%URGENT%'"
  belongs_to :employee
  def urgent_title? ; self[:title] == 'URGENT' ; end
  def urgent_body? ; self[:body] =~ /URGENT/ ; end
end

class Department < ActiveRecord::Base
  named_scope :it, :conditions => {:name => 'IT'}
  named_scope :hr, :conditions => {:name => 'Human Resources'}
  named_scope :finance, :conditions => {:name => 'Finance'}
  has_many :department_memberships
  has_many :employees, :through => :department_memberships
end

class DepartmentMembership < ActiveRecord::Base
  belongs_to :employee
  belongs_to :department
end

class LegacyEmployee < ActiveRecord::Base
  set_primary_key :email
  has_many :reports, :class_name => 'LegacyReport', :foreign_key => 'email'
  grouped_scope :reports
  alias_method :email=, :id=
end

class LegacyReport < ActiveRecord::Base
  belongs_to :employee, :class_name => 'LegacyEmployee', :foreign_key => 'email'
end

class FooBar < ActiveRecord::Base
  has_many :reports
  grouped_scope :reports
end