require 'spec_helper'

describe Squealer::Target do
  let(:export_dbc) { mock(Mysql) }
  let(:table_name) { :test_table }
  let(:row_id) { 0 }

  before(:each) do
    Squealer::Database.instance.should_receive(:export).at_least(:once).and_return(export_dbc)
    st = mock(Mysql::Stmt)
    export_dbc.should_receive(:prepare).at_least(:once).and_return(st)
    st.should_receive(:execute).at_least(:once)
  end

  it "sends the sql to the export database" do
    Squealer::Target.new(export_dbc, table_name, row_id) { nil }
  end

  describe "#target" do

    it "pushes itself onto the targets stack when starting" do
      @target1 = nil
      @target2 = nil
      target1 = Squealer::Target.new(export_dbc, table_name, row_id) do
        @target1 = Squealer::Target.current
        Squealer::Target.new(export_dbc, "#{table_name}_2", row_id) do
          @target2 = Squealer::Target.current
          @target2.should_not == @target1
        end
      end
      target1.should === @target1
    end

    it "pops itself off the targets stack when finished" do
      Squealer::Target.new(export_dbc, table_name, row_id) { nil }
      Squealer::Target.current.should be_nil
    end

    context "generates SQL command strings" do

      let(:target) { Squealer::Target.new(export_dbc, table_name, row_id) { nil } }

      it "targets the table" do
        target.sql.should =~ /^INSERT #{table_name} /
      end

      it "uses an INSERT ... ON DUPLICATE KEY UPDATE statement" do
        target.sql.should =~ /^INSERT .* ON DUPLICATE KEY UPDATE /
      end

      it "includes the primary key name in the INSERT" do
        target.sql.should =~ / \(id\) VALUES/
      end

      it "includes the primary key value in the INSERT" do
        # target.sql.should =~ / VALUES \('#{row_id}'\) /
        target.sql.should =~ / VALUES \(\?\) /
      end

    end

    context "with inner block" do

      it "yields inner blocks" do
        block_done = false
        target = Squealer::Target.new(export_dbc, table_name, row_id) { block_done = true }
        block_done.should be_true
      end

      it "yields inner blocks first" do
        Squealer::Target.new(export_dbc, table_name, row_id) { Squealer::Target.current.sql.should be_empty }
      end

      it "yields inner blocks first and they can assign to this target" do
        target = Squealer::Target.new(export_dbc, table_name, row_id) { Squealer::Target.current.assign(:colA) { 42 } }
        target.sql.should =~ /colA/
        # target.sql.should =~ /42/
        target.sql.should =~ /\?/
      end

      context "with 2 columns" do

        let(:value_1) { 42 }
        let(:target) do
          Squealer::Target.new(export_dbc, table_name, row_id) { Squealer::Target.current.assign(:colA) { value_1 } }
        end

        it "includes the column name in the INSERT" do
          target.sql.should =~ /\(id,colA\) VALUES/
        end

        it "includes the column value in the INSERT" do
          # target.sql.should =~ /VALUES \('#{row_id}','#{value_1}'\)/
          target.sql.should =~ /VALUES \(\?,\?\)/
        end

        it "includes the column name and value in the UPDATE" do
          # target.sql.should =~ /UPDATE colA='#{value_1}'/
          target.sql.should =~ /UPDATE colA=\?/
        end

      end

      context "with 3 columns" do

        let(:value_1) { 42 }
        let(:value_2) { 'foobar' }
        let(:target) do
          Squealer::Target.new(export_dbc, table_name, row_id) do
            Squealer::Target.current.assign(:colA) { value_1 }
            Squealer::Target.current.assign(:colB) { value_2 }
          end
        end

        it "includes the column names in the INSERT" do
          target.sql.should =~ /\(id,colA,colB\) VALUES/
        end

        it "includes the column values in the INSERT" do
          # target.sql.should =~ /VALUES \('#{row_id}','#{value_1}','#{value_2}'\)/
          target.sql.should =~ /VALUES \(\?,\?,\?\)/
        end

        it "includes the column names and values in the UPDATE" do
          # target.sql.should =~ /UPDATE colA='#{value_1}',colB='#{value_2}'/
          target.sql.should =~ /UPDATE colA=\?,colB=\?/
        end

      end

    end

  end

end