require File.expand_path(File.dirname(__FILE__) + '/../spec_helper')

describe SampleSetsController do
  include AuthenticatedSpecHelper

  before(:each) do
    login_as_user
    
    projects = [mock_model(Project), mock_model(Project)]
    @current_user.stub!(:accessible_projects).and_return(projects)
    NamingScheme.stub!(:find).and_return(
      [mock_model(NamingScheme), mock_model(NamingScheme)]
    )
    SamplePrepKit.stub!(:find).and_return(
      [mock_model(SamplePrepKit), mock_model(SamplePrepKit)]
    )
    ReferenceGenome.stub!(:find).and_return(
      [mock_model(ReferenceGenome), mock_model(ReferenceGenome)]
    )
  end
    
  describe "handling GET /sample_sets/new" do
    describe "step 1" do
      before(:each) do
        @sample_set = mock_model(SampleSet)
        SampleSet.stub!(:new).and_return(@sample_set)
      end

      def do_get
        get :new
      end

      it "should be successful" do
        do_get
        response.should be_success
      end

      it "should render new template" do
        do_get
        response.should render_template('new')
      end

      it "should create an new sample_set" do
        SampleSet.should_receive(:new).and_return(@sample_set)
        do_get
      end

      it "should not save the new sample_set" do
        @sample_set.should_not_receive(:save)
        do_get
      end

      it "should assign the new sample_set for the view" do
        do_get
        assigns[:sample_set].should equal(@sample_set)
      end      
    end
    
    describe "step 2" do
      before(:each) do
        @sample_set = mock_model(SampleSet)
        SampleSet.stub!(:new).and_return(@sample_set)
      end
      
      def do_get
        get :new, :step => "2", :sample_set => { :number_of_samples => 2 }
      end
      
      describe "with invalid sample set data" do
        before(:each) do
          @sample_set.stub!(:valid?).and_return(false)
        end
        
        it "should create an new sample_set" do
          SampleSet.should_receive(:new).and_return(@sample_set)
          do_get
        end
        
        it "should check the validity of the sample set" do
          @sample_set.should_receive(:valid?).and_return(false)
          do_get
        end
        
        it "should be successful" do
          do_get
          response.should be_success
        end

        it "should render new template" do
          do_get
          response.should render_template('new')
        end
      end
      
      describe "with valid sample set data" do
        before(:each) do
          @sample_set.stub!(:valid?).and_return(true)
          @sample_set.stub!(:submission_date).and_return('2008-02-01')
          @sample_set.stub!(:project_id).and_return(1)
          @sample_set.stub!(:sample_prep_kit_id).and_return(1)
          @sample_set.stub!(:budget_number).and_return("1234")
          @sample_set.stub!(:reference_genome_id).and_return(1)
          @sample_set.stub!(:desired_read_length).and_return(18)
          @sample_set.stub!(:alignment_start_position).and_return(1)
          @sample_set.stub!(:alignment_end_position).and_return(17)
          @sample_set.stub!(:insert_size).and_return(150)
          @sample = mock_model(Sample)
          @sample.stub!(:populate_default_visibilities_and_texts)
          Sample.stub!(:new).and_return( @sample )
        end

        describe "without a naming scheme" do
          before(:each) do
            @sample_set.stub!(:naming_scheme).and_return(nil)
            @sample_set.stub!(:naming_scheme_id).and_return(nil)
          end

          it "should create an new sample_set" do
            SampleSet.should_receive(:new).and_return(@sample_set)
            do_get
          end
          
          it "should check the validity of the sample set" do
            @sample_set.should_receive(:valid?).and_return(true)
            do_get
          end

          it "should be successful" do
            do_get
            response.should be_success
          end

          it "should render new template" do
            do_get
            response.should render_template('new')
          end

          it "should not save the new sample_set" do
            @sample_set.should_not_receive(:save)
            do_get
          end

          it "should assign the new sample_set for the view" do
            do_get
            assigns[:sample_set].should equal(@sample_set)
          end
        end

        describe "with a naming scheme" do
          before(:each) do
            @naming_scheme = mock_model(NamingScheme)
            @sample_set.stub!(:naming_scheme).and_return(@naming_scheme)
            @sample_set.stub!(:naming_scheme_id).and_return(1)
            @naming_scheme.stub!(:ordered_naming_elements).and_return(
              [mock_model(NamingElement), mock_model(NamingElement)]
            )
            NamingScheme.stub!(:find).and_return(nil)
          end
          
          it "should create an new sample_set" do
            SampleSet.should_receive(:new).and_return(@sample_set)
            do_get
          end
          
          it "should check the validity of the sample set" do
            @sample_set.should_receive(:valid?).and_return(true)
            do_get
          end

          it "should be successful" do
            do_get
            response.should be_success
          end

          it "should render new template" do
            do_get
            response.should render_template('new')
          end

          it "should not save the new sample_set" do
            @sample_set.should_not_receive(:save)
            do_get
          end

          it "should assign the new sample_set for the view" do
            do_get
            assigns[:sample_set].should equal(@sample_set)
          end
        end
      end
    end
  end
  
  describe "handling POST /sample_sets" do
    before(:each) do
      @sample_set = mock_model(SampleSet, :to_param => "1")
      SampleSet.stub!(:new).and_return(@sample_set)
      @sample = mock_model(Sample)
      Sample.stub!(:new).and_return(@sample)
      @sample_set.stub!(:samples=).and_return(true)
      Notifier.stub!(:deliver_sample_submission_notification)
    end
   
    describe "with a valid sample set" do
      before(:each) do
        @sample_set.stub!(:valid?).and_return(true)
        @sample.stub!(:save).and_return(true)
      end
  
      def do_post
        post :create,
          :sample_set => {"submission_date(2i)"=>"10", "naming_scheme_id"=>"",
                          "sample_prep_kit_id"=>"1", "number_of_samples"=>"2",
                          "submission_date(3i)"=>"6", "alignment_end_position"=>"18",
                          "desired_read_length"=>"18", "reference_genome_id"=>"2",
                          "alignment_start_position"=>"1", "lab_group_id"=>"1",
                          "budget_number"=>"1234", "insert_size"=>"150",
                          "submission_date(1i)"=>"2008"},
           :sample => {"0"=>{"status"=>"", "reference_genome_id"=>"1",
             "name_on_tube"=>"1121", "desired_read_length"=>"18",
             "schemed_name"=>{"Protocol"=>"203", "Cell Type"=>"197", "Time"=>"184",
             "Biological Replicate"=>"", "Antibody"=>"291", "Stimulus"=>"189",
             "Exclude From Analysis"=>"204", "Technical Replicate"=>"202", "Date"=>""},
             "submission_date"=>"2008-10-02", "budget_number"=>"1234", "insert_size"=>"150",
             "sample_prep_kit_id"=>"1"} }
      end
  
      it "should create a new sample_set" do
        SampleSet.should_receive(:new).and_return(@sample_set)
        do_post
      end
      
      it "should find the new sample set valid" do
        @sample_set.should_receive(:valid?).and_return(true)
        do_post
      end
      
      it "should save the sample" do
        @sample.should_receive(:save).and_return(true)
        do_post
      end      

      it "should redirect to the list of samples" do
        do_post
        response.should redirect_to(samples_url)
      end
      
      it "should send email notifications" do
        Notifier.should_receive(:deliver_sample_submission_notification)
        do_post
      end
    end
    
    describe "with an invalid sample set" do
      before(:each) do
        @sample_set.stub!(:valid?).and_return(false)
        @naming_scheme = mock_model(NamingScheme)
        @sample_set.stub!(:naming_scheme).and_return(@naming_scheme)
        @naming_scheme.stub!(:ordered_naming_elements).and_return( [mock_model(NamingElement)] )
      end

      def do_post
        post :create,
          :sample_set => {"submission_date(2i)"=>"10", "naming_scheme_id"=>"",
                          "sample_prep_kit_id"=>"1", "number_of_samples"=>"2",
                          "submission_date(3i)"=>"6", "alignment_end_position"=>"18",
                          "desired_read_length"=>"18", "reference_genome_id"=>"2",
                          "alignment_start_position"=>"1", "lab_group_id"=>"1",
                          "budget_number"=>"1234", "insert_size"=>"150",
                          "submission_date(1i)"=>"2008"},
           :sample => {"0"=>{"status"=>"", "reference_genome_id"=>"1",
             "name_on_tube"=>"1121", "desired_read_length"=>"18",
             "schemed_name"=>{"Protocol"=>"203", "Cell Type"=>"197", "Time"=>"184",
             "Biological Replicate"=>"", "Antibody"=>"291", "Stimulus"=>"189",
             "Exclude From Analysis"=>"204", "Technical Replicate"=>"202", "Date"=>""},
             "submission_date"=>"2008-10-02", "budget_number"=>"1234", "insert_size"=>"150",
             "sample_prep_kit_id"=>"1"} }
      end

      it "should find the new sample set invalid" do
        @sample_set.should_receive(:valid?).and_return(false)
        do_post
      end

      it "should get the naming scheme for the sample set" do
        @sample_set.should_receive(:naming_scheme).and_return(@naming_scheme)
        do_post
      end
      
      it "should get the naming elements for the naming scheme" do
        @naming_scheme.should_receive(:ordered_naming_elements).and_return( [mock_model(NamingElement)] )
        do_post
      end
      
      it "should set the step parameter to go to step 2" do
        do_post
        params[:step].should == "2"
      end
  
      it "should re-render 'new' template at step 2" do
        do_post
        response.should render_template('new')
      end
    end
  end
end