Skip to content

Commit bacfa9c

Browse files
committed
fix: mock_get_staking_module
1 parent b58e727 commit bacfa9c

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tests/modules/csm/test_csm_distribution.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66

77
from src.constants import UINT64_MAX
88
from src.modules.csm.csm import CSOracle, CSMError
9-
from src.modules.csm.log import ValidatorFrameSummary, OperatorFrameSummary
9+
from src.modules.csm.log import ValidatorFrameSummary
1010
from src.modules.csm.state import AttestationsAccumulator, State
1111
from src.types import NodeOperatorId, ValidatorIndex
1212
from src.web3py.extensions import CSM
@@ -15,8 +15,8 @@
1515

1616

1717
@pytest.fixture(autouse=True)
18-
def mock_get_module_id(monkeypatch: pytest.MonkeyPatch):
19-
monkeypatch.setattr(CSOracle, "_get_module_id", Mock())
18+
def mock_get_staking_module(monkeypatch: pytest.MonkeyPatch):
19+
monkeypatch.setattr(CSOracle, "_get_staking_module", Mock())
2020

2121

2222
@pytest.fixture()

0 commit comments

Comments
 (0)