policy_manager/fake_state.h

// Copyright (c) 2014 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_POLICY_MANAGER_FAKE_STATE_H_
#define CHROMEOS_PLATFORM_UPDATE_ENGINE_POLICY_MANAGER_FAKE_STATE_H_

#include "update_engine/policy_manager/fake_random_provider.h"
#include "update_engine/policy_manager/fake_shill_provider.h"
#include "update_engine/policy_manager/fake_system_provider.h"
#include "update_engine/policy_manager/fake_time_provider.h"
#include "update_engine/policy_manager/state.h"

namespace chromeos_policy_manager {

// A fake State class that creates fake providers for all the providers.
class FakeState : public State {
 public:
  // Creates and initializes the FakeState using fake providers. Returns NULL
  // if the initialization fails.
  static FakeState* Construct();

  virtual ~FakeState() {}

  // Downcasted getters, to allow access to the fake instances during testing.
  virtual FakeRandomProvider* random_provider() override {
    return reinterpret_cast<FakeRandomProvider*>(State::random_provider());
  }
  virtual FakeShillProvider* shill_provider() override {
    return reinterpret_cast<FakeShillProvider*>(State::shill_provider());
  }
  virtual FakeTimeProvider* time_provider() override {
    return reinterpret_cast<FakeTimeProvider*>(State::time_provider());
  }
  virtual FakeSystemProvider* system_provider() override {
    return reinterpret_cast<FakeSystemProvider*>(State::system_provider());
  }

 private:
  // Creates a FakeState instance using FakeProviders.
  FakeState();

  DISALLOW_COPY_AND_ASSIGN(FakeState);
};

}  // namespace chromeos_policy_manager

#endif  // CHROMEOS_PLATFORM_UPDATE_ENGINE_POLICY_MANAGER_FAKE_STATE_H_