require 'spec_helper'

describe Elasticity::CustomJarJob do

  describe ".new" do

    it "should have good defaults" do
      custom_jar = Elasticity::CustomJarJob.new("access", "secret")
      custom_jar.aws_access_key_id.should == "access"
      custom_jar.aws_secret_access_key.should == "secret"
      custom_jar.ec2_key_name.should == "default"
      custom_jar.hadoop_version.should == "0.20"
      custom_jar.instance_count.should == 2
      custom_jar.master_instance_type.should == "m1.small"
      custom_jar.name.should == "Elasticity Custom Jar Job"
      custom_jar.slave_instance_type.should == "m1.small"
      custom_jar.action_on_failure.should == "TERMINATE_JOB_FLOW"
      custom_jar.log_uri.should == nil
    end

  end

  describe "#run" do

    context "when there are arguments provided" do
      it "should run the script with the specified variables and return the jobflow_id" do
        aws = Elasticity::EMR.new("", "")
        aws.should_receive(:run_job_flow).with(
            {
                :name => "Elasticity Custom Jar Job",
                :log_uri => "s3n://slif-test/output/logs",
                :instances => {
                    :ec2_key_name => "default",
                    :hadoop_version => "0.20",
                    :instance_count => 2,
                    :master_instance_type => "m1.small",
                    :slave_instance_type => "m1.small",
                },
                :steps => [
                    {
                        :action_on_failure => "TERMINATE_JOB_FLOW",
                        :hadoop_jar_step => {
                            :jar => "s3n://elasticmapreduce/samples/cloudburst/cloudburst.jar",
                            :args => [
                                "s3n://elasticmapreduce/samples/cloudburst/input/s_suis.br",
                                    "s3n://elasticmapreduce/samples/cloudburst/input/100k.br",
                                    "s3n://slif_hadoop_test/cloudburst/output/2011-12-09",
                            ],
                        },
                        :name => "Execute Custom Jar"
                    }
                ]
            }).and_return("new_jobflow_id")
        Elasticity::EMR.should_receive(:new).with("access", "secret").and_return(aws)

        custom_jar = Elasticity::CustomJarJob.new("access", "secret")
        custom_jar.log_uri = "s3n://slif-test/output/logs"
        custom_jar.action_on_failure = "TERMINATE_JOB_FLOW"
        jobflow_id = custom_jar.run('s3n://elasticmapreduce/samples/cloudburst/cloudburst.jar', [
            "s3n://elasticmapreduce/samples/cloudburst/input/s_suis.br",
                "s3n://elasticmapreduce/samples/cloudburst/input/100k.br",
                "s3n://slif_hadoop_test/cloudburst/output/2011-12-09",
        ])
        jobflow_id.should == "new_jobflow_id"
      end
    end

    context "when there are no arguments provided" do
      it "should run the script with the specified variables and return the jobflow_id" do
        aws = Elasticity::EMR.new("", "")
        aws.should_receive(:run_job_flow).with(
            {
                :name => "Elasticity Custom Jar Job",
                :log_uri => "s3n://slif-test/output/logs",
                :instances => {
                    :ec2_key_name => "default",
                    :hadoop_version => "0.20",
                    :instance_count => 2,
                    :master_instance_type => "m1.small",
                    :slave_instance_type => "m1.small",
                },
                :steps => [
                    {
                        :action_on_failure => "TERMINATE_JOB_FLOW",
                        :hadoop_jar_step => {
                            :jar => "s3n://elasticmapreduce/samples/cloudburst/cloudburst.jar"
                        },
                        :name => "Execute Custom Jar"
                    }
                ]
            }).and_return("new_jobflow_id")
        Elasticity::EMR.should_receive(:new).with("access", "secret").and_return(aws)

        custom_jar = Elasticity::CustomJarJob.new("access", "secret")
        custom_jar.log_uri = "s3n://slif-test/output/logs"
        custom_jar.action_on_failure = "TERMINATE_JOB_FLOW"
        jobflow_id = custom_jar.run('s3n://elasticmapreduce/samples/cloudburst/cloudburst.jar')
        jobflow_id.should == "new_jobflow_id"
      end
    end

  end

  describe "integration happy path" do
    use_vcr_cassette "custom_jar_job/cloudburst", :record => :none
    it "should kick off the sample Amazion EMR Hive application" do
      custom_jar = Elasticity::CustomJarJob.new(AWS_ACCESS_KEY_ID, AWS_SECRET_KEY)
      custom_jar.ec2_key_name = "sharethrough_dev"
      jobflow_id = custom_jar.run('s3n://elasticmapreduce/samples/cloudburst/cloudburst.jar', [
          "s3n://elasticmapreduce/samples/cloudburst/input/s_suis.br",
              "s3n://elasticmapreduce/samples/cloudburst/input/100k.br",
              "s3n://slif_hadoop_test/cloudburst/output/2011-12-09",
      ])
      jobflow_id.should == "j-1IU6NM8OUPS9I"
    end
  end

end