[Testing] [Bug]: Wait timing optimization

[dhegberg]

Migration from aws/aws-durable-execution-sdk-python-testing#183

---

Expected Behavior

const resultPromise = context.step(async () => {
    await new Promise((resolve) => setTimeout(resolve, 7000));
    return "step completed";
});
await context.wait("wait-1", { seconds: 10 });
const result = await resultPromise; // Re-invokes after 10s, rather than 3s

For this example real backend reinvokes after 3s.

Actual Behavior

The function is re-invoked after the full wait duration of 10s instead of the remaining time (3s).

Steps to Reproduce

See above

SDK Version

Python Version

3.13

Is this a regression?

No

Last Working Version

No response

Additional Context

No response