summaryrefslogtreecommitdiffstats
path: root/pre_commit_hooks/tests_should_end_in_test.py
diff options
context:
space:
mode:
authorDaniel Baumann <daniel.baumann@progress-linux.org>2022-01-30 11:02:58 +0000
committerDaniel Baumann <daniel.baumann@progress-linux.org>2024-05-21 20:41:55 +0000
commitd4583dcad7d68d3c1503b04ec0d3364809304807 (patch)
tree82456c5d0bc77961759812ddd85414435ba89127 /pre_commit_hooks/tests_should_end_in_test.py
parentInitial commit. (diff)
downloadpre-commit-hooks-upstream/4.5.0+dfsg.tar.xz
pre-commit-hooks-upstream/4.5.0+dfsg.zip
Adding upstream version 4.5.0+dfsg.upstream/4.5.0+dfsgupstream
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'pre_commit_hooks/tests_should_end_in_test.py')
-rw-r--r--pre_commit_hooks/tests_should_end_in_test.py53
1 files changed, 53 insertions, 0 deletions
diff --git a/pre_commit_hooks/tests_should_end_in_test.py b/pre_commit_hooks/tests_should_end_in_test.py
new file mode 100644
index 0000000..e7842af
--- /dev/null
+++ b/pre_commit_hooks/tests_should_end_in_test.py
@@ -0,0 +1,53 @@
+from __future__ import annotations
+
+import argparse
+import os.path
+import re
+from typing import Sequence
+
+
+def main(argv: Sequence[str] | None = None) -> int:
+ parser = argparse.ArgumentParser()
+ parser.add_argument('filenames', nargs='*')
+ mutex = parser.add_mutually_exclusive_group()
+ mutex.add_argument(
+ '--pytest',
+ dest='pattern',
+ action='store_const',
+ const=r'.*_test\.py',
+ default=r'.*_test\.py',
+ help='(the default) ensure tests match %(const)s',
+ )
+ mutex.add_argument(
+ '--pytest-test-first',
+ dest='pattern',
+ action='store_const',
+ const=r'test_.*\.py',
+ help='ensure tests match %(const)s',
+ )
+ mutex.add_argument(
+ '--django', '--unittest',
+ dest='pattern',
+ action='store_const',
+ const=r'test.*\.py',
+ help='ensure tests match %(const)s',
+ )
+ args = parser.parse_args(argv)
+
+ retcode = 0
+ reg = re.compile(args.pattern)
+ for filename in args.filenames:
+ base = os.path.basename(filename)
+ if (
+ not reg.fullmatch(base) and
+ not base == '__init__.py' and
+ not base == 'conftest.py'
+ ):
+ retcode = 1
+ print(f'{filename} does not match pattern "{args.pattern}"')
+
+ return retcode
+
+
+if __name__ == '__main__':
+ raise SystemExit(main())