Created Mojo BudgetService implementation in chrome/browser

chrome/browser/budget_service/budget_manager_unittest.cc

 // 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 <stdint.h>
 #include <string>
 
 #include "base/memory/ptr_util.h"
 #include "base/run_loop.h"
 #include "base/test/simple_test_clock.h"
 #include "chrome/browser/budget_service/budget_manager.h"
 #include "chrome/browser/budget_service/budget_manager_factory.h"
 #include "chrome/browser/engagement/site_engagement_service.h"
 #include "chrome/common/pref_names.h"
 #include "chrome/test/base/testing_profile.h"
 #include "components/prefs/pref_service.h"
 #include "components/prefs/scoped_user_pref_update.h"
 #include "content/public/test/test_browser_thread_bundle.h"
 #include "testing/gtest/include/gtest/gtest.h"
+#include "third_party/WebKit/public/platform/modules/budget_service/budget_service.mojom.h"
 
 namespace {
 
 const char kTestOrigin[] = "https://example.com";
 const double kTestBudget = 10.0;
 const double kTestSES = 48.0;
 const double kLowSES = 1.0;
 const double kMaxSES = 100.0;
 // Mirrors definition in BudgetManager, this is 10 days of seconds.
 const double kSecondsToAccumulate = 864000.0;
@@ skipping 156 matching lines @@
   double budget = 0.0;
 
   // Measure over 200 hours. In each hour a message is received, and for 1 in
   // 10, budget is consumed.
   for (int i = 0; i < 200; i++) {
     // Query for the budget after 1 hour has passed.
     clock->Advance(base::TimeDelta::FromHours(1));
     budget = GetBudget();
 
     if (i % 10 == 0) {
-      double cost =
-          BudgetManager::GetCost(BudgetManager::CostType::SILENT_PUSH);
+      double cost = BudgetManager::GetCost(
+          blink::mojom::BudgetOperationType::SILENT_PUSH);
       StoreBudget(budget - cost);
     }
   }
 
   // With a SES of 48.0, the origin will get a budget of 2.4 per day, but the
   // old budget will also decay. At the end, we expect the budget to be lower
   // than the starting budget.
   EXPECT_GT(budget, 0.0);
   EXPECT_LT(budget, kTestBudget);
 }
@@ skipping 31 matching lines @@
 
   // Store the updated budget.
   StoreBudget(original_budget);
   EXPECT_NE(kTestBudget, original_budget);
 
   // Now move time backwards a day and make sure that the current
   // budget matches the budget of the most foward time.
   clock->SetNow(starting_time - base::TimeDelta::FromDays(1));
   EXPECT_NEAR(original_budget, GetBudget(), 0.01);
 }