use clippy_utils::diagnostics::span_lint_and_help;
+use clippy_utils::is_in_test_function;
use clippy_utils::ty::is_type_diagnostic_item;
use rustc_hir as hir;
use rustc_lint::LateContext;
use super::EXPECT_USED;
/// lint use of `expect()` for `Option`s and `Result`s
-pub(super) fn check(cx: &LateContext<'_>, expr: &hir::Expr<'_>, recv: &hir::Expr<'_>) {
+pub(super) fn check(cx: &LateContext<'_>, expr: &hir::Expr<'_>, recv: &hir::Expr<'_>, allow_expect_in_tests: bool) {
let obj_ty = cx.typeck_results().expr_ty(recv).peel_refs();
let mess = if is_type_diagnostic_item(cx, obj_ty, sym::Option) {
None
};
+ if allow_expect_in_tests && is_in_test_function(cx.tcx, expr.hir_id) {
+ return;
+ }
+
if let Some((lint, kind, none_value)) = mess {
span_lint_and_help(
cx,