Skip to content

Commit bdb490c

Browse files
committed
Stub out methods used to check check messages
1 parent fb6f55b commit bdb490c

File tree

1 file changed

+21
-0
lines changed

1 file changed

+21
-0
lines changed

tests/unit/recorders.py

Lines changed: 21 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@
1818
import datetime
1919
import pathlib
2020
from collections.abc import Awaitable, Callable
21+
from typing import Any
2122

2223
import atr.models.sql as sql
2324
import atr.tasks.checks as checks
@@ -71,6 +72,26 @@ async def _add(
7172
inputs_hash=None,
7273
)
7374

75+
async def exception(
76+
self, message: str, data: Any, primary_rel_path: str | None = None, member_rel_path: str | None = None
77+
) -> sql.CheckResult:
78+
return await self._add(sql.CheckResultStatus.EXCEPTION, message, data, primary_rel_path, member_rel_path)
79+
80+
async def failure(
81+
self, message: str, data: Any, primary_rel_path: str | None = None, member_rel_path: str | None = None
82+
) -> sql.CheckResult:
83+
return await self._add(sql.CheckResultStatus.FAILURE, message, data, primary_rel_path, member_rel_path)
84+
85+
async def success(
86+
self, message: str, data: Any, primary_rel_path: str | None = None, member_rel_path: str | None = None
87+
) -> sql.CheckResult:
88+
return await self._add(sql.CheckResultStatus.SUCCESS, message, data, primary_rel_path, member_rel_path)
89+
90+
async def warning(
91+
self, message: str, data: Any, primary_rel_path: str | None = None, member_rel_path: str | None = None
92+
) -> sql.CheckResult:
93+
return await self._add(sql.CheckResultStatus.WARNING, message, data, primary_rel_path, member_rel_path)
94+
7495

7596
def get_recorder(recorder: checks.Recorder) -> Callable[[], Awaitable[checks.Recorder]]:
7697
async def _recorder() -> checks.Recorder:

0 commit comments

Comments
 (0)