mongoid-ultra 9.0.0.0.alpha1

data/spec/support/macros.rb

@@ -0,0 +1,129 @@
+# frozen_string_literal: true
+
+module Mongoid
+  module Macros
+
+    def use_spec_mongoid_config
+      around do |example|
+        config_path = File.join(File.dirname(__FILE__), "..", "config", "mongoid.yml")
+
+        Mongoid::Clients.clear
+        Mongoid.load!(config_path, :test)
+
+        begin
+          example.run
+        ensure
+          Mongoid::Config.reset
+        end
+      end
+    end
+
+    def config_override(key, value)
+      around do |example|
+        existing = Mongoid.send(key)
+
+        Mongoid.send("#{key}=", value)
+
+        example.run
+
+        Mongoid.send("#{key}=", existing)
+      end
+    end
+
+    def with_config_values(key, *values, &block)
+      values.each do |value|
+        context "when #{key} is #{value}" do
+          config_override key, value
+
+          class_exec(value, &block)
+        end
+      end
+    end
+
+    def driver_config_override(key, value)
+      around do |example|
+        existing = Mongo.send(key)
+
+        Mongo.send("#{key}=", value)
+
+        example.run
+
+        Mongo.send("#{key}=", existing)
+      end
+    end
+
+    def with_driver_config_values(key, *values, &block)
+      values.each do |value|
+        context "when #{key} is #{value}" do
+          driver_config_override key, value
+
+          class_exec(value, &block)
+        end
+      end
+    end
+
+    def restore_config_clients
+      around do |example|
+        # Duplicate the config because some tests mutate it.
+        old_config = Mongoid::Config.clients.dup
+        example.run
+        Mongoid::Config.send(:clients=, old_config)
+      end
+    end
+
+    def query_cache_enabled
+      around do |example|
+        Mongoid::QueryCache.cache do
+          example.run
+        end
+      end
+    end
+
+    def override_query_cache(enabled)
+      around do |example|
+        cache_enabled = Mongoid::QueryCache.enabled?
+        Mongoid::QueryCache.enabled = enabled
+        example.run
+        Mongoid::QueryCache.enabled = cache_enabled
+      end
+    end
+
+    # Override the global persistence context.
+    #
+    # @param [ :client, :database ] component The component to override.
+    # @param [ Object ] value The value to override to.
+    def persistence_context_override(component, value)
+      around do |example|
+        meth = "#{component}_override"
+        old_value = Mongoid::Threaded.send(meth)
+        Mongoid::Threaded.send("#{meth}=", value)
+        example.run
+        Mongoid::Threaded.send("#{meth}=", old_value)
+      end
+    end
+
+    def time_zone_override(tz)
+      around do |example|
+        old_tz = Time.zone
+        Time.zone = tz
+        example.run
+        Time.zone = old_tz
+      end
+    end
+
+    def with_default_i18n_configs
+      around do |example|
+        I18n.locale = :en
+        I18n.default_locale = :en
+        I18n.try(:fallbacks=, I18n::Locale::Fallbacks.new)
+        I18n.enforce_available_locales = false
+        example.run
+      ensure
+        I18n.locale = :en
+        I18n.default_locale = :en
+        I18n.try(:fallbacks=, I18n::Locale::Fallbacks.new)
+        I18n.enforce_available_locales = false
+      end
+    end
+  end
+end

data/spec/support/models/account.rb

@@ -0,0 +1,38 @@
+# frozen_string_literal: true
+
+class Account
+  include Mongoid::Document
+
+  field :_id, type: String, overwrite: true, default: ->{ name.try(:parameterize) }
+
+  field :number, type: String
+  field :balance, type: Integer
+  field :nickname, type: String
+  field :name, type: String
+  field :balanced, type: Mongoid::Boolean, default: ->{ balance? ? true : false }
+
+  field :overridden, type: String
+
+  embeds_many :memberships
+  belongs_to :creator, class_name: "User", foreign_key: :creator_id
+  belongs_to :person
+  has_many :alerts, autosave: false
+  has_and_belongs_to_many :agents
+  has_one :comment, validate: false
+
+  validates_presence_of :name
+  validates_presence_of :nickname, on: :upsert
+  validates_length_of :name, maximum: 10, on: :create
+
+  def overridden
+    self[:overridden] = "not recommended"
+  end
+
+  # MONGOID-3365
+  field :period_started_at, type: Time
+  has_many :consumption_periods, dependent: :destroy, validate: false
+
+  def current_consumption
+    consumption_periods.find_or_create_by(started_at: period_started_at)
+  end
+end

data/spec/support/models/acolyte.rb

@@ -0,0 +1,19 @@
+# frozen_string_literal: true
+
+class Acolyte
+  include Mongoid::Document
+  include Mongoid::Attributes::Dynamic
+  field :status
+  field :name
+
+  embeds_many :versions, as: :memorable
+  belongs_to :church
+
+  default_scope ->{ asc(:name) }
+  scope :active, ->{ where(status: "active") }
+  scope :named, ->{ where(:name.exists => true) }
+
+  def callback_test?
+    name == "callback-test"
+  end
+end

data/spec/support/models/actor.rb

@@ -0,0 +1,20 @@
+# frozen_string_literal: true
+
+class Actor
+  include Mongoid::Document
+  field :name
+  field :after_custom_count, type: Integer, default: 0
+  has_and_belongs_to_many :tags
+  embeds_many :things, validate: false, cascade_callbacks: true
+  accepts_nested_attributes_for :things, allow_destroy: true
+
+  define_model_callbacks :custom
+
+  def do_something
+    run_callbacks(:custom) do
+      self.name = "custom"
+    end
+  end
+end
+
+require "support/models/actress"

data/spec/support/models/actress.rb

@@ -0,0 +1,4 @@
+# frozen_string_literal: true
+
+class Actress < Actor
+end

data/spec/support/models/address.rb

@@ -0,0 +1,87 @@
+# frozen_string_literal: true
+
+class Address
+  include Mongoid::Document
+
+  field :_id, type: String, overwrite: true, default: ->{ street.try(:parameterize) }
+
+  attr_accessor :mode
+
+  field :address_type
+  field :number, type: Integer
+  field :no, type: Integer
+  field :h, as: :house, type: Integer
+  field :street
+  field :city
+  field :state
+  field :post_code
+  field :parent_title
+  field :services, type: Array
+  field :aliases, as: :a, type: Array
+  field :test, type: Array
+  field :latlng, type: Array
+  field :map, type: Hash
+  field :move_in, type: DateTime
+  field :end_date, type: Date
+  field :s, type: String, as: :suite
+  field :name, localize: true
+
+  alias_attribute :n, :name
+
+  embeds_many :locations, validate: false
+  embeds_one :code, validate: false
+  embeds_one :target, as: :targetable, validate: false
+
+  embedded_in :addressable, polymorphic: true do
+    def extension
+      "Testing"
+    end
+    def doctor?
+      title == "Dr"
+    end
+  end
+
+  accepts_nested_attributes_for :code, :target
+  accepts_nested_attributes_for :locations, allow_destroy: true
+
+  belongs_to :account
+  belongs_to :band
+
+  scope :without_postcode, ->{ where(postcode: nil) }
+  scope :ordered, ->{ order_by(state: 1) }
+  scope :without_postcode_ordered, ->{ without_postcode.ordered }
+  scope :rodeo, ->{ where(street: "Rodeo Dr") } do
+    def mansion?
+      all? { |address| address.street == "Rodeo Dr" }
+    end
+  end
+
+  validates_presence_of :street, on: :update
+  validates_format_of :street, with: /\D/, allow_nil: true
+
+  def set_parent=(set = false)
+    self.parent_title = addressable.title if set
+  end
+
+  def <=>(other)
+    street <=> other.street
+  end
+
+  class << self
+    def california
+      where(state: "CA")
+    end
+
+    def homes
+      where(address_type: "Home")
+    end
+
+    def streets
+      all.map(&:street)
+    end
+
+    def city_and_state(city:, state:)
+      where(city: city, state: state)
+    end
+  end
+end

data/spec/support/models/address_component.rb

@@ -0,0 +1,7 @@
+# frozen_string_literal: true
+
+class AddressComponent
+  include Mongoid::Document
+  field :street, type: String
+  embedded_in :person
+end

data/spec/support/models/address_number.rb

@@ -0,0 +1,8 @@
+# frozen_string_literal: true
+
+class AddressNumber
+  include Mongoid::Document
+  field :country_code, type: Integer, default: 1
+  field :number
+  embedded_in :slave
+end

data/spec/support/models/agency.rb

@@ -0,0 +1,7 @@
+# frozen_string_literal: true
+
+class Agency
+  include Mongoid::Document
+  include Mongoid::Timestamps::Updated
+  has_many :agents, validate: false
+end

data/spec/support/models/agent.rb

@@ -0,0 +1,18 @@
+# frozen_string_literal: true
+
+class Agent
+  include Mongoid::Document
+  include Mongoid::Timestamps::Updated
+  field :title, type: String
+  field :number, type: String
+  field :dob, type: Time
+  embeds_many :names, as: :namable
+  embeds_one :address
+  belongs_to :game
+  belongs_to :agency, touch: true, autobuild: true
+
+  belongs_to :same_name, class_name: 'Band', inverse_of: :same_name
+
+  has_and_belongs_to_many :accounts
+  has_and_belongs_to_many :basics
+end

data/spec/support/models/album.rb

@@ -0,0 +1,20 @@
+# frozen_string_literal: true
+
+class Album
+  include Mongoid::Document
+
+  belongs_to :artist
+  before_destroy :set_parent_name
+
+  attr_accessor :before_add_called
+
+  private
+
+  def set_parent_name
+    artist.name = "destroyed" if artist
+  end
+
+  def set_parent_name_fail
+    throw(:abort)
+  end
+end

data/spec/support/models/alert.rb

@@ -0,0 +1,9 @@
+# frozen_string_literal: true
+
+class Alert
+  include Mongoid::Document
+  field :message, type: String
+  belongs_to :account
+  has_many :items
+  belongs_to :post
+end

data/spec/support/models/animal.rb

@@ -0,0 +1,28 @@
+# frozen_string_literal: true
+
+class Animal
+  include Mongoid::Document
+
+  field :_id, type: String, overwrite: true, default: ->{ name.try(:parameterize) }
+
+  field :name
+  field :height, type: Integer
+  field :weight, type: Integer
+  field :tags, type: Array
+
+  embedded_in :person
+  embedded_in :circus, class_name: 'Circus' # class_name is necessary because ActiveRecord think the singular of Circus
+                                            # is Circu
+
+  validates_format_of :name, without: /\$\$\$/
+
+  accepts_nested_attributes_for :person
+
+  def tag_list
+    tags.join(", ")
+  end
+
+  def tag_list=(_tag_list)
+    self.tags = _tag_list.split(",").map(&:strip)
+  end
+end

data/spec/support/models/answer.rb

@@ -0,0 +1,8 @@
+# frozen_string_literal: true
+
+class Answer
+  include Mongoid::Document
+  embedded_in :question
+
+  field :position, type: Integer
+end

data/spec/support/models/appointment.rb

@@ -0,0 +1,9 @@
+# frozen_string_literal: true
+
+class Appointment
+  include Mongoid::Document
+  field :active, type: Mongoid::Boolean, default: true
+  field :timed, type: Mongoid::Boolean, default: true
+  embedded_in :person
+  default_scope ->{ where(active: true) }
+end

data/spec/support/models/armrest.rb

@@ -0,0 +1,9 @@
+# frozen_string_literal: true
+
+class Armrest
+  include Mongoid::Document
+
+  embedded_in :seat
+
+  field :side
+end

data/spec/support/models/array_field.rb

@@ -0,0 +1,7 @@
+# frozen_string_literal: true
+
+class ArrayField
+  include Mongoid::Document
+
+  field :af, type: Array
+end

data/spec/support/models/article.rb

@@ -0,0 +1,14 @@
+# frozen_string_literal: true
+
+class Article
+  include Mongoid::Document
+
+  field :author_id, type: Integer
+  field :public, type: Mongoid::Boolean
+  field :title, type: String
+  field :is_rss, type: Mongoid::Boolean, default: false
+  field :user_login, type: String
+
+  has_and_belongs_to_many :tags, validate: false
+  has_and_belongs_to_many :preferences, inverse_of: nil, validate: false
+end

data/spec/support/models/artist.rb

@@ -0,0 +1,90 @@
+# frozen_string_literal: true
+
+class Artist
+  include Mongoid::Document
+
+  attr_accessor :before_add_called, :after_add_called, :before_add_referenced_called, :after_add_referenced_called, :before_remove_embedded_called, :after_remove_embedded_called, :before_remove_referenced_called, :after_remove_referenced_called
+
+  field :name, type: String
+
+  embeds_many :songs, before_add: [ :before_add_song, Proc.new { |artist, song| song.before_add_called = true } ], before_remove: :before_remove_song
+  embeds_many :labels, after_add: :after_add_label, after_remove: :after_remove_label
+  has_many :albums, dependent: :destroy, before_add: [:before_add_album, Proc.new { |artist, album| album.before_add_called = true} ], after_add: :after_add_album, before_remove: :before_remove_album, after_remove: :after_remove_album
+  belongs_to :band
+
+  before_create :before_create_stub
+  after_create :create_songs
+  before_save :before_save_stub
+  before_destroy :before_destroy_stub
+  before_update :before_update_stub
+
+  protected
+
+  def before_create_stub
+    true
+  end
+
+  def before_update_stub
+    true
+  end
+
+  def before_update_fail_stub
+    throw(:abort)
+  end
+
+  def before_save_fail_stub
+    throw(:abort)
+  end
+
+  def before_create_fail_stub
+    throw(:abort)
+  end
+
+  def before_save_stub
+    true
+  end
+
+  def before_destroy_fail_stub
+    throw(:abort)
+  end
+
+  def before_destroy_stub
+    true
+  end
+
+  def create_songs
+    2.times { |n| songs.create!(title: "#{n}") }
+  end
+
+  def before_add_song(song)
+    @before_add_called = true
+  end
+
+  def after_add_label(label)
+    @after_add_called = true
+  end
+
+  def before_add_album(album)
+    @before_add_referenced_called = true
+  end
+
+  def after_add_album(album)
+    @after_add_referenced_called = true
+  end
+
+  def before_remove_song(song)
+    @before_remove_embedded_called = true
+  end
+
+  def after_remove_label(label)
+    @after_remove_embedded_called = true
+  end
+
+  def before_remove_album(album)
+    @before_remove_referenced_called = true
+  end
+
+  def after_remove_album(album)
+    @after_remove_referenced_called = true
+  end
+end

data/spec/support/models/artwork.rb

@@ -0,0 +1,6 @@
+# frozen_string_literal: true
+
+class Artwork
+  include Mongoid::Document
+  has_and_belongs_to_many :exhibitors
+end

data/spec/support/models/audible_sound.rb

@@ -0,0 +1,3 @@
+# frozen_string_literal: true
+
+class AudibleSound < Sound; end

data/spec/support/models/audio.rb

@@ -0,0 +1,7 @@
+# frozen_string_literal: true
+
+class Audio
+  include Mongoid::Document
+  field :likes, type: Integer
+  default_scope ->{ self.or({:likes => nil}, {:likes.gt => 100}) }
+end

data/spec/support/models/augmentation.rb

@@ -0,0 +1,25 @@
+# frozen_string_literal: true
+
+class Augmentation
+  include Mongoid::Document
+
+  field :name
+
+  embedded_in :player, inverse_of: :augmentation
+
+  after_build do
+    self.name = "Infolink (#{player.frags})"
+  end
+
+  field :after_find_player
+  field :after_initialize_player
+  field :after_default_player, default: ->{ self.player&._id }
+
+  after_find do |doc|
+    doc.after_find_player = player&._id
+  end
+
+  after_initialize do |doc|
+    doc.after_initialize_player = player&._id
+  end
+end

data/spec/support/models/author.rb

@@ -0,0 +1,8 @@
+# frozen_string_literal: true
+
+class Author
+  include Mongoid::Document
+  field :id, type: Integer
+  field :author, type: Mongoid::Boolean
+  field :name, type: String
+end

data/spec/support/models/baby.rb

@@ -0,0 +1,6 @@
+# frozen_string_literal: true
+
+class Baby
+  include Mongoid::Document
+  embedded_in :kangaroo
+end

data/spec/support/models/band.rb

@@ -0,0 +1,45 @@
+# frozen_string_literal: true
+
+class Band
+  include Mongoid::Document
+  include Mongoid::Attributes::Dynamic
+
+  field :name, type: String
+  field :active, type: Mongoid::Boolean, default: true
+  field :origin, type: String
+  field :genres, type: Array
+  field :member_count, type: Integer
+  field :mems, as: :members, type: Array
+  field :likes, type: Integer
+  field :views, type: Integer
+  field :rating, type: Float
+  field :upserted, type: Mongoid::Boolean, default: false
+  field :created, type: DateTime
+  field :updated, type: Time
+  field :sales, type: BigDecimal
+  field :decimal, type: BSON::Decimal128
+  field :y, as: :years, type: Integer
+  field :founded, type: Date
+  field :decibels, type: Range
+  field :deleted, type: Boolean
+  field :mojo, type: Object
+  field :tags, type: Hash
+  field :fans
+
+  alias_attribute :d, :deleted
+
+  embeds_many :records, cascade_callbacks: true
+  embeds_many :notes, as: :noteable, cascade_callbacks: true, validate: false
+  embeds_many :labels
+  embeds_many :fanatics
+  embeds_one :label, cascade_callbacks: true
+
+  scope :highly_rated, -> { gte(rating: 7) }
+
+  has_many :artists
+  has_many :same_name, class_name: 'Agent', inverse_of: :same_name
+
+  after_upsert do |doc|
+    doc.upserted = true
+  end
+end

data/spec/support/models/bar.rb

@@ -0,0 +1,12 @@
+# frozen_string_literal: true
+
+class Bar
+  include Mongoid::Document
+  include Mongoid::Attributes::Dynamic
+  field :name, type: String
+  field :location, type: Array
+  field :lat_lng, type: LatLng
+
+  has_one :rating, as: :ratable
+  index location: "2d"
+end

data/spec/support/models/basic.rb

@@ -0,0 +1,8 @@
+# frozen_string_literal: true
+
+class Basic
+  include Mongoid::Document
+end
+
+class SubBasic < Basic
+end