Skip to content

Commit

Permalink
Merge pull request #214 from Aiven-Open/aris-tests-reexport
Browse files Browse the repository at this point in the history
tests: Fix re-exported utils module error
  • Loading branch information
kmichel-aiven authored Apr 16, 2024
2 parents 912dd4c + ec4d908 commit b9581ab
Showing 1 changed file with 2 additions and 7 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@
See LICENSE for details
"""

from astacus.common import ipc
from astacus.common import ipc, utils
from astacus.common.cassandra.schema import CassandraSchema
from astacus.coordinator.cluster import Cluster
from astacus.coordinator.config import CoordinatorNode
Expand Down Expand Up @@ -166,12 +166,7 @@ async def get_schema_hash(cluster, nodes):
return get_schema_steps.pop(0), "unused-error"

mocker.patch.object(restore_steps, "get_schema_hash", new=get_schema_hash)

mocker.patch.object(
restore_steps.utils, # type: ignore[attr-defined]
"exponential_backoff",
return_value=AsyncIterableWrapper(steps),
)
mocker.patch.object(utils, "exponential_backoff", return_value=AsyncIterableWrapper(steps))

step = restore_steps.WaitCassandraUpStep(duration=123, replaced_node_step=None)
context = base.StepsContext()
Expand Down

0 comments on commit b9581ab

Please sign in to comment.