partitioned 0.8.0

data/spec/partitioned/by_time_field_spec.rb

@@ -0,0 +1,182 @@
+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 ByTimeField do
+
+    include TablesSpecHelper
+
+    module TimeField
+      class Employee < Partitioned::ByTimeField
+        belongs_to :company, :class_name => 'Company'
+        attr_accessible :created_at, :name, :company_id
+
+        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 # TimeField
+
+    before(:all) do
+      @employee = TimeField::Employee
+      create_tables
+      dates = @employee.partition_generate_range(DATE_NOW,
+                                                 DATE_NOW + 1.day)
+      @employee.create_new_partition_tables(dates)
+      ActiveRecord::Base.connection.execute <<-SQL
+        insert into employees_partitions.
+          p#{DATE_NOW.strftime('%Y%m%d')}
+          (company_id,name) values (1,'Keith');
+      SQL
+    end
+
+    after(:all) do
+      drop_tables
+    end
+
+    let(:class_by_time_field) { ::Partitioned::ByTimeField }
+
+    describe "model is abstract class" do
+
+      it "returns true" do
+        class_by_time_field.abstract_class.should be_true
+      end
+
+    end # model is abstract class
+
+    describe "#partition_generate_range" do
+
+      it "returns dates array" do
+        class_by_time_field.
+            partition_generate_range(Date.parse('2011-01-05'), Date.parse('2011-01-07')).
+            should == [Date.parse('2011-01-05'), Date.parse('2011-01-06'), Date.parse('2011-01-07')]
+      end
+
+    end # #partition_generate_range
+
+    describe "#partition_normalize_key_value" do
+
+      it "returns date" do
+        class_by_time_field.
+            partition_normalize_key_value(Date.parse('2011-01-05')).
+            should == Date.parse('2011-01-05')
+      end
+
+    end # #partition_normalize_key_value
+
+    describe "#partition_table_size" do
+
+      it "returns 1.day" do
+        class_by_time_field.partition_table_size.should == 1.day
+      end
+
+    end # #partition_table_size
+
+    describe "#partition_time_field" do
+
+      it "raises MethodNotImplemented" do
+        lambda {
+          class_by_time_field.partition_time_field
+        }.should raise_error(MethodNotImplemented)
+      end
+
+    end # #partition_time_field
+
+    describe "partitioned block" do
+
+      let(:data) do
+        class_by_time_field.configurator_dsl.data
+      end
+
+      context "checks data in the on_field is Proc" do
+
+        it "returns Proc" do
+          data.on_field.should be_is_a Proc
+        end
+
+      end # checks data in the on_field is Proc
+
+      context "checks data in the indexes is Proc" do
+
+        it "returns Proc" do
+          data.indexes.first.should be_is_a Proc
+        end
+
+      end # checks data in the indexes is Proc
+
+      context "checks data in the base_name is Proc" do
+
+        it "returns Proc" do
+          data.base_name.should be_is_a Proc
+        end
+
+      end # checks data in the base_name is Proc
+
+      context "checks data in the check_constraint is Proc" do
+
+        it "returns Proc" do
+          data.check_constraint.should be_is_a Proc
+        end
+
+      end # checks data in the check_constraint is Proc
+
+      context "checks data in the on_field" do
+
+        it "returns on_field" do
+          data.on_field.call(@employee).should == :created_at
+        end
+
+      end # checks data in the on_field
+
+      context "checks data in the indexes" do
+
+        it "returns :created_at" do
+          data.indexes.first.call(@employee, nil).field.should == :created_at
+        end
+
+        it "returns empty options hash" do
+          data.indexes.first.call(@employee, nil).options.should == {}
+        end
+
+      end # checks data in the indexes
+
+      context "checks data in the last_partitions_order_by_clause" do
+
+        it "returns last_partitions_order_by_clause" do
+          data.last_partitions_order_by_clause.should == "tablename desc"
+        end
+
+      end # checks data in the last_partitions_order_by_clause
+
+      context "checks data in the base_name" do
+
+        it "returns base_name" do
+          data.base_name.call(@employee, Date.parse('2011-01-05')).should == "20110105"
+        end
+
+      end # checks data in the base_name
+
+      context "checks data in the check_constraint" do
+
+        it "returns check_constraint" do
+          data.check_constraint.
+              call(@employee, Date.parse('2011-01-05')).
+              should == "created_at >= '2011-01-05' AND created_at < '2011-01-06'"
+        end
+
+      end # checks data in the check_constraint
+
+    end # partitioned block
+
+    it_should_behave_like "check that basic operations with postgres works correctly for time key", TimeField::Employee
+
+  end # ByTimeField
+
+end # Partitioned

data/spec/partitioned/by_weekly_time_field_spec.rb

@@ -0,0 +1,100 @@
+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'
+        attr_accessible :name, :company_id, :created_at
+
+        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
+        class_by_weekly_time_field.abstract_class.should be_true
+      end
+
+    end # model is abstract class
+
+    describe "#partition_normalize_key_value" do
+
+      it "returns date with day set to 1st of the week" do
+        class_by_weekly_time_field.
+            partition_normalize_key_value(Date.parse('2011-01-05')).
+            should == Date.parse('2011-01-03')
+      end
+
+    end # #partition_normalize_key_value
+
+    describe "#partition_table_size" do
+
+      it "returns 1.week" do
+        class_by_weekly_time_field.partition_table_size.should == 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
+          data.base_name.should 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
+          data.base_name.call(@employee, Date.parse('2011-01-05')).should == "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

data/spec/partitioned/multi_level/configurator/dsl_spec.rb

@@ -0,0 +1,88 @@
+require 'spec_helper'
+
+describe Partitioned::MultiLevel::Configurator::Dsl do
+
+  before(:all) do
+    class Employee
+
+    end
+  end
+
+  after(:all) do
+    Object.send(:remove_const, :Employee)
+  end
+
+  let!(:dsl) { Partitioned::MultiLevel::Configurator::Dsl.new(Employee) }
+
+  describe "initialize" do
+
+    let!(:data_stubs) do
+      {
+        "on_field" => nil,
+        "indexes" => [],
+        "foreign_keys" => [],
+        "last_partitions_order_by_clause" => nil,
+        "schema_name" => nil,
+        "name_prefix" => nil,
+        "base_name" => nil,
+        "part_name" => nil,
+        "table_name" => nil,
+        "parent_table_schema_name" => nil,
+        "parent_table_name" => nil,
+        "check_constraint" => nil,
+        "encoded_name" => nil,
+        "using_classes" => []
+      }
+    end
+
+    context "when try to create the new object" do
+
+      context "check the model name" do
+
+        it "returns Employer" do
+          dsl.model.should == Employee
+        end
+
+      end # check the model name
+
+      context "check the object data" do
+
+        it "returns data" do
+          dsl.data.instance_values.should == data_stubs
+        end
+
+      end # check the object data
+
+    end # when try to create a new object
+
+  end # initialize
+
+  describe "on" do
+
+    context "when try to set the field which used to partition child tables" do
+
+      it "raises InvalidForMultiLevelPartitioning" do
+        lambda {
+          dsl.on
+        }.should raise_error(Partitioned::MultiLevel::Configurator::Dsl::InvalidForMultiLevelPartitioning)
+      end
+
+    end # when try to set the field which used to partition child tables
+
+  end # on
+
+  describe "using_classes" do
+
+    context "when try to set the using_classes field" do
+
+      it "returns using_classes" do
+        dsl.using_classes("ByCompanyId", "ByCreatedAt")
+        dsl.data.using_classes.first.should == "ByCompanyId"
+        dsl.data.using_classes.last.should == "ByCreatedAt"
+      end
+
+    end # when try to set the using_classes field
+
+  end # using_classes
+
+end

data/spec/partitioned/multi_level/configurator/reader_spec.rb

@@ -0,0 +1,147 @@
+require 'spec_helper'
+
+module Partitioned
+  class MultiLevel
+    module Configurator
+      describe Reader do
+
+        before(:all) do
+          class Partitioned::ById
+            def self.partition_field
+              :id
+            end
+            def self.partition_table_size
+              return 1
+            end
+          end
+          class Partitioned::ByCreatedAt
+            def self.partition_field
+              :created_at
+            end
+          end
+          class Employee < MultiLevel
+            def self.first_partition_field
+              :id
+            end
+            def self.second_partition_field
+              :created_at
+            end
+            partitioned do |partition|
+              partition.index :id, :unique => true
+              partition.using_classes Partitioned::ById, Partitioned::ByCreatedAt
+            end
+          end
+        end
+
+        after(:all) do
+          Partitioned::MultiLevel::Configurator.send(:remove_const, :Employee)
+        end
+
+        let!(:reader) do
+          Partitioned::MultiLevel::Configurator::Reader.new(Employee)
+        end
+
+        describe "using_configurators" do
+
+          context "checking arrays length" do
+
+            it "returns 6" do
+              reader.send(:using_configurators).length.should == 6
+            end
+
+          end # checking array length
+
+          context "checking models class for each of using_configurators" do
+
+            it "returns 'Partitioned::ById'" do
+              for i in 0..2
+                reader.send(:using_configurators)[i].model.to_s.should == "Partitioned::ById"
+              end
+            end
+
+            it "returns 'Partitioned::ByCreatedAt'" do
+              for i in 3..5
+                reader.send(:using_configurators)[i].model.to_s.should == "Partitioned::ByCreatedAt"
+              end
+            end
+
+          end # checking models class for each of using_configurators
+
+        end # using_configurators
+
+        describe "on_fields" do
+
+          context "when on_field value is set by default" do
+
+            it "returns [:id, :created_at]" do
+              reader.on_fields.sort{|a,b| a.to_s <=> b.to_s}.
+                  should == [:id, :created_at].sort{|a,b| a.to_s <=> b.to_s}
+            end
+
+          end # when on_filed value is set by default
+
+        end # on_fields
+
+        describe "parent_table_schema_name" do
+
+          context "when parent_table_schema_name value is set without options" do
+
+            it "returns public" do
+              reader.parent_table_schema_name.should == "public"
+            end
+
+          end # when parent_table_schema_name value is set by default
+
+          context "when parent_table_schema_name value is set with options" do
+
+            it "returns employees_partitions" do
+              reader.parent_table_schema_name(1, Date.parse("2011-01-03")).should == "employees_partitions"
+            end
+
+          end # when parent_table_schema_name value is set by value
+
+        end # parent_table_schema_name
+
+        describe "parent_table_name" do
+
+          context "when parent_table_name value is set without options" do
+
+            it "returns employees" do
+              reader.parent_table_name.should == "employees"
+            end
+
+          end # when parent_table_name value is set without options
+
+          context "when parent_table_name value is set with options" do
+
+            it "returns employees_partitions.p0" do
+              reader.parent_table_name(1, Date.parse("2011-01-03")).should == "employees_partitions.p1"
+            end
+
+          end # when parent_table_name value is set with options
+
+        end # parent_table_name
+
+        describe "check_constraint" do
+
+          it "returns check_constraint" do
+            reader.check_constraint(1).
+                should == "( id = 1 )"
+            reader.check_constraint(1, Date.parse('2011-10-10')).
+                should == "created_at >= '2011-10-10' AND created_at < '2011-10-17'"
+          end
+
+        end # check_constraint
+
+        describe "base_name" do
+
+          it "returns base_name" do
+            reader.base_name(1, Date.parse('2011-10-10')).should == "1_20111010"
+          end
+
+        end # base_name
+
+      end # Reader
+    end # Configurator
+  end # MultiLevel
+end # Partitioned