Home | History | Annotate | Download | only in extensions 
      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.
      4 
      5 #include "chrome/browser/extensions/external_provider_impl.h"
      6 
      7 #include "base/command_line.h"
      8 #include "base/files/file_path.h"
      9 #include "base/files/file_util.h"
     10 #include "base/memory/scoped_ptr.h"
     11 #include "base/path_service.h"
     12 #include "base/prefs/testing_pref_service.h"
     13 #include "base/strings/stringprintf.h"
     14 #include "base/test/scoped_path_override.h"
     15 #include "chrome/browser/chrome_notification_types.h"
     16 #include "chrome/browser/extensions/extension_service.h"
     17 #include "chrome/browser/extensions/extension_service_test_base.h"
     18 #include "chrome/browser/extensions/updater/extension_cache_fake.h"
     19 #include "chrome/common/chrome_paths.h"
     20 #include "chrome/common/chrome_switches.h"
     21 #include "chrome/common/extensions/extension_constants.h"
     22 #include "chrome/common/pref_names.h"
     23 #include "chrome/test/base/testing_browser_process.h"
     24 #include "chrome/test/base/testing_profile.h"
     25 #include "content/public/browser/notification_service.h"
     26 #include "content/public/test/test_utils.h"
     27 #include "net/test/embedded_test_server/embedded_test_server.h"
     28 #include "net/test/embedded_test_server/http_request.h"
     29 #include "net/test/embedded_test_server/http_response.h"
     30 #include "testing/gmock/include/gmock/gmock.h"
     31 
     32 #if defined(OS_CHROMEOS)
     33 #include "chrome/browser/chromeos/customization_document.h"
     34 #include "chrome/browser/chromeos/login/users/fake_user_manager.h"
     35 #include "chrome/browser/chromeos/login/users/scoped_user_manager_enabler.h"
     36 #include "chromeos/system/mock_statistics_provider.h"
     37 #include "chromeos/system/statistics_provider.h"
     38 #endif
     39 
     40 using ::testing::_;
     41 using ::testing::NotNull;
     42 using ::testing::Return;
     43 
     44 namespace extensions {
     45 
     46 namespace {
     47 
     48 using namespace net::test_server;
     49 
     50 const char kManifestPath[] = "/update_manifest";
     51 const char kAppPath[] = "/app.crx";
     52 
     53 class ExternalProviderImplTest : public ExtensionServiceTestBase {
     54  public:
     55   ExternalProviderImplTest() {}
     56   virtual ~ExternalProviderImplTest() {}
     57 
     58   void InitServiceWithExternalProviders() {
     59 #if defined(OS_CHROMEOS)
     60     chromeos::ScopedUserManagerEnabler scoped_user_manager(
     61         new chromeos::FakeUserManager);
     62 #endif
     63     InitializeExtensionServiceWithUpdater();
     64 
     65     // Don't install default apps. Some of the default apps are downloaded from
     66     // the webstore, ignoring the url we pass to kAppsGalleryUpdateURL, which
     67     // would cause the external updates to never finish install.
     68     profile_->GetPrefs()->SetString(prefs::kDefaultApps, "");
     69 
     70     ProviderCollection providers;
     71     extensions::ExternalProviderImpl::CreateExternalProviders(
     72         service_, profile_.get(), &providers);
     73 
     74     for (ProviderCollection::iterator i = providers.begin();
     75          i != providers.end();
     76          ++i) {
     77       service_->AddProviderForTesting(i->release());
     78     }
     79   }
     80 
     81   // ExtensionServiceTestBase overrides:
     82   virtual void SetUp() OVERRIDE {
     83     ExtensionServiceTestBase::SetUp();
     84     test_server_.reset(new EmbeddedTestServer());
     85 
     86 #if defined(OS_CHROMEOS)
     87     TestingBrowserProcess::GetGlobal()->SetLocalState(&local_state_);
     88     chromeos::ServicesCustomizationDocument::RegisterPrefs(
     89         local_state_.registry());
     90 #endif
     91 
     92     ASSERT_TRUE(test_server_->InitializeAndWaitUntilReady());
     93     test_server_->RegisterRequestHandler(
     94         base::Bind(&ExternalProviderImplTest::HandleRequest,
     95                    base::Unretained(this)));
     96 
     97     test_extension_cache_.reset(new ExtensionCacheFake());
     98 
     99     CommandLine* cmdline = CommandLine::ForCurrentProcess();
    100     cmdline->AppendSwitchASCII(switches::kAppsGalleryUpdateURL,
    101                                test_server_->GetURL(kManifestPath).spec());
    102 #if defined(OS_CHROMEOS)
    103     chromeos::system::StatisticsProvider::SetTestProvider(
    104         &mock_statistics_provider_);
    105     EXPECT_CALL(mock_statistics_provider_, GetMachineStatistic(_, NotNull()))
    106         .WillRepeatedly(Return(false));
    107 #endif
    108   }
    109 
    110   virtual void TearDown() OVERRIDE {
    111 #if defined(OS_CHROMEOS)
    112     chromeos::system::StatisticsProvider::SetTestProvider(NULL);
    113     TestingBrowserProcess::GetGlobal()->SetLocalState(NULL);
    114 #endif
    115   }
    116 
    117  private:
    118   scoped_ptr<HttpResponse> HandleRequest(const HttpRequest& request) {
    119     GURL url = test_server_->GetURL(request.relative_url);
    120     if (url.path() == kManifestPath) {
    121       scoped_ptr<BasicHttpResponse> response(new BasicHttpResponse);
    122       response->set_code(net::HTTP_OK);
    123       response->set_content(base::StringPrintf(
    124           "<?xml version='1.0' encoding='UTF-8'?>\n"
    125           "<gupdate xmlns='http://www.google.com/update2/response' "
    126               "protocol='2.0'>\n"
    127           "  <app appid='%s'>\n"
    128           "    <updatecheck codebase='%s' version='1.0' />\n"
    129           "  </app>\n"
    130           "</gupdate>",
    131           extension_misc::kInAppPaymentsSupportAppId,
    132           test_server_->GetURL(kAppPath).spec().c_str()));
    133       response->set_content_type("text/xml");
    134       return response.PassAs<HttpResponse>();
    135     }
    136     if (url.path() == kAppPath) {
    137       base::FilePath test_data_dir;
    138       PathService::Get(chrome::DIR_TEST_DATA, &test_data_dir);
    139       std::string contents;
    140       base::ReadFileToString(
    141           test_data_dir.AppendASCII("extensions/dummyiap.crx"),
    142           &contents);
    143       scoped_ptr<BasicHttpResponse> response(new BasicHttpResponse);
    144       response->set_code(net::HTTP_OK);
    145       response->set_content(contents);
    146       return response.PassAs<HttpResponse>();
    147     }
    148 
    149     return scoped_ptr<HttpResponse>();
    150   }
    151 
    152   scoped_ptr<EmbeddedTestServer> test_server_;
    153   scoped_ptr<ExtensionCacheFake> test_extension_cache_;
    154 #if defined(OS_CHROMEOS)
    155   chromeos::system::MockStatisticsProvider mock_statistics_provider_;
    156   TestingPrefServiceSimple local_state_;
    157 #endif
    158 
    159   DISALLOW_COPY_AND_ASSIGN(ExternalProviderImplTest);
    160 };
    161 
    162 }  // namespace
    163 
    164 TEST_F(ExternalProviderImplTest, InAppPayments) {
    165   InitServiceWithExternalProviders();
    166 
    167   scoped_refptr<content::MessageLoopRunner> runner =
    168       new content::MessageLoopRunner;
    169   service_->set_external_updates_finished_callback_for_test(
    170       runner->QuitClosure());
    171 
    172   service_->CheckForExternalUpdates();
    173   runner->Run();
    174 
    175   EXPECT_TRUE(service_->GetInstalledExtension(
    176       extension_misc::kInAppPaymentsSupportAppId));
    177   EXPECT_TRUE(service_->IsExtensionEnabled(
    178       extension_misc::kInAppPaymentsSupportAppId));
    179 }
    180 
    181 }  // namespace extensions
    182