Contents

module DbHelpers

  def self.pg?
    ENV.has_key? "PG"
  end

  def db_setup(opts={})
    use_time = opts[:use_time]
    DB.drop_table(:room_versions) if DB.table_exists?(:room_versions)
    DB.drop_table(:rooms) if DB.table_exists?(:rooms)
    DB.create_table! :rooms do
      primary_key :id
      Boolean     :disabled, null: false, default: false
    end
    DB.create_table! :room_versions do
      primary_key :id
      foreign_key :master_id, :rooms
      String      :name
      Fixnum      :price
      Fixnum      :length
      Fixnum      :width
      send(use_time ? :Time : :Date, :created_at)
      send(use_time ? :Time : :Date, :expired_at)
      send(use_time ? :Time : :Date, :valid_from)
      send(use_time ? :Time : :Date, :valid_to)
    end
    @version_class = Class.new Sequel::Model do
      set_dataset :room_versions
      def validate
        super
        errors.add(:name, "is required") unless name
        errors.add(:price, "is required") unless price
      end
    end

    bitemporal_options = {version_class: @version_class}
    bitemporal_options[:audit_class] = opts[:audit_class] if opts[:audit_class]
    bitemporal_options[:audit_updated_by_method] = opts[:audit_updated_by_method] if opts[:audit_updated_by_method]
    bitemporal_options[:ranges] = opts[:ranges] if opts[:ranges]

    @master_class = Class.new Sequel::Model do
      set_dataset :rooms
      plugin :bitemporal, bitemporal_options
    end
  end

  def db_truncate
    if DbHelpers.pg?
      @version_class.truncate cascade: true
      @master_class.truncate cascade: true
    else
      @version_class.truncate
      @master_class.truncate
    end
  end
end

Version data entries

5 entries across 5 versions & 1 rubygems

Version	Path
sequel_bitemporal-0.6.16	spec/support/db.rb
sequel_bitemporal-0.6.15	spec/support/db.rb
sequel_bitemporal-0.6.14	spec/support/db.rb
sequel_bitemporal-0.6.13	spec/support/db.rb
sequel_bitemporal-0.6.12	spec/support/db.rb