require 'rspec'

require 'active_model'
require 'active_model/validations'
require 'active_record'
require 'action_view'
require 'timecop'
require 'rspec_tag_matchers'

require 'validates_timeliness'

require 'support/test_model'
require 'support/model_helpers'
require 'support/config_helper'

ValidatesTimeliness.setup do |c|
  c.extend_orms = [ :active_record ]
  c.enable_date_time_select_extension!
  c.enable_multiparameter_extension!
  c.default_timezone = :utc
end

Time.zone = 'Australia/Melbourne'

LOCALE_PATH = File.expand_path(File.dirname(__FILE__) + '/../lib/generators/validates_timeliness/templates/en.yml')
I18n.load_path.unshift(LOCALE_PATH)

# Extend TestModel as you would another ORM/ODM module
module TestModelShim
  extend ActiveSupport::Concern
  include ValidatesTimeliness::AttributeMethods

  module ClassMethods
    # Hook method for attribute method generation
    def define_attribute_methods(attr_names)
      super
      define_timeliness_methods
    end

    # Hook into native time zone handling check, if any
    def timeliness_attribute_timezone_aware?(attr_name)
      false
    end
  end
end

class Person
  include TestModel
  attribute :birth_date, :date
  attribute :birth_time, :time
  attribute :birth_datetime, :datetime
  validates_date :birth_date
  validates_time :birth_time
  validates_datetime :birth_datetime
  define_attribute_methods model_attributes.keys
end

class PersonWithShim < Person
  include TestModelShim
end

ActiveRecord::Base.time_zone_aware_attributes = true
ActiveRecord::Base.establish_connection({:adapter => 'sqlite3', :database => ':memory:'})
ActiveRecord::Migration.verbose = false
ActiveRecord::Schema.define(:version => 1) do
  create_table :employees, :force => true do |t|
    t.string   :first_name
    t.string   :last_name
    t.date     :birth_date
    t.time     :birth_time
    t.datetime :birth_datetime
  end
end

class Employee < ActiveRecord::Base
  validates_date :birth_date
  validates_time :birth_time
  validates_datetime :birth_datetime
  define_attribute_methods

  attr_accessor :redefined_birth_date_called

  def birth_date=(value)
    self.redefined_birth_date_called = true
    super
  end
end

Rspec.configure do |c|
  c.mock_with :rspec
  c.include(RspecTagMatchers)
  c.include(ModelHelpers)
  c.include(ConfigHelper)
  c.before do
    Person.reset_callbacks(:validate)
    PersonWithShim.timeliness_validated_attributes = []
    Person._validators.clear
    Employee.reset_callbacks(:validate)
    Employee.timeliness_validated_attributes = []
    Employee._validators.clear
  end
end