arc: Add unit tests for ArcKioskBridge

BUG=710896
TEST=Run components_unittests

Review-Url: https://codereview.chromium.org/2811303003
Cr-Commit-Position: refs/heads/master@{#464379}

components/arc/BUILD.gn

@@ -216,6 +216,7 @@ source_set("unit_tests") {
     "intent_helper/link_handler_model_impl_unittest.cc",
     "intent_helper/local_activity_resolver_unittest.cc",
     "intent_helper/page_transition_util_unittest.cc",
+    "kiosk/arc_kiosk_bridge_unittest.cc",
   ]
 
   deps = [
@@ -225,6 +226,7 @@ source_set("unit_tests") {
     "//chromeos",
     "//device/bluetooth",
     "//mojo/public/cpp/system:system",
+    "//testing/gmock",
     "//testing/gtest",
     "//ui/aura",
     "//ui/aura:test_support",

components/arc/kiosk/arc_kiosk_bridge_unittest.cc

+// Copyright 2017 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 <memory>
+
+#include "base/macros.h"
+#include "base/memory/ptr_util.h"
+#include "components/arc/arc_bridge_service.h"
+#include "components/arc/kiosk/arc_kiosk_bridge.h"
+#include "testing/gmock/include/gmock/gmock.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace {
+
+class MockArcKioskBridgeDelegate : public arc::ArcKioskBridge::Delegate {
+ public:
+  MockArcKioskBridgeDelegate() = default;
+
+  MOCK_METHOD0(OnMaintenanceSessionCreated, void());
+  MOCK_METHOD0(OnMaintenanceSessionFinished, void());
+};
+
+}  // namespace
+
+namespace arc {
+
+class ArcKioskBridgeTest : public testing::Test {
+ public:
+  ArcKioskBridgeTest()
+      : bridge_service_(base::MakeUnique<ArcBridgeService>()),
+        delegate_(base::MakeUnique<MockArcKioskBridgeDelegate>()),
+        kiosk_bridge_(base::MakeUnique<ArcKioskBridge>(bridge_service_.get(),
+                                                       delegate_.get())) {}
+
+ protected:
+  std::unique_ptr<ArcBridgeService> bridge_service_;
+  std::unique_ptr<MockArcKioskBridgeDelegate> delegate_;
+  std::unique_ptr<ArcKioskBridge> kiosk_bridge_;
+
+ private:
+  DISALLOW_COPY_AND_ASSIGN(ArcKioskBridgeTest);
+};
+
+TEST_F(ArcKioskBridgeTest, MaintenanceSessionFinished) {
+  EXPECT_CALL(*delegate_, OnMaintenanceSessionCreated()).Times(1);
+  kiosk_bridge_->OnMaintenanceSessionCreated(1);
+  EXPECT_CALL(*delegate_, OnMaintenanceSessionFinished()).Times(1);
+  kiosk_bridge_->OnMaintenanceSessionFinished(1, true);
+}
+
+TEST_F(ArcKioskBridgeTest, MaintenanceSessionNotFinished) {
+  EXPECT_CALL(*delegate_, OnMaintenanceSessionCreated()).Times(1);
+  kiosk_bridge_->OnMaintenanceSessionCreated(1);
+  EXPECT_CALL(*delegate_, OnMaintenanceSessionFinished()).Times(0);
+  kiosk_bridge_->OnMaintenanceSessionFinished(2, true);
+}
+
+}  // namespace arc