15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2011 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#ifndef CHROME_INSTALLER_UTIL_REGISTRY_TEST_DATA_H_ 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#define CHROME_INSTALLER_UTIL_REGISTRY_TEST_DATA_H_ 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <windows.h> 95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <string> 115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/basictypes.h" 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// A helper class for use by unit tests that need some registry space and data. 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// BEWARE: Instances of this class irrevocably and recursively delete keys and 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// values from the registry. Carefully read the comments for Initialize and 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Reset before use. 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class RegistryTestData { 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) RegistryTestData(); 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Invokes Reset() on its way out. 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ~RegistryTestData(); 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Resets this instance, deletes the key rooted at |base_path|, and then 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // populates |base_path| with: 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // \EmptyKey 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // \NonEmptyKey (default value = "|base_path|\NonEmptyKey") 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // \NonEmptyKey\Subkey ("SomeValue" = DWORD 1) 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool Initialize(HKEY root_key, const wchar_t* base_path); 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Deletes the key rooted at base_path and clears all state. 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void Reset(); 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Fires Google Test expectations in the hopes that |path| contains the same 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // data as originally placed in |non_empty_key| by Initialize(). 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void ExpectMatchesNonEmptyKey(HKEY root_key, const wchar_t* path); 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) HKEY root_key() const { return root_key_; } 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const std::wstring& base_path() const { return base_path_; } 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const std::wstring& empty_key_path() const { return empty_key_path_; } 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const std::wstring& non_empty_key_path() const { return non_empty_key_path_; } 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Fires Google Test expectations in the hopes that |path| is an empty key 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // (exists but has no values or subkeys). 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) static void ExpectEmptyKey(HKEY root_key, const wchar_t* path); 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) private: 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) static bool DeleteKey(HKEY root_key, const wchar_t* path); 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) HKEY root_key_; 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::wstring base_path_; 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::wstring empty_key_path_; 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::wstring non_empty_key_path_; 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DISALLOW_COPY_AND_ASSIGN(RegistryTestData); 565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif // CHROME_INSTALLER_UTIL_REGISTRY_TEST_DATA_H_ 59