Skip to content

Commit

Permalink
Fix parser test monkeypatch for yaml
Browse files Browse the repository at this point in the history
  • Loading branch information
VladimirFilonov committed Jan 19, 2025
1 parent 5ba6b47 commit 9d46903
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions tests/test_parser.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,6 +13,7 @@
from keep.api.core.dependencies import SINGLE_TENANT_UUID
from keep.api.models.db.action import Action
from keep.contextmanager.contextmanager import ContextManager
from keep.functions import cyaml
from keep.parser.parser import Parser, ParserUtils
from keep.providers.mock_provider.mock_provider import MockProvider
from keep.providers.models.provider_config import ProviderConfig
Expand Down Expand Up @@ -203,7 +204,7 @@ def mock_safeload(*args, **kwargs):
monkeypatch.setattr(
builtins, "open", mocker.mock_open(read_data="does not matter")
)
monkeypatch.setattr(yaml, "safe_load", mock_safeload)
monkeypatch.setattr(cyaml, "safe_load", mock_safeload)

# ACT
parse_file_setup(context_manager)
Expand All @@ -223,7 +224,7 @@ def mock_safeload(*args, **kwargs):
monkeypatch.setattr(
builtins, "open", mocker.mock_open(read_data="does not matter")
)
monkeypatch.setattr(yaml, "safe_load", mock_safeload)
monkeypatch.setattr(cyaml, "safe_load", mock_safeload)

# ACT/ASSERT
with pytest.raises(yaml.YAMLError):
Expand Down

0 comments on commit 9d46903

Please sign in to comment.