Skip to content

Commit 2a4c95e

Browse files
committed
MK8S-184 - Add a mock for Salt cp.get_file_str
1 parent 52eb865 commit 2a4c95e

File tree

1 file changed

+11
-0
lines changed
  • salt/tests/unit/formulas/fixtures

1 file changed

+11
-0
lines changed

salt/tests/unit/formulas/fixtures/salt.py

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -412,6 +412,17 @@ def slsutil_renderer(salt_mock: SaltMock, source: str, **_kwargs: Any) -> Any:
412412
register_basic("file.find")(MagicMock(return_value=[]))
413413
register_basic("file.join")(lambda *args: "/".join(args))
414414
register_basic("file.read")(MagicMock(return_value="<file contents>"))
415+
416+
417+
@register_basic("cp.get_file_str")
418+
def cp_get_file_str(source: str, **_kwargs: Any) -> str:
419+
"""Read the relevant file directly from local sources."""
420+
assert source.startswith("salt://")
421+
path = paths.SALT_DIR / source[len("salt://") :]
422+
with path.open("r") as src_file:
423+
return src_file.read()
424+
425+
415426
register_basic("hashutil.base64_b64decode")(lambda input_data: input_data)
416427
register_basic("hashutil.base64_encodefile")(
417428
MagicMock(return_value="<b64-encoded data>")

0 commit comments

Comments
 (0)