require File.expand_path("../../../../base", __FILE__)

describe Vagrant::Action::Builtin::Lock do
  let(:app) { lambda { |env| } }
  let(:env) { {} }
  let(:lock_path) do
    @__lock_path = Tempfile.new("vagrant-test-lock")
    @__lock_path.path.to_s
  end

  let(:options) do
    {
      :exception => Class.new(StandardError),
      :path      => lock_path
    }
  end

  it "should require a path" do
    expect { described_class.new(app, env) }.
      to raise_error(ArgumentError)

    expect { described_class.new(app, env, :path => "foo") }.
      to raise_error(ArgumentError)

    expect { described_class.new(app, env, :exception => "foo") }.
      to raise_error(ArgumentError)

    expect { described_class.new(app, env, :path => "bar", :exception => "foo") }.
      to_not raise_error
  end

  it "should allow the path to be a proc" do
    inner_acquire = true
    app = lambda do |env|
      File.open(lock_path, "w+") do |f|
        inner_acquire = f.flock(File::LOCK_EX | File::LOCK_NB)
      end
    end

    options[:path] = lambda { |env| lock_path }

    instance = described_class.new(app, env, options)
    instance.call(env)

    inner_acquire.should == false
  end

  it "should allow the exception to be a proc" do
    exception = options[:exception]
    options[:exception] = lambda { |env| exception }

    File.open(lock_path, "w+") do |f|
      # Acquire lock
      f.flock(File::LOCK_EX | File::LOCK_NB).should == 0

      # Test!
      instance = described_class.new(app, env, options)
      expect { instance.call(env) }.
        to raise_error(exception)
    end
  end

  it "should call the middleware with the lock held" do
    inner_acquire = true
    app = lambda do |env|
      File.open(lock_path, "w+") do |f|
        inner_acquire = f.flock(File::LOCK_EX | File::LOCK_NB)
      end
    end

    instance = described_class.new(app, env, options)
    instance.call(env)

    inner_acquire.should == false
  end

  it "should raise an exception if the lock is already held" do
    File.open(lock_path, "w+") do |f|
      # Acquire lock
      f.flock(File::LOCK_EX | File::LOCK_NB).should == 0

      # Test!
      instance = described_class.new(app, env, options)
      expect { instance.call(env) }.
        to raise_error(options[:exception])
    end
  end

  it "should allow nesting locks within the same middleware sequence" do
    called = false
    app = lambda { |env| called = true }
    inner = described_class.new(app, env, options)
    outer = described_class.new(inner, env, options)
    outer.call(env)

    called.should == true
  end
end
