1
0
Fork 0
mirror of https://github.com/thoughtbot/factory_bot.git synced 2022-11-09 11:43:51 -05:00
thoughtbot--factory_bot/spec/acceptance/transient_attributes_spec.rb
Josh Clayton and Jason Draper bef5a01b31 Introduce new callback syntax
Instead of calling before_create, after_build, after_create, or
after_stub, you can now call:

    before(:create) {|instance| instance.name = "overridden!" }
    after(:create)  {|instance| instance.name = "overridden!" }
    after(:build)   {|instance| instance.name = "overridden!" }
    after(:stub)    {|instance| instance.name = "overridden!" }

Additionally, you can declare callbacks longhand:

    callback(:after_stub) {|instance| instance.name = "overridden!" }

This allows for custom callbacks to be defined:

    callback(:custom_callback) {|instance| instance.name = "overridden!" }

Which can then be used from a custom strategy:

    class CustomStrategy
      def association(runner); end

      def result(evaluation)
        evaluation.object.tap do |instance|
          evaluation.notify(:custom_callback, instance)
        end
      end
    end

    FactoryGirl.register_strategy(:custom, CustomStrategy)

This would allow for calling:

    FactoryGirl.custom(:user)

Which would return the user instance but execute the :custom_callback callback
on the user instance first.
2012-05-04 17:59:34 -04:00

124 lines
3.4 KiB
Ruby

require "spec_helper"
describe "transient attributes" do
before do
define_model("User", name: :string, email: :string)
FactoryGirl.define do
sequence(:name) {|n| "John #{n}" }
factory :user do
ignore do
four { 2 + 2 }
rockstar true
upcased false
end
name { "#{FactoryGirl.generate(:name)}#{" - Rockstar" if rockstar}" }
email { "#{name.downcase}#{four}@example.com" }
after(:create) do |user, evaluator|
user.name.upcase! if evaluator.upcased
end
end
end
end
context "returning attributes for a factory" do
subject { FactoryGirl.attributes_for(:user, rockstar: true) }
it { should_not have_key(:four) }
it { should_not have_key(:rockstar) }
it { should_not have_key(:upcased) }
it { should have_key(:name) }
it { should have_key(:email) }
end
context "with a transient variable assigned" do
let(:rockstar) { FactoryGirl.create(:user, rockstar: true, four: "1234") }
let(:rockstar_with_name) { FactoryGirl.create(:user, name: "Jane Doe", rockstar: true) }
let(:upcased_rockstar) { FactoryGirl.create(:user, rockstar: true, upcased: true) }
let(:groupie) { FactoryGirl.create(:user, rockstar: false) }
it "generates the correct attributes on a rockstar" do
rockstar.name.should == "John 1 - Rockstar"
rockstar.email.should == "john 1 - rockstar1234@example.com"
end
it "generates the correct attributes on an upcased rockstar" do
upcased_rockstar.name.should == "JOHN 1 - ROCKSTAR"
upcased_rockstar.email.should == "john 1 - rockstar4@example.com"
end
it "generates the correct attributes on a groupie" do
groupie.name.should == "John 1"
groupie.email.should == "john 14@example.com"
end
it "generates the correct attributes on a rockstar with a name" do
rockstar_with_name.name.should == "Jane Doe"
rockstar_with_name.email.should == "jane doe4@example.com"
end
end
context "without transient variables assigned" do
let(:rockstar) { FactoryGirl.create(:user) }
it "uses the default value of the attribute" do
rockstar.name.should == "John 1 - Rockstar"
end
end
end
describe "transient sequences" do
before do
define_model("User", name: :string)
FactoryGirl.define do
factory :user do
ignore do
sequence(:counter)
end
name { "John Doe #{counter}" }
end
end
end
it "increments sequences correctly" do
FactoryGirl.build(:user).name.should == "John Doe 1"
FactoryGirl.build(:user).name.should == "John Doe 2"
end
end
describe "assigning values from a transient attribute" do
before do
define_class("User") do
attr_accessor :foo_id, :foo_name
end
define_class("Foo") do
attr_accessor :id, :name
def initialize(id, name)
@id = id
@name = name
end
end
FactoryGirl.define do
factory :user do
ignore do
foo { Foo.new('id-of-foo', 'name-of-foo')}
end
foo_id { foo.id }
foo_name { foo.name }
end
end
end
it "does not ignore an _id attribute that is an alias for a transient attribute" do
user = FactoryGirl.build(:user, :foo => Foo.new('passed-in-id-of-foo', 'passed-in-name-of-foo'))
user.foo_id.should == 'passed-in-id-of-foo'
user.foo_name.should == 'passed-in-name-of-foo'
end
end