require File.dirname(__FILE__) + '/../../spec_helper'
require 'active_record'
require 'database_cleaner/active_record/truncation'


module ActiveRecord
  module ConnectionAdapters
    [MysqlAdapter, Mysql2Adapter, SQLite3Adapter, JdbcAdapter, PostgreSQLAdapter, IBM_DBAdapter].each do |adapter|
      describe adapter, "#truncate_table" do
        it "responds" do
          adapter.new("foo").should respond_to(:truncate_table)
        end
        it "should truncate the table"
      end
    end
  end
end

module DatabaseCleaner
  module ActiveRecord

    describe Truncation do
      let(:connection) { mock('connection') }


      before(:each) do
        connection.stub!(:disable_referential_integrity).and_yield
        connection.stub!(:views).and_return([])
        ::ActiveRecord::Base.stub!(:connection).and_return(connection)
      end

      it "should truncate all tables except for schema_migrations" do
        connection.stub!(:tables).and_return(%w[schema_migrations widgets dogs])
        
        connection.should_receive(:truncate_tables).with(['widgets', 'dogs'])
        Truncation.new.clean
      end

      it "should only truncate the tables specified in the :only option when provided" do
        connection.stub!(:tables).and_return(%w[schema_migrations widgets dogs])

        connection.should_receive(:truncate_tables).with(['widgets'])

        Truncation.new(:only => ['widgets']).clean
      end

      it "should not truncate the tables specified in the :except option" do
        connection.stub!(:tables).and_return(%w[schema_migrations widgets dogs])

        connection.should_receive(:truncate_tables).with(['dogs'])

        Truncation.new(:except => ['widgets']).clean
      end

      it "should raise an error when :only and :except options are used" do
        running {
          Truncation.new(:except => ['widgets'], :only => ['widgets'])
        }.should raise_error(ArgumentError)
      end

      it "should raise an error when invalid options are provided" do
        running { Truncation.new(:foo => 'bar') }.should raise_error(ArgumentError)
      end

      it "should not truncate views" do
        connection.stub!(:tables).and_return(%w[widgets dogs])
        connection.stub!(:views).and_return(["widgets"])

        connection.should_receive(:truncate_tables).with(['dogs'])

        Truncation.new.clean
      end

    end
  end
end