$LOAD_PATH.unshift(File.dirname(__FILE__))
$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))

require 'rubygems'
require 'bundler/setup'
require 'active_record'

require 'locker'

ActiveRecord::Base.time_zone_aware_attributes = true
ActiveRecord::Base.default_timezone = :utc

if File.exist?(File.join(File.dirname(__FILE__), 'database.yml'))
  config = YAML.load_file(File.join(File.dirname(__FILE__), 'database.yml'))
else
  puts "database.yml did not exist, using defaults"
  config = { "reconnect" => false,
             "database"  => "locker_test",
             "adapter"   => "postgresql",
             "password"  => nil,
             "user"      => "postgres",
             "encoding"  => "utf8",
             "pool"      => 5 }
end

begin
  ActiveRecord::Base.establish_connection(config.merge('database' => 'postgres', 'schema_search_path' => 'public'))
  ActiveRecord::Base.connection.create_database(config['database'], config.merge("encoding" => config['encoding'] || ENV['CHARSET'] || 'utf8'))
rescue ActiveRecord::StatementInvalid => e
  raise unless e.message =~ /database "locker_test" already exists/
end

ActiveRecord::Base.establish_connection(config)

ActiveRecord::Base.connection.execute("DROP TABLE IF EXISTS locks")
ActiveRecord::Base.connection.create_table(:locks) do |t|
  t.string :locked_by
  t.string :key
  t.integer :sequence, :default => 0
  t.datetime :locked_at
  t.datetime :locked_until
end
ActiveRecord::Base.connection.add_index :locks, :key, :unique => true

class Lock < ActiveRecord::Base
end

class FakeLock
  attr_accessor :locked_by, :key, :locked_at, :locked_until

  cattr_accessor :fake_locks
  self.fake_locks = {}

  def self.find_by_key(key)
    fake_locks[key]
  end

  def self.find_by_key_and_locked_by(key, locked_by)
    lock = fake_locks[key]

    if lock && lock.locked_by == locked_by
      lock
    end
  end

  def self.create(attributes={})
    fake_locks[attributes[:key]] = new(attributes)
    true
  end

  def initialize(attributes={})
    attributes.each do |key, value|
      send("#{key}=", value)
    end
  end
end

RSpec.configure do |c|
  c.before do
    if ActiveRecord::VERSION::MAJOR < 4
      ActiveRecord::Base.connection.increment_open_transactions
      ActiveRecord::Base.connection.begin_db_transaction
    else
      ActiveRecord::Base.connection.begin_transaction
    end
  end
  c.after do
    if ActiveRecord::VERSION::MAJOR < 4
      ActiveRecord::Base.connection.rollback_db_transaction
      ActiveRecord::Base.connection.decrement_open_transactions
    else
      ActiveRecord::Base.connection.rollback_transaction
    end
  end
end