diff options
Diffstat (limited to 'testing/embedder_test_timer_handling_delegate.h')
-rw-r--r-- | testing/embedder_test_timer_handling_delegate.h | 11 |
1 files changed, 8 insertions, 3 deletions
diff --git a/testing/embedder_test_timer_handling_delegate.h b/testing/embedder_test_timer_handling_delegate.h index 709dd22c17..31966159eb 100644 --- a/testing/embedder_test_timer_handling_delegate.h +++ b/testing/embedder_test_timer_handling_delegate.h @@ -38,9 +38,11 @@ class EmbedderTestTimerHandlingDelegate : public EmbedderTest::Delegate { } int SetTimer(int msecs, TimerCallback fn) override { - expiry_to_timer_map_.insert(std::pair<int, Timer>( - msecs + fake_elapsed_msecs_, {++next_timer_id_, msecs, fn})); - return next_timer_id_; + int id = fail_next_timer_ ? 0 : ++next_timer_id_; + expiry_to_timer_map_.insert( + std::pair<int, Timer>(msecs + fake_elapsed_msecs_, {id, msecs, fn})); + fail_next_timer_ = false; + return id; } void KillTimer(int id) override { @@ -73,8 +75,11 @@ class EmbedderTestTimerHandlingDelegate : public EmbedderTest::Delegate { const std::vector<AlertRecord>& GetAlerts() const { return alerts_; } + void SetFailNextTimer() { fail_next_timer_ = true; } + protected: std::multimap<int, Timer> expiry_to_timer_map_; // Keyed by timeout. + bool fail_next_timer_ = false; int next_timer_id_ = 0; int fake_elapsed_msecs_ = 0; std::vector<AlertRecord> alerts_; |