Connect Mojo budget_service to BudgetManager implementation of Reserve.

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 "chrome/browser/budget_service/budget_manager.h"
@@ skipping 25 matching lines @@
   }
 
   void SetSiteEngagementScore(double score) {
     SiteEngagementService* service = SiteEngagementService::Get(&profile_);
     service->ResetScoreForURL(GURL(kTestOrigin), score);
   }
 
   Profile* profile() { return &profile_; }
   const url::Origin origin() const { return origin_; }
 
+  void ReserveCallback(base::Closure run_loop_closure,
+                       blink::mojom::BudgetServiceErrorType error,
+                       bool success) {
+    success_ = (error == blink::mojom::BudgetServiceErrorType::NONE) && success;
+    run_loop_closure.Run();
+  }
+
   void StatusCallback(base::Closure run_loop_closure, bool success) {
     success_ = success;
     run_loop_closure.Run();
   }
 
   bool ReserveBudget(blink::mojom::BudgetOperationType type) {
     base::RunLoop run_loop;
     GetManager()->Reserve(
         origin(), type,
-        base::Bind(&BudgetManagerTest::StatusCallback, base::Unretained(this),
+        base::Bind(&BudgetManagerTest::ReserveCallback, base::Unretained(this),
                    run_loop.QuitClosure()));
     run_loop.Run();
     return success_;
   }
 
   bool ConsumeBudget(blink::mojom::BudgetOperationType type) {
     base::RunLoop run_loop;
     GetManager()->Consume(
         origin(), type,
         base::Bind(&BudgetManagerTest::StatusCallback, base::Unretained(this),
@@ skipping 27 matching lines @@
   ASSERT_FALSE(ReserveBudget(type));
 
   // Try to consume for the 24 messages reserved.
   for (int i = 0; i < 24; i++)
     ASSERT_TRUE(ConsumeBudget(type));
 
   // The next consume should fail, since there is no reservation or budget
   // available.
   ASSERT_FALSE(ConsumeBudget(type));
 }