Skip to content

Commit

Permalink
Update test_client.py
Browse files Browse the repository at this point in the history
  • Loading branch information
oiadebayo committed Oct 25, 2024
1 parent 53bab4f commit 0ab6a9a
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions integrations/snyk/tests/snyk/test_client.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
import pytest
from unittest.mock import patch, AsyncMock, MagicMock
from typing import Any, Dict, List
from typing import Any, Dict, List, Generator
from snyk.client import SnykClient
from port_ocean.exceptions.context import PortOceanContextAlreadyInitializedError
from port_ocean.context.ocean import initialize_port_ocean_context
Expand Down Expand Up @@ -35,7 +35,7 @@ def mock_ocean_context() -> None:


@pytest.fixture
def mock_event_context() -> MagicMock:
def mock_event_context() -> Generator[MagicMock, None, None]:
"""Fixture to mock the event context."""
mock_event = MagicMock(spec=EventContext)
mock_event.event_type = "test_event"
Expand Down

0 comments on commit 0ab6a9a

Please sign in to comment.