如何在RSpec测试中捕获PG::NotNullViolation非空约束错误

问题场景 #

使用RSpec编写测试时，存在如下参数处理方法：

def update_section_params
  params.require(:section).permit(:type_id, :by_team)
end

调用@section.update(update_section_params)更新Section时，若传入by_team: nil，会触发PostgreSQL非空约束错误：

PG::NotNullViolation: ERROR:  null value in column 

当前测试代码如下：

payload = { by_team: nil }
put ".../.../sections/#{@section1['id']}", params: payload

解决方案 #

1. 模型层测试：直接捕获数据库异常 #

如果是在模型测试中直接调用update方法，可以通过raise_error匹配器捕获PG::NotNullViolation异常：

it "抛出非空约束错误当by_team为nil时" do
  expect { @section.update(by_team: nil) }.to raise_error(PG::NotNullViolation)
end

2. 请求/控制器层测试：验证响应状态或内容 #

在请求测试中，Rails会自动捕获数据库异常并返回对应HTTP响应，此时可以验证响应状态或错误信息：

注意：你的测试参数结构存在问题，update_section_params要求参数嵌套在:section键下，否则会先触发ActionController::ParameterMissing错误，需要先修正payload结构：

it "返回错误响应当by_team为nil时" do
  payload = { section: { by_team: nil } }
  put section_path(@section1), params: payload

  # 验证响应状态码，默认是500，若有自定义错误处理则调整
  expect(response).to have_http_status(:internal_server_error)
  
  # 若应用返回JSON格式错误信息，可进一步验证
  # expect(JSON.parse(response.body)["message"]).to include("null value in column")
end

额外优化建议 #

比起捕获数据库底层异常，更推荐在模型层添加业务验证，提前拦截无效数据：

# Section模型代码
class Section < ApplicationRecord
  validates :by_team, presence: true
end

对应的测试可以更友好地验证错误信息：

it "不允许by_team为nil" do
  @section.update(by_team: nil)
  expect(@section.errors[:by_team]).to include("can't be blank")
end

内容的提问来源于stack exchange，提问作者tunosi st