src/chrome/browser/extensions/external_provider_impl_unittest.cc

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