Connect Mojo budget_service to BudgetManager implementation of Reserve.

chrome/browser/budget_service/budget_database_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 "chrome/browser/budget_service/budget_database.h"
 
 #include <vector>
 
 #include "base/run_loop.h"
 #include "base/test/histogram_tester.h"
@@ skipping 19 matching lines @@
 }  // namespace
 
 class BudgetDatabaseTest : public ::testing::Test {
  public:
   BudgetDatabaseTest()
       : success_(false),
         db_(&profile_,
             profile_.GetPath().Append(FILE_PATH_LITERAL("BudgetDatabase")),
             base::ThreadTaskRunnerHandle::Get()) {}
 
-  void WriteBudgetComplete(base::Closure run_loop_closure, bool success) {
-    success_ = success;
+  void WriteBudgetComplete(base::Closure run_loop_closure,
+                           blink::mojom::BudgetServiceErrorType error,
+                           bool success) {
+    success_ = (error == blink::mojom::BudgetServiceErrorType::NONE) && success;
     run_loop_closure.Run();
   }
 
   // Spend budget for the origin.
   bool SpendBudget(const GURL& origin, double amount) {
     base::RunLoop run_loop;
     db_.SpendBudget(origin, amount,
                     base::Bind(&BudgetDatabaseTest::WriteBudgetComplete,
                                base::Unretained(this), run_loop.QuitClosure()));
     run_loop.Run();
@@ skipping 253 matching lines @@
 
   std::vector<base::Bucket> low_budget_buckets =
       GetHistogramTester()->GetAllSamples(
           "PushMessaging.SESForLowBudgetOrigin");
   ASSERT_EQ(2U, low_budget_buckets.size());
   EXPECT_EQ(engagement, low_budget_buckets[0].min);
   EXPECT_EQ(1, low_budget_buckets[0].count);
   EXPECT_EQ(engagement * 2, low_budget_buckets[1].min);
   EXPECT_EQ(1, low_budget_buckets[1].count);
 }