Framework and tests for WorkerNavigatorBudget

third_party/WebKit/Source/modules/budget/WorkerNavigatorBudget.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 "modules/budget/WorkerNavigatorBudget.h"
+
+#include "core/workers/WorkerNavigator.h"
+#include "modules/budget/BudgetService.h"
+
+namespace blink {
+
+WorkerNavigatorBudget::WorkerNavigatorBudget()
+{
+}
+
+// static
+const char* WorkerNavigatorBudget::supplementName()
+{
+    return "WorkerNavigatorBudget";
+}
+
+// static
+WorkerNavigatorBudget& WorkerNavigatorBudget::from(WorkerNavigator& workerNavigator)
+{
+    // Get the unique WorkerNavigatorBudget associated with this workerNavigator.
+    WorkerNavigatorBudget* workerNavigatorBudget = static_cast<WorkerNavigatorBudget*>(Supplement<WorkerNavigator>::from(workerNavigator, supplementName()));
+    if (!workerNavigatorBudget) {
+        // If there isn't one already, create it now and associate it.
+        workerNavigatorBudget = new WorkerNavigatorBudget();
+        Supplement<WorkerNavigator>::provideTo(workerNavigator, supplementName(), workerNavigatorBudget);
+    }
+    return *workerNavigatorBudget;
+}
+
+BudgetService* WorkerNavigatorBudget::budget()
+{
+    if (!m_budget)
+        m_budget = BudgetService::create();
+    return m_budget.get();
+}
+
+// static
+BudgetService* WorkerNavigatorBudget::budget(WorkerNavigator& workerNavigator)
+{
+    return WorkerNavigatorBudget::from(workerNavigator).budget();
+}
+
+DEFINE_TRACE(WorkerNavigatorBudget)
+{
+    visitor->trace(m_budget);
+    Supplement<WorkerNavigator>::trace(visitor);
+}
+
+} // namespace blink