Start plumbing connections from the BudgetManager to the BudgetDatabase

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"
@@ skipping 58 matching lines @@
     run_loop_closure.Run();
   }
 
   void StoreBudget(double budget) {
     const GURL origin(kTestOrigin);
     base::RunLoop run_loop;
     GetManager()->StoreBudget(origin, budget, run_loop.QuitClosure());
     run_loop.Run();
   }
 
-  // Budget for callbacks to set.
+  void StatusCallback(base::Closure run_loop_closure, bool success) {
+    success_ = success;
+    run_loop_closure.Run();
+  }
+
+  bool ReserveBudget(blink::mojom::BudgetOperationType type) {
+    const GURL origin(kTestOrigin);
+    base::RunLoop run_loop;
+    GetManager()->Reserve(
+        origin, type,
+        base::Bind(&BudgetManagerTest::StatusCallback, base::Unretained(this),
+                   run_loop.QuitClosure()));
+    run_loop.Run();
+    return success_;
+  }
+
+  // Members for callbacks to set.
   double budget_;
+  bool success_;
 
  private:
   content::TestBrowserThreadBundle thread_bundle_;
   TestingProfile profile_;
 };
 
 TEST_F(BudgetManagerTest, GetBudgetNoBudgetOrSES) {
   EXPECT_DOUBLE_EQ(GetBudget(), 0.0);
 }
 
@@ skipping 149 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);
 }
+
+TEST_F(BudgetManagerTest, ReserveBudgetTest) {
+  // Reserve without any budget allocated should fail.
+  ASSERT_FALSE(ReserveBudget(blink::mojom::BudgetOperationType::SILENT_PUSH));
+}