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