require 'spec_helper'
require 'generator_spec/test_case'
require './lib/generators/appsignal/appsignal_generator'

# The generator doesn't know we're testing
# So change the path while running the generator
# Change it back upon completion
def run_generator_in_tmp(args=[])
  FileUtils.cd("spec/tmp") do
    @output = run_generator(args)
  end
end

describe AppsignalGenerator do
  include GeneratorSpec::TestCase
  destination File.expand_path("../../../tmp", __FILE__)

  context "with key" do
    context "known key" do
      before do
        prepare_destination
        authcheck = double
        Appsignal::AuthCheck.should_receive(:new).and_return(authcheck)
        authcheck.should_receive(:perform_with_result).
          and_return(['200', 'everything ok'])
        run_generator_in_tmp %w(production my_app_key)
      end

      specify "should mention successful auth check" do
        @output.should include('success  everything ok')
      end
    end

    context "invalid key" do
      before do
        prepare_destination
        authcheck = double
        Appsignal::AuthCheck.should_receive(:new).and_return(authcheck)
        authcheck.should_receive(:perform_with_result).
          and_return(['401', 'unauthorized'])
        run_generator_in_tmp %w(production my_app_key)
      end

      specify "should mention invalid key" do
        @output.should include('error  unauthorized')
      end
    end

    context "failed check" do
      before do
        prepare_destination
        authcheck = double
        Appsignal::AuthCheck.should_receive(:new).and_return(authcheck)
        authcheck.should_receive(:perform_with_result).
          and_return(['500', 'error!'])
        run_generator_in_tmp %w(production my_app_key)
      end

      specify "should mention failed check" do
        @output.should include('error  error!')
      end
    end

    context "internal failure" do
      before do
        prepare_destination
        authcheck = Appsignal::AuthCheck.new
        Appsignal::AuthCheck.should_receive(:new).and_return(authcheck)
        authcheck.should_receive(:perform).
          and_throw(:error)
        run_generator_in_tmp %w(production my_app_key)
      end

      specify "should mention internal failure" do
        @output.should include('Something went wrong while trying to '\
          'authenticate with AppSignal:')
      end
    end
  end

  context "without key" do
    before do
      prepare_destination
      run_generator_in_tmp %w()
    end

    specify "no config files are created" do
      destination_root.should have_structure {
        directory 'config' do
          no_file 'appsignal.yml'
          no_file 'deploy.rb'
        end
      }
    end
  end

  context "without capistrano" do
    before :all do
      prepare_destination
      run_generator_in_tmp %w(production my_app_key)
    end

    specify "config file is created" do
      destination_root.should have_structure {
        directory 'config' do
          file 'appsignal.yml'
          no_file 'deploy.rb'
        end
      }
    end

    specify "should mention the deploy task" do
      @output.should include('No capistrano setup detected!')
      @output.should include('appsignal notify_of_deploy -h')
    end
  end

  context "with capistrano" do
    before :all do
      prepare_destination
      cap_file = File.expand_path('Capfile', destination_root)
      File.open(cap_file, 'w') {}
      FileUtils.mkdir(File.expand_path('config', destination_root))
      deploy_file = File.expand_path(File.join('config', 'deploy.rb'),
        destination_root)
      File.open(deploy_file, 'w') {}
      run_generator_in_tmp %w(production my_app_key)
    end

    specify "config file is created and capistrano deploy file modified" do
      destination_root.should have_structure {
        file 'Capfile'
        directory 'config' do
          file 'appsignal.yml'
          file 'deploy.rb' do
            contains "require 'appsignal/capistrano'"
          end
        end
      }
    end

    specify "should not mention the deploy task" do
      @output.should_not include('No capistrano setup detected!')
      @output.should_not include('appsignal notify_of_deploy -h')
    end
  end

  context "with custom environment" do
    before do
      prepare_destination
      run_generator_in_tmp %w(development my_app_key)
    end

    specify "config file is created" do
      destination_root.should have_structure {
        directory 'config' do
          file 'appsignal.yml'
          no_file 'deploy.rb'
        end
      }
    end
  end

  context "with multiple environments" do
    context "with new environment" do
      before :all do
        prepare_destination
        FileUtils.mkdir(File.expand_path('config', destination_root))
        config_file = File.join('config', 'appsignal.yml')
        File.open(File.expand_path(config_file, destination_root), 'w') do |f|
          f.write("production:\n  api_key: 111")
        end
        run_generator_in_tmp %w(development my_app_key)
      end

      it "exits and tells to manually edit config/appsignal.yml" do
        @output.should include('error  Looks like you already have a config file')
      end
    end
  end
end