require_relative '../../helpers'
require 'symphony/tasks/ssh'
require 'symphony/tasks/sshscript'

context Symphony::Task::SSHScript do

	before( :each ) do
		Symphony::Task::SSH.configure(
			key:  '/tmp/sekrit.rsa',
			user: 'symphony'
		)
	end


	describe 'utility' do

		it "can generate an appropriate tempfile name" do
			instance = Class.new( described_class ).new( 'queue' )
			tmpname = instance.send( :make_remote_filename, "fancy-script.tmpl" )
			expect( tmpname ).to match( %r|^/tmp/fancy-script-[[:xdigit:]]{6}| )

			tmpname = instance.send( :make_remote_filename, "fancy-script.tmpl", "/var/tmp/" )
			expect( tmpname ).to match( %r|/var/tmp/fancy-script-[[:xdigit:]]{6}| )

			tmpname = instance.send( :make_remote_filename, "fancy-script.tmpl", '' )
			expect( tmpname ).to match( %r|fancy-script-[[:xdigit:]]{6}| )
		end
	end


	describe 'subclassed' do
		let( :instance ) { Class.new(described_class).new('queue') }
		let( :payload ) {
			{ 'template' => 'script', 'host' => 'example.com' }
		}
		let( :opts ) {
			opts = described_class::DEFAULT_SSH_OPTIONS
			opts.merge!(
				:port    => 22,
				:keys    => ['/tmp/sekrit.rsa']
			)
			opts
		}
		let( :template ) { Inversion::Template.new("Hi there, <?attr name?>!") }

		before( :each ) do
			allow( Inversion::Template ).to receive( :load ).and_return( template )
			allow( instance ).to receive( :make_remote_filename ).and_return( "/tmp/script_temp" )
		end

		it "aborts if there is no template in the payload" do
			expect {
				instance.work( {}, {} )
			}.to raise_exception( ArgumentError, /missing required option 'template'/i )
		end

		it "aborts if there is no host in the payload" do
			expect {
				instance.work({ 'template' => 'boop' }, {} )
			}.to raise_exception( ArgumentError, /missing required option 'host'/i )
		end

		it "adds debugging output if specified in the payload" do
			payload[ 'debug' ] = true

			options = opts.dup
			options.merge!(
				:logger  => Loggability[ Net::SSH ],
				:verbose => :debug
			)

			expect( Net::SSH ).to receive( :start ).with( 'example.com', 'symphony', options )
			instance.work( payload, {} )
		end

		it "attaches attributes to the scripts from the payload" do
			payload[ 'attributes' ] = { :name => 'Handsome' }

			conn = double( :ssh_connection )
			expect( instance ).to receive( :upload_script ).
				with( conn, "Hi there, Handsome!", "/tmp/script_temp" )
			expect( conn ).to receive( :exec! ).with( "/tmp/script_temp" )
			expect( conn ).to receive( :exec! ).with( "rm /tmp/script_temp" )

			expect( Net::SSH ).to receive( :start ).
				with( 'example.com', 'symphony', opts ).and_yield( conn )

			instance.work( payload, {} )
		end

		it "uploads the file and sets it executable" do
			conn = double( :ssh_connection )
			sftp = double( :sftp_connection )
			file = double( :remote_file_obj )
			fh   = double( :remote_filehandle )

			expect( conn ).to receive( :sftp ).and_return( sftp )
			expect( sftp ).to receive( :file ).and_return( file )

			expect( file ).to receive( :open ).
				with( "/tmp/script_temp", "w", 0755 ).and_yield( fh )
			expect( fh ).to receive( :print ).with( "Hi there, !" )

			expect( conn ).to receive( :exec! ).with( "/tmp/script_temp" )
			expect( conn ).to receive( :exec! ).with( "rm /tmp/script_temp" )

			expect( Net::SSH ).to receive( :start ).
				with( 'example.com', 'symphony', opts ).and_yield( conn )

			instance.work( payload, {} )
		end

		it "can override how it cleans the remote script up" do
			payload[ 'delete_cmd' ] = 'del'

			conn = double( :ssh_connection )
			expect( instance ).to receive( :upload_script ).
				with( conn, "Hi there, !", "/tmp/script_temp" )
			expect( conn ).to receive( :exec! ).with( "/tmp/script_temp" )
			expect( conn ).to receive( :exec! ).with( "del /tmp/script_temp" )

			expect( Net::SSH ).to receive( :start ).
				with( 'example.com', 'symphony', opts ).and_yield( conn )

			instance.work( payload, {} )
		end

		it "can run the script with a specific interpreter" do
			payload[ 'run_binary' ] = 'ruby'

			conn = double( :ssh_connection )
			expect( instance ).to receive( :upload_script ).
				with( conn, "Hi there, !", "/tmp/script_temp" )
			expect( conn ).to receive( :exec! ).with( "ruby /tmp/script_temp" )
			expect( conn ).to receive( :exec! ).with( "rm /tmp/script_temp" )

			expect( Net::SSH ).to receive( :start ).
				with( 'example.com', 'symphony', opts ).and_yield( conn )

			instance.work( payload, {} )
		end

		it "leaves the remote script in place if asked" do
			payload[ 'nocleanup' ] = true

			conn = double( :ssh_connection )
			expect( instance ).to receive( :upload_script ).
				with( conn, "Hi there, !", "/tmp/script_temp" )
			expect( conn ).to receive( :exec! ).with( "/tmp/script_temp" )
			expect( conn ).to_not receive( :exec! ).with( "rm /tmp/script_temp" )

			expect( Net::SSH ).to receive( :start ).
				with( 'example.com', 'symphony', opts ).and_yield( conn )

			instance.work( payload, {} )
		end

		it "remembers the output of the remote script" do
			conn = double( :ssh_connection )
			expect( instance ).to receive( :upload_script ).
				with( conn, "Hi there, !", "/tmp/script_temp" )
			expect( conn ).to receive( :exec! ).with( "/tmp/script_temp" ).and_return( "Hi there, !" )
			expect( conn ).to receive( :exec! ).with( "rm /tmp/script_temp" )

			expect( Net::SSH ).to receive( :start ).
				with( 'example.com', 'symphony', opts ).and_yield( conn )

			instance.work( payload, {} )
			output = instance.instance_variable_get( :@output )
			expect( output ).to eq( 'Hi there, !' )
		end
	end
end