自定义验证检查

Question

我有以下休假模型。我要做的是根据休假类型、休假次数和休假分配来检查可用的休假数。似乎我的“check_leave”由于某种原因没有得到验证而没有任何错误。感谢任何帮助！

class Leave < ActiveRecord::Base
  belongs_to :staff
  validates :staff, :leave_type, :start_date, :end_date, :number_of_days, :approved_by, presence: true
  enum leave_type: {Medical: 0, Annual: 1, Urgent: 3, "Birth Leave": 4}
  validate :check_leave, if: -> {self.number_of_days.present?}

protected

  def check_leave
    if self.leave_type == 0
      if ( self.number_of_days = self.number_of_days + LeaveAllocation.last.medical_leave_counter ) > LeaveAllocation.last.medical_leave
        self.errors.add(:number_of_days, "Days exceeded the limit")
      end
    elsif self.leave_type == 1
      if ( self.number_of_days = self.number_of_days + LeaveAllocation.last.annual_leave_counter ) > LeaveAllocation.last.annual_leave
        self.errors.add(:number_of_days, "Days exceeded the limit")
      end
    end
  end
end

Answer 1

在我看来，您只需要在您的条件（两者）中删除 self.number_of_days 的分配。如果我了解您要查找的内容，则应该这样做：

def check_leave
  if self.leave_type == 0
    if (self.number_of_days + LeaveAllocation.last.medical_leave_counter) > LeaveAllocation.last.medical_leave
      errors.add(:number_of_days, "Days exceeded the limit")
    end
  elsif self.leave_type == 1
    if (self.number_of_days + LeaveAllocation.last.annual_leave_counter) > LeaveAllocation.last.annual_leave
      errors.add(:number_of_days, "Days exceeded the limit")
    end
  end
end

我还从errors.add 中删除了self.，因为没有理由拥有它。您可能可以在此方法中删除所有 self. 的使用，但我会留给您，以防有我不知道的冲突。

我创建了一个简短的测试程序，以使用我创建的一些代表性数据来测试条件。测试程序在这里：

test_data = [
  { leave_type: 0, number_of_days:  2, leave_counter: 2, leave_limit: 13, result: true },
  { leave_type: 0, number_of_days: 12, leave_counter: 2, leave_limit: 13, result: false },
  { leave_type: 0, number_of_days: 12, leave_counter: 2, leave_limit: 14, result: true },
  { leave_type: 1, number_of_days:  5, leave_counter: 1, leave_limit:  9, result: true },
  { leave_type: 1, number_of_days: 12, leave_counter: 2, leave_limit: 13, result: false },
  { leave_type: 1, number_of_days: 12, leave_counter: 2, leave_limit: 14, result: true },
]

def check_leave(leave_type, number_of_days, leave_counter, leave_limit)
  if leave_type == 0
    if (number_of_days + leave_counter) > leave_limit
      return false
    end
  elsif leave_type == 1
    if (number_of_days + leave_counter) > leave_limit
      return false
    end
  end

  return true
end

test_data.each_with_index do |test, i|
  puts "#{check_leave(test[:leave_type], test[:number_of_days], test[:leave_counter], test[:leave_limit]) == test[:result] ? 'Pass' : 'Fail'}: #{i}"
end

结果在这里：

Pass: 0
Pass: 1
Pass: 2
Pass: 3
Pass: 4
Pass: 5

这是 6 个特定的测试条件，它们分别为 2 个 leave_types 确认以下 3 个条件：

• (number_of_days + leave_counter)
• (number_of_days + leave_counter) > leave_limit（预计验证失败）
• (number_of_days + leave_counter) == leave_limit（期望验证通过）

这表明检查正在工作。验证失败几乎可以肯定是因为测试数据不正确。此验证中使用的 4 个字段中的任何一个都可能是错误的：

• leave_type
• number_of_days
• leave_counter
• leave_limit

剩下的唯一步骤是确认应用程序测试中使用的数据，以进行据报道无效的验证。至少需要一组测试数据（4 个必填字段的值）才能继续进行。

Answer 2

试试这个，因为 leave_type 是一个哈希，你需要提供正确的键来获取值...

改变

  if self.leave_type == 0
  ...
  elsif self.leave_type == 1
  ...
  end

到

  if self.leave_type[:Medical] == 0
  ...
  elsif self.leave_type[:Annual] == 1
  ...
  end