require 'spec_helper'
require 'dpl/provider/heroku'
require 'faraday'

RSpec.shared_context 'with faraday' do
  let(:api_key) { 'foo' }
  let(:stubs) { Faraday::Adapter::Test::Stubs.new }
  let(:faraday) {
    Faraday.new do |builder|
      builder.adapter :test, stubs do |stub|
        stub.get("/account") {|env| [200, response_headers, account_response_body]}
        stub.get("/apps/example") {|env| [200, response_headers, app_response_body]}
        stub.post("/apps/example/builds") {|env| [201, response_headers, builds_response_body]}
        stub.get("/apps/example/builds/01234567-89ab-cdef-0123-456789abcdef") {|env| [200, response_headers, build_response_body]}
        stub.post("/sources") {|env| [201, response_headers, source_response_body] }
        stub.post("/apps/example/dynos") {|env| [201, response_headers, dynos_create_response_body]}
        stub.delete("/apps/example/dynos") {|env| [202, response_headers, '{}'] }
      end
    end
  }

  let(:response_headers) {
    {'Content-Type' => 'application/json'}
  }

  let(:app_response_body) {
    '{
      "acm": false,
      "archived_at": "2012-01-01T12:00:00Z",
      "buildpack_provided_description": "Ruby/Rack",
      "build_stack": {
        "id": "01234567-89ab-cdef-0123-456789abcdef",
        "name": "cedar-14"
      },
      "created_at": "2012-01-01T12:00:00Z",
      "git_url": "https://git.heroku.com/example.git",
      "id": "01234567-89ab-cdef-0123-456789abcdef",
      "maintenance": false,
      "name": "example",
      "owner": {
        "email": "username@example.com",
        "id": "01234567-89ab-cdef-0123-456789abcdef"
      },
      "organization": {
        "id": "01234567-89ab-cdef-0123-456789abcdef",
        "name": "example"
      },
      "team": {
        "id": "01234567-89ab-cdef-0123-456789abcdef",
        "name": "example"
      },
      "region": {
        "id": "01234567-89ab-cdef-0123-456789abcdef",
        "name": "us"
      },
      "released_at": "2012-01-01T12:00:00Z",
      "repo_size": 0,
      "slug_size": 0,
      "space": {
        "id": "01234567-89ab-cdef-0123-456789abcdef",
        "name": "nasa",
        "shield": true
      },
      "stack": {
        "id": "01234567-89ab-cdef-0123-456789abcdef",
        "name": "cedar-14"
      },
      "updated_at": "2012-01-01T12:00:00Z",
      "web_url": "https://example.herokuapp.com/"
    }'
  }

  let(:account_response_body) {
    '{
      "allow_tracking": true,
      "beta": false,
      "created_at": "2012-01-01T12:00:00Z",
      "email": "username@example.com",
      "federated": false,
      "id": "01234567-89ab-cdef-0123-456789abcdef",
      "identity_provider": {
        "id": "01234567-89ab-cdef-0123-456789abcdef",
        "organization": {
          "name": "example"
        }
      },
      "last_login": "2012-01-01T12:00:00Z",
      "name": "Tina Edmonds",
      "sms_number": "+1 ***-***-1234",
      "suspended_at": "2012-01-01T12:00:00Z",
      "delinquent_at": "2012-01-01T12:00:00Z",
      "two_factor_authentication": false,
      "updated_at": "2012-01-01T12:00:00Z",
      "verified": false,
      "default_organization": {
        "id": "01234567-89ab-cdef-0123-456789abcdef",
        "name": "example"
      }
    }'
  }

  let(:builds_response_body) {
    '{
      "app": {
        "id": "01234567-89ab-cdef-0123-456789abcdef"
      },
      "buildpacks": [
        {
          "url": "https://github.com/heroku/heroku-buildpack-ruby"
        }
      ],
      "created_at": "2012-01-01T12:00:00Z",
      "id": "01234567-89ab-cdef-0123-456789abcdef",
      "output_stream_url": "https://build-output.heroku.com/streams/01234567-89ab-cdef-0123-456789abcdef",
      "source_blob": {
        "checksum": "SHA256:e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855",
        "url": "https://example.com/source.tgz?token=xyz",
        "version": "v1.3.0"
      },
      "release": {
        "id": "01234567-89ab-cdef-0123-456789abcdef"
      },
      "slug": {
        "id": "01234567-89ab-cdef-0123-456789abcdef"
      },
      "status": "succeeded",
      "updated_at": "2012-01-01T12:00:00Z",
      "user": {
        "id": "01234567-89ab-cdef-0123-456789abcdef",
        "email": "username@example.com"
      }
    }'
  }

  let(:source_response_body) {
    '{
      "source_blob": {
        "get_url": "https://api.heroku.com/sources/1234.tgz",
        "put_url": "https://api.heroku.com/sources/1234.tgz"
      }
    }'
  }

  let(:dynos_create_response_body) {
    '{
      "attach_url": "rendezvous://rendezvous.runtime.heroku.com:5000/rendezvous",
      "command": "bash",
      "created_at": "2012-01-01T12:00:00Z",
      "id": "01234567-89ab-cdef-0123-456789abcdef",
      "name": "run.1",
      "release": {
        "id": "01234567-89ab-cdef-0123-456789abcdef",
        "version": 11
      },
      "app": {
        "name": "example",
        "id": "01234567-89ab-cdef-0123-456789abcdef"
      },
      "size": "standard-1X",
      "state": "up",
      "type": "run",
      "updated_at": "2012-01-01T12:00:00Z"
    }'
  }

  let(:build_response_body) {
  '{
    "id": "01234567-89ab-cdef-0123-456789abcdef",
    "status": "succeeded",
    "output_stream_url": "https://build-output.heroku.com/streams/01234567-89ab-cdef-0123-456789abcdef"
  }'
  }

  let(:build_response_body_failure) {
    '{
      "id": "01234567-89ab-cdef-0123-456789abcdef",
      "status": "failed",
      "output_stream_url": "https://build-output.heroku.com/streams/01234567-89ab-cdef-0123-456789abcdef"
    }'
  }

  let(:build_response_body_in_progress) {
    '{
      "id": "01234567-89ab-cdef-0123-456789abcdef",
      "status": "pending",
      "output_stream_url": "https://build-output.heroku.com/streams/01234567-89ab-cdef-0123-456789abcdef"
    }'
  }
end

describe DPL::Provider::Heroku, :api do
  include_context 'with faraday'

  subject(:provider) do
    described_class.new(DummyContext.new, provider_options.merge({ :api_key => api_key}))
  end

  let(:provider_options) {
    {:app => 'example', :key_name => 'key', :strategy => "api"}
  }

  describe "#ssh" do
    it "doesn't require an ssh key" do
      expect(provider.needs_key?).to eq(false)
    end
  end

  describe "#api" do
    it 'accepts an api key' do
      expect(provider).to receive(:faraday).at_least(:once).and_return(faraday)
      provider.check_auth
    end

    context "when api_key is not given" do
      let(:provider) { described_class.new(DummyContext.new, provider_options) }
      it 'raises DPL::Error' do
        provider.options.update(:user => "foo", :password => "bar")
        expect { provider.check_auth }.to raise_error(DPL::Error)
      end
    end
  end

  describe "#trigger_build" do
    it "does not initiate legacy API object" do
      expect(provider).to receive(:faraday).at_least(:once).and_return(faraday)
      provider.trigger_build
    end

    example do
      expect(provider).to receive(:log).with('triggering new deployment')
      expect(provider).to receive(:faraday).at_least(:once).and_return(faraday)
      expect(provider).to receive(:get_url).and_return 'http://example.com/source.tgz'
      expect(provider).to receive(:version).and_return 'v1.3.0'
      expect(provider.context).to receive(:shell).with("curl https://build-output.heroku.com/streams/01234567-89ab-cdef-0123-456789abcdef -H 'Accept: application/vnd.heroku+json; version=3' -H 'User-Agent: dpl/#{DPL::VERSION}'")
      provider.trigger_build
      expect(provider.build_id).to eq('01234567-89ab-cdef-0123-456789abcdef')
    end

    context 'when $stdout is not a TTY' do
      before do
        @old_stdout = $stdout
        $stdout = StringIO.new
      end

      after do
        $stdout = @old_stdout
      end

      example do
        expect(provider).to receive(:log).with('triggering new deployment')
        expect(provider).to receive(:faraday).at_least(:once).and_return(faraday)
        expect(provider).to receive(:get_url).and_return 'http://example.com/source.tgz'
        expect(provider).to receive(:version).and_return 'v1.3.0'
        expect(provider.context).to receive(:shell).with("curl -sS https://build-output.heroku.com/streams/01234567-89ab-cdef-0123-456789abcdef -H 'Accept: application/vnd.heroku+json; version=3' -H 'User-Agent: dpl/#{DPL::VERSION}'")
        provider.trigger_build
        expect(provider.build_id).to eq('01234567-89ab-cdef-0123-456789abcdef')
      end

    end
  end

  describe "#verify_build" do
    context 'when build succeeds' do
      example do
        expect(provider).to receive(:faraday).at_least(:once).and_return(faraday)
        expect(provider).to receive(:build_id).at_least(:once).and_return('01234567-89ab-cdef-0123-456789abcdef')
        expect{ provider.verify_build }.not_to raise_error
      end
    end

    context 'when build fails' do
      example do
        expect(provider).to receive(:faraday).at_least(:once).and_return(faraday)
        expect(provider).to receive(:build_id).at_least(:once).and_return('01234567-89ab-cdef-0123-456789abcdef')
        stubs.get("/apps/example/builds/01234567-89ab-cdef-0123-456789abcdef") {|env| [200, response_headers, build_response_body_failure]}
        expect{ provider.verify_build }.to raise_error("deploy failed")
      end
    end

    context 'when build is pending, then succeeds' do
      example do
        expect(provider).to receive(:faraday).at_least(:once).and_return(faraday)
        expect(provider).to receive(:build_id).at_least(:once).and_return('01234567-89ab-cdef-0123-456789abcdef')
        stubs.get("/apps/example/builds/01234567-89ab-cdef-0123-456789abcdef") {|env| [200, response_headers, build_response_body_in_progress]}
        expect(provider).to receive(:sleep).with(5).and_return(true)
        expect{ provider.verify_build }.not_to raise_error
      end
    end

  end

end


describe DPL::Provider::Heroku, :git do
  include_context "with faraday"

  subject :provider do
    described_class.new(DummyContext.new, :app => 'example', :key_name => 'key', :api_key => api_key, :strategy => "git")
  end

  describe "#api" do
    it 'accepts an api key' do
      expect(provider).to receive(:faraday).at_least(:once).and_return(faraday)
      provider.check_auth
    end
  end

  context "with faraday" do
    before :each do
      expect(provider).to receive(:faraday).at_least(:once).and_return(faraday)
    end

    describe "#check_auth" do
      example do
        expect(provider).to receive(:log).with("authentication succeeded")
        provider.check_auth
      end
    end

    describe "#check_app" do
      example do
        expect(provider).to receive(:log).at_least(1).times.with(/example/)
        provider.check_app
      end
    end

    describe "#run" do
      example do
        expect(Rendezvous).to receive(:start).with(:url => "rendezvous://rendezvous.runtime.heroku.com:5000/rendezvous")
        provider.run("that command")
      end
    end

    describe "#restart" do
      example do
        provider.restart
      end
    end
  end

  context "without faraday" do
    describe "#push_app" do
      example do
        provider.options[:git] = "git://something"
        expect(provider.context).to receive(:shell).with("git fetch origin $TRAVIS_BRANCH --unshallow")
        expect(provider.context).to receive(:shell).with("git push git://something HEAD:refs/heads/master -f")
        provider.push_app
        expect(provider.context.env['GIT_HTTP_USER_AGENT']).to include("dpl/#{DPL::VERSION}")
      end
    end
  end
end