blob: 8b9000ae08523c27efd4dcb2df3cef8193df29e3 [file] [log] [blame]
// Copyright 2015 PDFium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef TESTING_EMBEDDER_TEST_TIMER_HANDLING_DELEGATE_H_
#define TESTING_EMBEDDER_TEST_TIMER_HANDLING_DELEGATE_H_
#include <map>
#include <string>
#include <utility>
#include <vector>
#include "testing/embedder_test.h"
#include "testing/fx_string_testhelpers.h"
class EmbedderTestTimerHandlingDelegate final : public EmbedderTest::Delegate {
public:
struct AlertRecord {
std::wstring message;
std::wstring title;
int type;
int icon;
};
struct Timer {
int id;
int interval;
TimerCallback fn;
};
int Alert(FPDF_WIDESTRING message,
FPDF_WIDESTRING title,
int type,
int icon) override {
alerts_.push_back(
{GetPlatformWString(message), GetPlatformWString(title), type, icon});
return 0;
}
int SetTimer(int msecs, TimerCallback fn) override {
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 {
for (auto iter = expiry_to_timer_map_.begin();
iter != expiry_to_timer_map_.end(); ++iter) {
if (iter->second.id == id) {
expiry_to_timer_map_.erase(iter);
break;
}
}
}
void AdvanceTime(int increment_msecs) {
fake_elapsed_msecs_ += increment_msecs;
while (true) {
auto iter = expiry_to_timer_map_.begin();
if (iter == expiry_to_timer_map_.end()) {
break;
}
if (iter->first > fake_elapsed_msecs_) {
break;
}
Timer t = iter->second;
expiry_to_timer_map_.erase(iter);
expiry_to_timer_map_.insert(
std::pair<int, Timer>(fake_elapsed_msecs_ + t.interval, t));
t.fn(t.id); // Fire timer.
}
}
const std::vector<AlertRecord>& GetAlerts() const { return alerts_; }
void SetFailNextTimer() { fail_next_timer_ = true; }
private:
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_;
};
#endif // TESTING_EMBEDDER_TEST_TIMER_HANDLING_DELEGATE_H_