require 'spec_helper'

shared_examples "a proxy" do |args|

  context "basic features" do
    before(:each) do
      @proxy = Defog::Proxy.new(args)
    end

    it "should have a nice to_s" do
      @proxy.to_s.should include @proxy.provider.to_s
      @proxy.to_s.should include @proxy.location
    end

    it "file should return a handle" do
      handle = @proxy.file(key)
      handle.proxy.should == @proxy
      handle.key.should == key
    end

    it "file should yield a handle" do
      ret = @proxy.file(key) do |handle|
        handle.proxy.should == @proxy
        handle.key.should == key
        123
      end
      ret.should == 123
    end

    it "should forward file open to handle" do
      Defog::Handle.should_receive(:new).with(@proxy, key).and_return { mock('Handle').tap { |handle|
        handle.should_receive(:open).with("r+", :persist => true)
      } }
      @proxy.file(key, "r+", :persist => true)
    end

    it "should return fog storage" do
      @proxy.fog_connection.should == @proxy.fog_directory.service
    end

    it "should return fog directory" do
      create_remote("hello")
      @proxy.fog_directory.files.get(key).body.should == "hello"
    end
  end

  context "settings" do
    it "should set default for :persist => true" do
      @proxy = Defog::Proxy.new(args.merge(:persist => true))
      Defog::File.should_receive(:open).with(hash_including :persist => true)
      @proxy.file(key, "w") do end
    end
    it "should set default for :synchronize => :async" do
      @proxy = Defog::Proxy.new(args.merge(:synchronize => :async))
      Defog::File.should_receive(:open).with(hash_including :synchronize => :async)
      @proxy.file(key, "w") do end
    end
  end


  context "iteration" do

    before(:each) do
      @proxy = Defog::Proxy.new(args)
      @proxy.fog_directory.files.all.each do |model| model.destroy end
      create_other_remote("i0")
      create_other_remote("i1")
      create_other_remote("i2")
    end

    it "should iterate through remotes" do
      seen = []
      @proxy.each do |handle|
        seen << handle.key
      end
      seen.should =~ [other_key("i0"), other_key("i1"), other_key("i2")]
    end

    it "should return an enumerator" do
      @proxy.each.map(&:key).should =~ [other_key("i0"), other_key("i1"), other_key("i2")]
    end

  end

  context "prefix" do
    it "should return its prefix" do
      prefix = "me-first"
      @proxy = Defog::Proxy.new(args.merge(:prefix => prefix))
      @proxy.prefix.should == prefix
    end

    it "should use a prefix" do
      prefix = "me-first"
      @proxy = Defog::Proxy.new(args.merge(:prefix => prefix))
      @proxy.file(key, "w") { |f| f.puts "hello" }
      @proxy.file(key).fog_model.key.should == prefix + key
    end

    it "should iterate only matches to prefix" do
      @proxy = Defog::Proxy.new(args.merge(:prefix => "yes-"))
      @proxy.fog_directory.files.all.each do |model| model.destroy end
      create_other_remote("no-n1")
      create_other_remote("no-n2")
      create_other_remote("no-n3")
      create_other_remote("yes-y1")
      create_other_remote("yes-y2")
      create_other_remote("yes-y3")
      @proxy.each.map(&:key).should =~[other_key("y1"), other_key("y2"), other_key("y3")]
    end

  end

  context "proxy root location" do
    it "should default proxy root to tmpdir/defog" do
      proxy = Defog::Proxy.new(args)
      proxy.proxy_root.should == Pathname.new(Dir.tmpdir) + "defog" + "#{proxy.provider.to_s}-#{proxy.location}"
    end

    it "should default proxy root to Rails.root" do
      with_rails_defined do
        proxy = Defog::Proxy.new(args)
        proxy.proxy_root.should == Rails.root + "tmp/defog" + "#{proxy.provider.to_s}-#{proxy.location}"
      end
    end

    it "should accept proxy root parameter" do
      path = Pathname.new("/a/random/path")
      proxy = Defog::Proxy.new(args.merge(:proxy_root => path))
      proxy.proxy_root.should == path
    end
  end

  context "cache management" do
    before(:each) do
      @proxy = Defog::Proxy.new(args.merge(:max_cache_size => 100, :persist => true))
      @proxy.proxy_root.rmtree if @proxy.proxy_root.exist?
      @proxy.proxy_root.mkpath
    end

    it "should fail normally when trying to proxy a file that doesn't exist" do
      expect { @proxy.file("nonesuch", "r") }.to raise_error(Defog::Error::NoCloudFile)
    end

    it "should raise an error trying to proxy a file larger than the cache" do
      create_remote("x" * 101)
      expect { @proxy.file(key, "r") }.to raise_error(Defog::Error::CacheFull)
      proxy_path.should_not be_exist
    end

    it "should not count existing proxy in total" do
      create_proxy("y" * 70)
      create_remote("x" * 70)
      expect { @proxy.file(key, "r") do end }.to_not raise_error(Defog::Error::CacheFull)
      proxy_path.should be_exist
      proxy_path.read.should == remote_body
    end

    it "should delete proxies to make room" do
      create_other_proxy("a", 10)
      create_other_proxy("b", 30)
      create_other_proxy("c", 40)
      create_remote("x" * 80)
      expect { @proxy.file(key, "r") do end }.to_not raise_error(Defog::Error::CacheFull)
      proxy_path.should be_exist
      other_proxy_path("a").should be_exist
      other_proxy_path("b").should_not be_exist
      other_proxy_path("c").should_not be_exist
    end

    it "should not fail when proxies get deleted by another process" do
      create_other_proxy("a", 10)
      create_other_proxy("b", 30)
      create_other_proxy("c", 40)
      create_remote("x" * 80)
      Pathname.any_instance.should_receive(:unlink).and_raise Errno::ENOENT
      expect { @proxy.file(key, "r") do end }.to_not raise_error(Errno::ENOENT)
    end

    it "should delete proxies to make room for hinted size" do
      create_other_proxy("a", 10)
      create_other_proxy("b", 30)
      create_other_proxy("c", 40)
      expect { @proxy.file(key, "w", :size_hint => 80) do end }.to_not raise_error(Defog::Error::CacheFull)
      proxy_path.should be_exist
      other_proxy_path("a").should be_exist
      other_proxy_path("b").should_not be_exist
      other_proxy_path("c").should_not be_exist
    end

    it "should not delete proxies that are open" do
      create_other_proxy("a", 20)
      create_other_proxy("b", 20)
      create_other_remote("R", 30)
      create_remote("x" * 30)
      @proxy.file(other_key("R"), "r") do
        @proxy.file(other_key("S"), "w") do
          create_other_proxy("S", 30)
          expect { @proxy.file(key, "r") do end }.to_not raise_error(Defog::Error::CacheFull)
          proxy_path.should be_exist
          other_proxy_path("R").should be_exist
          other_proxy_path("S").should be_exist
          other_proxy_path("a").should_not be_exist
          other_proxy_path("b").should_not be_exist
        end
      end
    end

    it "should delete proxies that are no longer open" do
      create_other_remote("R", 60)
      create_remote("z" * 60)
      @proxy.file(other_key("R"), "r") do end
      other_proxy_path("R").should be_exist
      expect { @proxy.file(key, "r") do end }.to_not raise_error(Defog::Error::CacheFull)
      proxy_path.should be_exist
      other_proxy_path("R").should_not be_exist
    end

    it "should not delete proxies if there wouldn't be enough space" do
      create_other_proxy("a", 20)
      create_other_proxy("b", 20)
      create_other_remote("R", 30)
      create_other_remote("S", 30)
      create_remote("z" * 50)
      @proxy.file(other_key("R"), "r") do
        @proxy.file(other_key("S"), "r") do
          expect { @proxy.file(key, "r") do end }.to raise_error(Defog::Error::CacheFull)
          proxy_path.should_not be_exist
          other_proxy_path("a").should be_exist
          other_proxy_path("b").should be_exist
          other_proxy_path("R").should be_exist
          other_proxy_path("S").should be_exist
        end
      end
    end

  end

  private

  def other_key(okey)
    "#{okey}-#{key}"
  end

  def create_other_proxy(okey, size)
    path = other_proxy_path(okey)
    path.dirname.mkpath
    path.open("w") do |f|
      f.write("x" * size)
    end
  end

  def other_proxy_path(okey)
    @proxy.file(other_key(okey)).proxy_path
  end

  def create_other_remote(okey, size=10)
    @proxy.fog_directory.files.create(:key => other_key(okey), :body => "x" * size)
  end

end

describe Defog::Proxy do

  context "Local" do
    before(:all) do
      Fog.unmock!
    end

    args = {:provider => :local, :local_root => LOCAL_CLOUD_PATH}

    it_should_behave_like "a proxy", args

    it "should use the deslashed local_root as the location" do
      Defog::Proxy.new(args).location.should == LOCAL_CLOUD_PATH.to_s.gsub(%r{/},"-")
    end

  end

  context "AWS" do
    before(:all) do
      Fog.mock!
    end

    args = {:provider => :AWS, :aws_access_key_id => "dummyid", :aws_secret_access_key => "dummysecret", :region => "eu-west-1", :bucket => "tester"}
    it_should_behave_like "a proxy", args

    it "should use the bucket name as the location" do
      Defog::Proxy.new(args).location.should == args[:bucket]
    end

    it "should share fog connection with same bucket" do
      proxy1 = Defog::Proxy.new(args)
      proxy2 = Defog::Proxy.new(args)
      proxy1.fog_connection.should be_equal proxy2.fog_connection
    end

    it "should share fog connection with different bucket" do
      proxy1 = Defog::Proxy.new(args)
      proxy2 = Defog::Proxy.new(args.merge(:bucket => "other"))
      proxy1.fog_connection.should be_equal proxy2.fog_connection
    end

    it "should not share fog connection with different connection args" do
      proxy1 = Defog::Proxy.new(args)
      proxy2 = Defog::Proxy.new(args.merge(:aws_access_key_id => "other"))
      proxy1.fog_connection.should_not be_equal proxy2.fog_connection
    end

  end

  it "should raise error on bad provider" do
    expect { Defog::Proxy.new(:provider => :nonesuch) }.to raise_error(ArgumentError)
  end

end