rspec model test failing

Question

Я пишу приложение, чтобы помочь мне отслеживать мои бюджеты на рекламу в социальных сетях. Когда вы вводите новое объявление, оно должно рассчитывать и обновлять сумму, потраченную на бюджет, из которого он берется. Вот моя модель, которая достигает этого.

class Advert < ActiveRecord::Base 
    belongs_to :budget 
    before_save :update_budget 
    after_destroy :update_budget 

    validates :budget_id, :name, :platform, :ad_type, :amount, :start_date, :end_date, presence: true 
    validates :amount, numericality: true 
    validate :check_budget 

     # Checks to see if there is enough budget remaining to set up advert 
     def check_budget 
     if self.amount > self.budget.amount_remaining 
     errors.add(:amount, " cannot exceed amount remaining in budget.") 
     end 
    end 



    # Updates the amount remaining in the budget on advert save. 
    def update_budget 
     budget = Budget.find(self.budget_id) 
     @adverts = Advert.all 
     total_spent = self.amount 
     @adverts.each do |advert| 
      if advert.budget_id == self.budget_id 
       total_spent += advert.amount 
      end 
     end 
     budget.amount_spent = total_spent 
     budget.save 
    end 
end 

Это все работает, но я в настоящее время преподаю себе писать тесты, так я думал, что написать тест в RSpec для него.

require 'rails_helper' 

describe Advert do 
    it "updates budget before save" do 
     advert = create(:advert) 
     budget = advert.budget 

     expect(budget.amount_spent).to eq(advert.amount) 
     expect(budget.amount_remaining).to eq(budget.amount - budget.amount_spent) 
    end 

end 

Однако этот тест, если не удается, но я не могу понять, почему. Вот код ошибки.

1) Advert updates budget before save 
    Failure/Error: expect(budget.amount_spent).to eq(advert.amount) 

     expected: 7.0 (#<BigDecimal:7ffa61358b18,'0.7E1',9(18)>) 
      got: 0.0 (#<BigDecimal:7ffa6026a9a0,'0.0',9(18)>) 

     (compared using ==) 
    # ./spec/models/advert_spec.rb:27:in `block (2 levels) in <top (required)>' 

И вот соответствующий протокол испытаний.

SQL (0.3ms) INSERT INTO "budgets" ("name", "amount", "client_id", "amount_remaining", "created_at", "updated_at") VALUES ($1, $2, $3, $4, $5, $6) RETURNING "id" [["name", "eos"], ["amount", "432.0"], ["client_id", 102], ["amount_remaining", "432.0"], ["created_at", "2016-03-12 18:08:54.607999"], ["updated_at", "2016-03-12 18:08:54.607999"]] 
    (0.1ms) RELEASE SAVEPOINT active_record_1 
    (0.2ms) SAVEPOINT active_record_1 
    Budget Load (0.4ms) SELECT "budgets".* FROM "budgets" WHERE "budgets"."id" = $1 LIMIT 1 [["id", 49]] 
    Advert Load (0.5ms) SELECT "adverts".* FROM "adverts" 
    SQL (0.4ms) UPDATE "budgets" SET "amount_spent" = $1, "amount_remaining" = $2, "updated_at" = $3 WHERE "budgets"."id" = $4 [["amount_spent", "7.0"], ["amount_remaining", "425.0"], ["updated_at", "2016-03-12 18:08:54.616491"], ["id", 49]] 
    SQL (0.4ms) INSERT INTO "adverts" ("budget_id", "name", "platform", "ad_type", "amount", "start_date", "end_date", "created_at", "updated_at") VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9) RETURNING "id" [["budget_id", 49], ["name", "ut"], ["platform", "voluptate"], ["ad_type", "facere"], ["amount", "7.0"], ["start_date", "2016-03-01"], ["end_date", "2016-04-12"], ["created_at", "2016-03-12 18:08:54.619698"], ["updated_at", "2016-03-12 18:08:54.619698"]] 
    (0.2ms) RELEASE SAVEPOINT active_record_1 
    (0.2ms) ROLLBACK 

Интересно, если я прокомментирую первый «ожидание» теста. Это похоже на то, что он не может получить доступ к advert.amount, поэтому установите его как 0.

У кого-нибудь есть идеи?

Answer 1

Это решило мою проблему.

describe Advert do 
    it "updates budget before save" do 
     advert = build(:advert) 
     budget = advert.budget 

     expect(budget.amount_spent).to eq(0) 

     advert.save 
     budget.reload 

     expect(budget.amount_spent).to eq(advert.amount) 
     expect(budget.amount_remaining).to eq(budget.amount - budget.amount_spent) 
end 

Я думаю, источник моей проблемы не было перегрузки моего бюджета это означало, что я пытался получить доступ к атрибуту, прежде чем он был обновлен.