1 // Copyright 2013 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.
5 #ifndef CHROME_BROWSER_COMPONENT_UPDATER_TEST_COMPONENT_UPDATER_SERVICE_UNITTEST_H_
6 #define CHROME_BROWSER_COMPONENT_UPDATER_TEST_COMPONENT_UPDATER_SERVICE_UNITTEST_H_
14 #include "base/basictypes.h"
15 #include "base/compiler_specific.h"
16 #include "base/files/file_path.h"
17 #include "base/memory/ref_counted.h"
18 #include "base/memory/scoped_ptr.h"
19 #include "chrome/browser/component_updater/component_updater_service.h"
20 #include "chrome/browser/component_updater/test/url_request_post_interceptor.h"
21 #include "content/public/test/test_browser_thread_bundle.h"
22 #include "content/test/net/url_request_prepackaged_interceptor.h"
23 #include "net/url_request/url_request_test_util.h"
24 #include "testing/gmock/include/gmock/gmock.h"
25 #include "testing/gtest/include/gtest/gtest.h"
27 namespace component_updater {
31 // Intercepts HTTP GET requests sent to "localhost".
32 typedef content::URLLocalHostRequestPrepackagedInterceptor GetInterceptor;
34 // Intercepts HTTP POST requests sent to "localhost2".
35 class InterceptorFactory : public URLRequestPostInterceptorFactory {
38 ~InterceptorFactory();
40 URLRequestPostInterceptor* CreateInterceptor();
43 DISALLOW_COPY_AND_ASSIGN(InterceptorFactory);
46 class PartialMatch : public URLRequestPostInterceptor::RequestMatcher {
48 explicit PartialMatch(const std::string& expected) : expected_(expected) {}
49 virtual bool Match(const std::string& actual) const OVERRIDE;
52 const std::string expected_;
54 DISALLOW_COPY_AND_ASSIGN(PartialMatch);
57 // component 1 has extension id "jebgalgnebhfojomionfpkfelancnnkf", and
58 // the RSA public key the following hash:
59 const uint8 jebg_hash[] = {0x94, 0x16, 0x0b, 0x6d, 0x41, 0x75, 0xe9, 0xec,
60 0x8e, 0xd5, 0xfa, 0x54, 0xb0, 0xd2, 0xdd, 0xa5,
61 0x6e, 0x05, 0x6b, 0xe8, 0x73, 0x47, 0xf6, 0xc4,
62 0x11, 0x9f, 0xbc, 0xb3, 0x09, 0xb3, 0x5b, 0x40};
63 // component 2 has extension id "abagagagagagagagagagagagagagagag", and
64 // the RSA public key the following hash:
65 const uint8 abag_hash[] = {0x01, 0x06, 0x06, 0x06, 0x06, 0x06, 0x06, 0x06,
66 0x06, 0x06, 0x06, 0x06, 0x06, 0x06, 0x06, 0x06,
67 0x06, 0x06, 0x06, 0x06, 0x06, 0x06, 0x06, 0x06,
68 0x06, 0x06, 0x06, 0x06, 0x06, 0x06, 0x06, 0x01};
69 // component 3 has extension id "ihfokbkgjpifnbbojhneepfflplebdkc", and
70 // the RSA public key the following hash:
71 const uint8 ihfo_hash[] = {0x87, 0x5e, 0xa1, 0xa6, 0x9f, 0x85, 0xd1, 0x1e,
72 0x97, 0xd4, 0x4f, 0x55, 0xbf, 0xb4, 0x13, 0xa2,
73 0xe7, 0xc5, 0xc8, 0xf5, 0x60, 0x19, 0x78, 0x1b,
74 0x6d, 0xe9, 0x4c, 0xeb, 0x96, 0x05, 0x42, 0x17};
76 class TestConfigurator : public ComponentUpdateService::Configurator {
79 virtual ~TestConfigurator();
81 // Overrrides for ComponentUpdateService::Configurator.
82 virtual int InitialDelay() OVERRIDE;
83 virtual int NextCheckDelay() OVERRIDE;
84 virtual int StepDelay() OVERRIDE;
85 virtual int StepDelayMedium() OVERRIDE;
86 virtual int MinimumReCheckWait() OVERRIDE;
87 virtual int OnDemandDelay() OVERRIDE;
88 virtual GURL UpdateUrl() OVERRIDE;
89 virtual GURL PingUrl() OVERRIDE;
90 virtual std::string ExtraRequestParams() OVERRIDE;
91 virtual size_t UrlSizeLimit() OVERRIDE;
92 virtual net::URLRequestContextGetter* RequestContext() OVERRIDE;
93 virtual bool InProcess() OVERRIDE;
94 virtual bool DeltasEnabled() const OVERRIDE;
95 virtual bool UseBackgroundDownloader() const OVERRIDE;
97 typedef std::pair<CrxComponent*, int> CheckAtLoopCount;
98 void SetLoopCount(int times);
99 void SetRecheckTime(int seconds);
100 void SetOnDemandTime(int seconds);
101 void SetComponentUpdateService(ComponentUpdateService* cus);
102 void SetQuitClosure(const base::Closure& quit_closure);
103 void SetInitialDelay(int seconds);
111 ComponentUpdateService* cus_;
112 scoped_refptr<net::TestURLRequestContextGetter> context_;
113 base::Closure quit_closure_;
116 class ComponentUpdaterTest : public testing::Test {
118 enum TestComponents {
124 ComponentUpdaterTest();
126 virtual ~ComponentUpdaterTest();
128 virtual void SetUp();
130 virtual void TearDown();
132 ComponentUpdateService* component_updater();
134 // Makes the full path to a component updater test file.
135 const base::FilePath test_file(const char* file);
137 TestConfigurator* test_configurator();
139 ComponentUpdateService::Status RegisterComponent(CrxComponent* com,
140 TestComponents component,
141 const Version& version,
142 TestInstaller* installer);
146 void RunThreadsUntilIdle();
148 scoped_ptr<InterceptorFactory> interceptor_factory_;
149 URLRequestPostInterceptor* post_interceptor_; // Owned by the factory.
151 scoped_ptr<GetInterceptor> get_interceptor_;
154 TestConfigurator* test_config_;
155 base::FilePath test_data_dir_;
156 content::TestBrowserThreadBundle thread_bundle_;
157 scoped_ptr<ComponentUpdateService> component_updater_;
160 const char expected_crx_url[] =
161 "http://localhost/download/jebgalgnebhfojomionfpkfelancnnkf.crx";
163 class MockServiceObserver : public ServiceObserver {
165 MockServiceObserver();
166 ~MockServiceObserver();
167 MOCK_METHOD2(OnEvent, void(Events event, const std::string&));
170 class OnDemandTester {
172 static ComponentUpdateService::Status OnDemand(
173 ComponentUpdateService* cus,
174 const std::string& component_id);
177 } // namespace component_updater
179 #endif // CHROME_BROWSER_COMPONENT_UPDATER_TEST_COMPONENT_UPDATER_SERVICE_UNITTEST_H_