1 files changed, 87 insertions, 0 deletions
diff --git a/include/nativepower/power_manager_stub.h b/include/nativepower/power_manager_stub.h
new file mode 100644
index 0000000..ab303c2
--- /dev/null
+++ b/include/nativepower/power_manager_stub.h
@@ -0,0 +1,87 @@
+/*
+ * Copyright (C) 2015 The Android Open Source Project
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#include <map>
+#include <string>
+
+#include <base/macros.h>
+#include <nativepower/BnPowerManager.h>
+
+namespace android {
+
+// Stub implementation of BnPowerManager for use in tests.
+class PowerManagerStub : public BnPowerManager {
+ public:
+  PowerManagerStub();
+  ~PowerManagerStub() override;
+
+  // Constructs a string that can be compared with one returned by
+  // GetLockString().
+  static std::string ConstructLockString(const std::string& tag,
+                                         const std::string& package,
+                                         int uid);
+
+  size_t num_locks() const { return locks_.size(); }
+
+  // Returns a string describing the lock registered for |binder|, or an empty
+  // string if no lock is present.
+  std::string GetLockString(const sp<IBinder>& binder) const;
+
+  // BnPowerManager:
+  status_t acquireWakeLock(int flags,
+                           const sp<IBinder>& lock,
+                           const String16& tag,
+                           const String16& packageName,
+                           bool isOneWay=false) override;
+  status_t acquireWakeLockWithUid(int flags,
+                                  const sp<IBinder>& lock,
+                                  const String16& tag,
+                                  const String16& packageName,
+                                  int uid,
+                                  bool isOneWay=false) override;
+  status_t releaseWakeLock(const sp<IBinder>& lock,
+                           int flags,
+                           bool isOneWay=false) override;
+  status_t updateWakeLockUids(const sp<IBinder>& lock,
+                              int len,
+                              const int* uids,
+                              bool isOneWay=false) override;
+  status_t powerHint(int hintId, int data) override;
+
+ private:
+  // Contains information passed to acquireWakeLock() or
+  // acquireWakeLockWithUid().
+  struct LockInfo {
+    LockInfo();
+    LockInfo(const LockInfo& info);
+    LockInfo(const std::string& tag,
+             const std::string& package,
+             int uid);
+
+    std::string tag;
+    std::string package;
+
+    // -1 if acquireWakeLock() was used.
+    int uid;
+  };
+
+  using LockInfoMap = std::map<sp<IBinder>, LockInfo>;
+  LockInfoMap locks_;
+
+  DISALLOW_COPY_AND_ASSIGN(PowerManagerStub);
+};
+
+}  // namespace android