third_party/WebKit/Source/web/tests/VirtualTimeTest.cpp

// Copyright 2016 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "platform/testing/UnitTestHelpers.h"
#include "public/platform/Platform.h"
#include "public/platform/WebViewScheduler.h"
#include "public/web/WebLocalFrame.h"
#include "public/web/WebScriptExecutionCallback.h"
#include "public/web/WebScriptSource.h"
#include "public/web/WebView.h"
#include "web/tests/sim/SimTest.h"

namespace blink {

namespace {
class ScriptExecutionCallbackHelper : public WebScriptExecutionCallback {
public:
    const String result() const { return m_result; }

private:
    void completed(const WebVector<v8::Local<v8::Value>>& values) override
    {
        if (!values.isEmpty() && !values[0].IsEmpty() && values[0]->IsString()) {
            m_result = toCoreString(v8::Local<v8::String>::Cast(values[0]));
        }
    }

    String m_result;
};
} // namespace

class VirtualTimeTest : public SimTest {
protected:
    String ExecuteJavaScript(String scriptSource)
    {
        ScriptExecutionCallbackHelper callbackHelper;
        webView().mainFrame()->toWebLocalFrame()->requestExecuteScriptAndReturnValue(
            WebScriptSource(WebString(scriptSource)), false, &callbackHelper);
        testing::runPendingTasks();
        return callbackHelper.result();
    }
};

TEST_F(VirtualTimeTest, DOMTimersFireInExpectedOrder)
{
    ExecuteJavaScript(
        "var run_order = [];"
        "function timerFn(delay, value) {"
        "  setTimeout(function() { run_order.push(value); }, delay);"
        "};"
        "var one_hour = 60 * 60 * 1000;"
        "timerFn(one_hour * 100, 'a');"
        "timerFn(one_hour * 10, 'b');"
        "timerFn(one_hour, 'c');");

    webView().scheduler()->enableVirtualTime();

    // Normally the JS runs pretty much instantly but the timer callbacks will
    // take 100h to fire, but thanks to timer fast forwarding we can make them
    // fire immediatly.
    testing::runPendingTasks();

    EXPECT_EQ("c, b, a", ExecuteJavaScript("run_order.join(', ')"));
}

TEST_F(VirtualTimeTest, SetInterval)
{
    ExecuteJavaScript(
        "var run_order = [];"
        "var count = 10;"
        "var interval_handle = setInterval(function() {"
        "  if (--window.count == 0) {"
        "     clearInterval(interval_handle);"
        "  }"
        "  run_order.push(count);"
        "}, 1000);"
        "setTimeout(function() { run_order.push('timer'); }, 1500);");

    webView().scheduler()->enableVirtualTime();
    testing::runPendingTasks();

    // If virtual time is not supplied to TimerBase then the setInterval
    // won't fire 10x.
    EXPECT_EQ("9, timer, 8, 7, 6, 5, 4, 3, 2, 1, 0", ExecuteJavaScript("run_order.join(', ')"));
}

TEST_F(VirtualTimeTest, AllowVirtualTimeToAdvance)
{
    ExecuteJavaScript(
        "var run_order = [];"
        "timerFn = function(delay, value) {"
        "  setTimeout(function() { run_order.push(value); }, delay);"
        "};"
        "timerFn(100, 'a');"
        "timerFn(10, 'b');"
        "timerFn(1, 'c');");

    webView().scheduler()->enableVirtualTime();
    webView().scheduler()->setAllowVirtualTimeToAdvance(false);
    testing::runPendingTasks();

    EXPECT_EQ("", ExecuteJavaScript("run_order.join(', ')"));

    webView().scheduler()->setAllowVirtualTimeToAdvance(true);
    testing::runPendingTasks();

    EXPECT_EQ("c, b, a", ExecuteJavaScript("run_order.join(', ')"));
}

} // namespace blink