require 'spec_helper'

describe Heirloom do

    before do
      @config_mock = double 'config'
      @logger_mock = double 'logger'
      @config_mock.should_receive(:logger).and_return(@logger_mock)
      @downloader = Heirloom::ArtifactDownloader.new :config => @config_mock,
                                                     :name   => 'tim',
                                                     :id     => '123'
    end

    it "should download an artifact" do
      s3_downloader_mock = mock 's3 downloader'
      Heirloom::Downloader::S3.should_receive(:new).
                               with(:config => @config_mock,
                                    :logger => @logger_mock,
                                    :region => 'us-west-1').
                               and_return s3_downloader_mock
      artifact_reader_mock = mock 'artifact_reader'
      @downloader.should_receive(:artifact_reader).
                  exactly(2).times.
                  and_return artifact_reader_mock
      artifact_reader_mock.should_receive(:get_bucket).
                           with(:region => 'us-west-1').
                           and_return 'bucket-us-west-1'
      artifact_reader_mock.should_receive(:get_key).
                           with(:region => 'us-west-1').
                           and_return 'key'

      @logger_mock.should_receive(:info).
                   with "Downloading s3://bucket-us-west-1/key from us-west-1."

      s3_downloader_mock.should_receive(:download_file).
                         with(:bucket => 'bucket-us-west-1',
                              :key    => 'key').
                         and_return 'filename'

      @logger_mock.should_receive(:info).
                   with "Writing file to /tmp/file."

      file_mock = mock 'file'

      File.should_receive(:open).with('/tmp/file', 'w').
                                 and_return file_mock

      @logger_mock.should_receive(:info).with "Download complete."

      @downloader.download(:output => '/tmp/file',
                           :region => 'us-west-1')
    end

    it "should download the artifact to the current path if output is unspecficief" do
      s3_downloader_mock = mock 's3 downloader'
      Heirloom::Downloader::S3.should_receive(:new).
                               with(:config => @config_mock,
                                    :logger => @logger_mock,
                                    :region => 'us-west-1').
                               and_return s3_downloader_mock
      artifact_reader_mock = mock 'artifact_reader'
      @downloader.should_receive(:artifact_reader).
                  exactly(2).times.
                  and_return artifact_reader_mock
      artifact_reader_mock.should_receive(:get_bucket).
                           with(:region => 'us-west-1').
                           and_return 'bucket-us-west-1'
      artifact_reader_mock.should_receive(:get_key).
                           with(:region => 'us-west-1').
                           and_return 'key'

      @logger_mock.should_receive(:info).
                   with "Downloading s3://bucket-us-west-1/key from us-west-1."

      s3_downloader_mock.should_receive(:download_file).
                         with(:bucket => 'bucket-us-west-1',
                              :key    => 'key').
                         and_return 'filename'

      @logger_mock.should_receive(:info).
                   with "Writing file to ./key."

      file_mock = mock 'file'

      File.should_receive(:open).with('./key', 'w').
                                 and_return file_mock

      @logger_mock.should_receive(:info).with "Download complete."

      @downloader.download(:region => 'us-west-1')
    end

end