require 'spec_helper'
require "#{File.dirname(__FILE__)}/../support/tables_spec_helper"
require "#{File.dirname(__FILE__)}/../support/shared_example_spec_helper_for_time_key"

module Partitioned

  describe ByWeeklyTimeField do

    include TablesSpecHelper

    module WeeklyTimeField
      class Employee < ByWeeklyTimeField
        belongs_to :company, :class_name => 'Company'

        def self.partition_time_field
          return :created_at
        end

        partitioned do |partition|
          partition.index :id, :unique => true
          partition.foreign_key :company_id
        end
      end # Employee
    end # WeeklyTimeField

    before(:all) do
      @employee = WeeklyTimeField::Employee
      create_tables
      dates = @employee.partition_generate_range(DATE_NOW,
                                                 DATE_NOW + 7.days)
      @employee.create_new_partition_tables(dates)
      ActiveRecord::Base.connection.execute <<-SQL
        insert into employees_partitions.
          p#{DATE_NOW.at_beginning_of_week.strftime('%Y%m%d')}
          (company_id,name) values (1,'Keith');
      SQL
    end

    after(:all) do
      drop_tables
    end

    let(:class_by_weekly_time_field) { ::Partitioned::ByWeeklyTimeField }

    describe "model is abstract class" do

      it "returns true" do
        expect(class_by_weekly_time_field.abstract_class).to be_truthy
      end

    end # model is abstract class

    describe "#partition_normalize_key_value" do

      it "returns date with day set to 1st of the week" do
        expect(class_by_weekly_time_field.
            partition_normalize_key_value(Date.parse('2011-01-05'))).
            to eq(Date.parse('2011-01-03'))
      end

    end # #partition_normalize_key_value

    describe "#partition_table_size" do

      it "returns 1.week" do
        expect(class_by_weekly_time_field.partition_table_size).to eq(1.week)
      end

    end # #partition_table_size

    describe "partitioned block" do

      let(:data) do
        class_by_weekly_time_field.configurator_dsl.data
      end

      context "checks data in the base_name is Proc" do

        it "returns Proc" do
          expect(data.base_name).to be_is_a Proc
        end

      end # checks data in the base_name is Proc

      context "checks data in the base_name" do

        it "returns base_name" do
          expect(data.base_name.call(@employee, Date.parse('2011-01-05'))).to eq("20110103")
        end

      end # checks data in the base_name

    end # partitioned block

    it_should_behave_like "check that basic operations with postgres works correctly for time key", WeeklyTimeField::Employee

  end # ByWeeklyTimeField

end # Partitioned