update_attempter_mock.h

// Copyright (c) 2010 The Chromium OS Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef CHROMEOS_PLATFORM_UPDATE_ENGINE_UPDATE_ATTEMPTER_MOCK_H_
#define CHROMEOS_PLATFORM_UPDATE_ENGINE_UPDATE_ATTEMPTER_MOCK_H_

#include <gmock/gmock.h>

#include "update_engine/mock_dbus_wrapper.h"
#include "update_engine/mock_system_state.h"
#include "update_engine/update_attempter.h"

namespace chromeos_update_engine {

class MockSystemState;

class UpdateAttempterMock : public UpdateAttempter {
 public:
  explicit UpdateAttempterMock(MockSystemState* mock_system_state,
                               MockDBusWrapper* dbus)
      : UpdateAttempter(mock_system_state, dbus) {}

  MOCK_METHOD5(Update, void(const std::string& app_version,
                            const std::string& omaha_url,
                            bool obey_proxies,
                            bool interactive,
                            bool is_test));
};

}  // namespace chromeos_update_engine

#endif  // CHROMEOS_PLATFORM_UPDATE_ENGINE_UPDATE_ATTEMPTER_MOCK_H_