Skip to content

Commit a020996

Browse files
committed
Improving coverage.
1 parent 523ce17 commit a020996

File tree

1 file changed

+3
-9
lines changed

1 file changed

+3
-9
lines changed

mocket/mocket.py

Lines changed: 3 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -36,9 +36,7 @@ def enable(
3636
if truesocket_recording_dir is not None:
3737
recording_dir = Path(truesocket_recording_dir)
3838

39-
if not recording_dir.is_dir():
40-
# JSON dumps will be saved here
41-
raise AssertionError
39+
assert recording_dir.is_dir(), f"Not a directory: {recording_dir}"
4240

4341
cls._record_storage = MocketRecordStorage(
4442
directory=recording_dir,
@@ -118,15 +116,11 @@ def has_requests(cls) -> bool:
118116

119117
@classmethod
120118
def get_namespace(cls) -> str | None:
121-
if not cls._record_storage:
122-
return None
123-
return cls._record_storage.namespace
119+
return cls._record_storage.namespace if cls._record_storage else None
124120

125121
@classmethod
126122
def get_truesocket_recording_dir(cls) -> str | None:
127-
if not cls._record_storage:
128-
return None
129-
return str(cls._record_storage.directory)
123+
return str(cls._record_storage.directory) if cls._record_storage else None
130124

131125
@classmethod
132126
def assert_fail_if_entries_not_served(cls) -> None:

0 commit comments

Comments
 (0)