chrome/browser/chromeos/settings/mock_owner_key_util.h

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #ifndef CHROME_BROWSER_CHROMEOS_SETTINGS_MOCK_OWNER_KEY_UTIL_H_
   6 #define CHROME_BROWSER_CHROMEOS_SETTINGS_MOCK_OWNER_KEY_UTIL_H_
   7
   8 #include <vector>
   9
  10 #include "base/basictypes.h"
  11 #include "base/compiler_specific.h"
  12 #include "base/files/file_path.h"
  13 #include "base/memory/scoped_ptr.h"
  14 #include "chrome/browser/chromeos/settings/owner_key_util.h"
  15
  16 namespace chromeos {
  17
  18 class MockOwnerKeyUtil : public OwnerKeyUtil {
  19  public:
  20   MockOwnerKeyUtil();
  21
  22   // OwnerKeyUtil:
  23   virtual bool ImportPublicKey(std::vector<uint8>* output) OVERRIDE;
  24   virtual crypto::RSAPrivateKey* FindPrivateKey(
  25       const std::vector<uint8>& key) OVERRIDE;
  26   virtual bool IsPublicKeyPresent() OVERRIDE;
  27
  28   // Clears the public and private keys.
  29   void Clear();
  30
  31   // Configures the mock to return the given public key.
  32   void SetPublicKey(const std::vector<uint8>& key);
  33
  34   // Sets the public key to use from the given private key, but doesn't
  35   // configure the private key.
  36   void SetPublicKeyFromPrivateKey(const crypto::RSAPrivateKey& key);
  37
  38   // Sets the private key (also configures the public key).
  39   void SetPrivateKey(scoped_ptr<crypto::RSAPrivateKey> key);
  40
  41  protected:
  42   virtual ~MockOwnerKeyUtil();
  43
  44  private:
  45   std::vector<uint8> public_key_;
  46   scoped_ptr<crypto::RSAPrivateKey> private_key_;
  47
  48   DISALLOW_COPY_AND_ASSIGN(MockOwnerKeyUtil);
  49 };
  50
  51 }  // namespace chromeos
  52
  53 #endif  // CHROME_BROWSER_CHROMEOS_SETTINGS_MOCK_OWNER_KEY_UTIL_H_