extension_rlz_apitest.cc revision 3345a6884c488ff3a535c2c9acdd33d74b37e311
1// Copyright (c) 2010 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#include <map> 6 7#include "base/registry.h" 8#include "chrome/browser/browser_process.h" 9#include "chrome/browser/extensions/extension_function.h" 10#include "chrome/browser/extensions/extension_function_dispatcher.h" 11#include "chrome/browser/extensions/extension_apitest.h" 12#include "chrome/browser/extensions/extension_rlz_module.h" 13#include "chrome/common/chrome_switches.h" 14#include "chrome/common/extensions/extension.h" 15#include "rlz/win/lib/rlz_lib.h" 16 17class MockRlzSendFinancialPingFunction : public RlzSendFinancialPingFunction { 18 virtual bool RunImpl(); 19 20 static int expected_count_; 21 22 public: 23 static int expected_count() { 24 return expected_count_; 25 } 26}; 27 28int MockRlzSendFinancialPingFunction::expected_count_ = 0; 29 30bool MockRlzSendFinancialPingFunction::RunImpl() { 31 EXPECT_TRUE(RlzSendFinancialPingFunction::RunImpl()); 32 ++expected_count_; 33 return true; 34} 35 36ExtensionFunction* MockRlzSendFinancialPingFunctionFactory() { 37 return new MockRlzSendFinancialPingFunction(); 38} 39 40IN_PROC_BROWSER_TEST_F(ExtensionApiTest, Rlz) { 41 CommandLine::ForCurrentProcess()->AppendSwitch( 42 switches::kEnableExperimentalExtensionApis); 43 44 // Before running the tests, clear the state of the RLZ products used. 45 rlz_lib::AccessPoint access_points[] = { 46 rlz_lib::GD_WEB_SERVER, 47 rlz_lib::GD_OUTLOOK, 48 rlz_lib::NO_ACCESS_POINT, 49 }; 50 rlz_lib::ClearProductState(rlz_lib::PINYIN_IME, access_points); 51 rlz_lib::ClearProductState(rlz_lib::DESKTOP, access_points); 52 53 // Check that the state has really been cleared. 54 RegKey key(HKEY_CURRENT_USER, L"Software\\Google\\Common\\Rlz\\Events\\N", 55 KEY_READ); 56 ASSERT_FALSE(key.Valid()); 57 58 key.Open(HKEY_CURRENT_USER, L"Software\\Google\\Common\\Rlz\\Events\\D", 59 KEY_READ); 60 ASSERT_FALSE(key.Valid()); 61 62 // Mock out experimental.rlz.sendFinancialPing(). 63 ASSERT_TRUE(ExtensionFunctionDispatcher::OverrideFunction( 64 "experimental.rlz.sendFinancialPing", 65 MockRlzSendFinancialPingFunctionFactory)); 66 67 // Set the access point that the test code is expecting. 68 ASSERT_TRUE(rlz_lib::SetAccessPointRlz(rlz_lib::GD_DESKBAND, "rlz_apitest")); 69 70 // Now run all the tests. 71 ASSERT_TRUE(RunExtensionTest("rlz")) << message_; 72 73 ASSERT_EQ(3, MockRlzSendFinancialPingFunction::expected_count()); 74 ExtensionFunctionDispatcher::ResetFunctions(); 75 76 // Now make sure we recorded what was expected. If the code in test.js 77 // changes, need to make appropriate changes here. 78 key.Open(HKEY_CURRENT_USER, L"Software\\Google\\Common\\Rlz\\Events\\N", 79 KEY_READ); 80 ASSERT_TRUE(key.Valid()); 81 82 DWORD value; 83 ASSERT_TRUE(key.ReadValueDW(L"D3I", &value)); 84 ASSERT_EQ(1, value); 85 ASSERT_TRUE(key.ReadValueDW(L"D3S", &value)); 86 ASSERT_EQ(1, value); 87 ASSERT_TRUE(key.ReadValueDW(L"D3F", &value)); 88 ASSERT_EQ(1, value); 89 90 ASSERT_TRUE(key.ReadValueDW(L"D4I", &value)); 91 ASSERT_EQ(1, value); 92 93 key.Open(HKEY_CURRENT_USER, L"Software\\Google\\Common\\Rlz\\Events\\D", 94 KEY_READ); 95 ASSERT_FALSE(key.Valid()); 96 97 // Cleanup. 98 rlz_lib::ClearProductState(rlz_lib::PINYIN_IME, access_points); 99} 100