feat(linter): eslint-plugin-jest: prefer-called-with (#2163)

Rule Detail:
[link](https://github.com/jest-community/eslint-plugin-jest/blob/main/src/rules/prefer-called-with.ts)
This commit is contained in:
cin 2024-01-25 14:01:51 +08:00 committed by GitHub
parent 80a454630a
commit ac1d318dc3
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 157 additions and 0 deletions

View file

@ -139,6 +139,7 @@ mod jest {
pub mod no_standalone_expect;
pub mod no_test_prefixes;
pub mod no_test_return_statement;
pub mod prefer_called_with;
pub mod prefer_todo;
pub mod valid_describe_callback;
pub mod valid_expect;
@ -422,6 +423,7 @@ oxc_macros::declare_all_lint_rules! {
jest::no_standalone_expect,
jest::no_test_prefixes,
jest::no_test_return_statement,
jest::prefer_called_with,
jest::prefer_todo,
jest::valid_describe_callback,
jest::valid_expect,

View file

@ -0,0 +1,128 @@
use crate::{
context::LintContext,
rule::Rule,
utils::{collect_possible_jest_call_node, parse_expect_jest_fn_call, PossibleJestNode},
};
use oxc_ast::AstKind;
use oxc_diagnostics::{
miette::{self, Diagnostic},
thiserror::Error,
};
use oxc_macros::declare_oxc_lint;
use oxc_span::Span;
#[derive(Debug, Error, Diagnostic)]
enum PreferCalledWithDiagnostic {
#[error("eslint-plugin-jest(prefer-called-with): Suggest using `toBeCalledWith()` or `toHaveBeenCalledWith()`.")]
#[diagnostic(severity(warning), help("Prefer toBeCalledWith(/* expected args */)"))]
UseToBeCalledWith(#[label] Span),
#[error("eslint-plugin-jest(prefer-called-with): Suggest using `toBeCalledWith()` or `toHaveBeenCalledWith()`.")]
#[diagnostic(severity(warning), help("Prefer toHaveBeenCalledWith(/* expected args */)"))]
UseHaveBeenCalledWith(#[label] Span),
}
#[derive(Debug, Default, Clone)]
pub struct PreferCalledWith;
declare_oxc_lint!(
/// ### What it does
/// Suggest using `toBeCalledWith()` or `toHaveBeenCalledWith()`
///
/// ### Example
///
/// ```javascript
///
/// // valid
/// expect(noArgsFunction).toBeCalledWith();
/// expect(roughArgsFunction).toBeCalledWith(expect.anything(), expect.any(Date));
/// expect(anyArgsFunction).toBeCalledTimes(1);
/// expect(uncalledFunction).not.toBeCalled();
///
/// // invalid
/// expect(someFunction).toBeCalled();
/// expect(someFunction).toHaveBeenCalled();
/// ```
///
PreferCalledWith,
style,
);
impl Rule for PreferCalledWith {
fn run_once(&self, ctx: &LintContext<'_>) {
for possible_jest_node in &collect_possible_jest_call_node(ctx) {
Self::run(possible_jest_node, ctx);
}
}
}
impl PreferCalledWith {
pub fn run<'a>(possible_jest_node: &PossibleJestNode<'a, '_>, ctx: &LintContext<'a>) {
let node = possible_jest_node.node;
let AstKind::CallExpression(call_expr) = node.kind() else {
return;
};
let Some(jest_fn_call) = parse_expect_jest_fn_call(call_expr, possible_jest_node, ctx)
else {
return;
};
let mut has_not_modifier = false;
for modifier in jest_fn_call.modifiers() {
if let Some(modifier_name) = modifier.name() {
if modifier_name.eq("not") {
has_not_modifier = true;
break;
}
}
}
if has_not_modifier {
return;
}
if let Some(matcher_property) = jest_fn_call.matcher() {
if let Some(matcher_name) = matcher_property.name() {
if matcher_name == "toBeCalled" {
ctx.diagnostic(PreferCalledWithDiagnostic::UseToBeCalledWith(
matcher_property.span,
));
} else if matcher_name == "toHaveBeenCalled" {
ctx.diagnostic(PreferCalledWithDiagnostic::UseHaveBeenCalledWith(
matcher_property.span,
));
}
}
}
}
}
#[test]
fn test() {
use crate::tester::Tester;
let pass = vec![
("expect(fn).toBeCalledWith();", None),
("expect(fn).toHaveBeenCalledWith();", None),
("expect(fn).toBeCalledWith(expect.anything());", None),
("expect(fn).toHaveBeenCalledWith(expect.anything());", None),
("expect(fn).not.toBeCalled();", None),
("expect(fn).rejects.not.toBeCalled();", None),
("expect(fn).not.toHaveBeenCalled();", None),
("expect(fn).not.toBeCalledWith();", None),
("expect(fn).not.toHaveBeenCalledWith();", None),
("expect(fn).resolves.not.toHaveBeenCalledWith();", None),
("expect(fn).toBeCalledTimes(0);", None),
("expect(fn).toHaveBeenCalledTimes(0);", None),
("expect(fn);", None),
];
let fail = vec![
("expect(fn).toBeCalled();", None),
("expect(fn).resolves.toBeCalled();", None),
("expect(fn).toHaveBeenCalled();", None),
];
Tester::new(PreferCalledWith::NAME, pass, fail).with_jest_plugin(true).test_and_snapshot();
}

View file

@ -0,0 +1,27 @@
---
source: crates/oxc_linter/src/tester.rs
assertion_line: 150
expression: prefer_called_with
---
⚠ eslint-plugin-jest(prefer-called-with): Suggest using `toBeCalledWith()` or `toHaveBeenCalledWith()`.
╭─[prefer_called_with.tsx:1:1]
1 │ expect(fn).toBeCalled();
· ──────────
╰────
help: Prefer toBeCalledWith(/* expected args */)
⚠ eslint-plugin-jest(prefer-called-with): Suggest using `toBeCalledWith()` or `toHaveBeenCalledWith()`.
╭─[prefer_called_with.tsx:1:1]
1 │ expect(fn).resolves.toBeCalled();
· ──────────
╰────
help: Prefer toBeCalledWith(/* expected args */)
⚠ eslint-plugin-jest(prefer-called-with): Suggest using `toBeCalledWith()` or `toHaveBeenCalledWith()`.
╭─[prefer_called_with.tsx:1:1]
1 │ expect(fn).toHaveBeenCalled();
· ────────────────
╰────
help: Prefer toHaveBeenCalledWith(/* expected args */)