This source file includes following definitions.
- TEST
- TEST
- TEST
- CreateURLFetcher
- AddMimeHeader
- SetUp
- TearDown
- RunUntilIdle
- AddCustomizationIdToVp
- AddExpectedManifest
- AddManifestNotFound
- CreateProfile
- TEST_F
- TEST_F
- TEST_F
- TEST_F
#include "chrome/browser/chromeos/customization_document.h"
#include "base/message_loop/message_loop.h"
#include "base/prefs/testing_pref_service.h"
#include "base/run_loop.h"
#include "base/strings/stringprintf.h"
#include "chrome/browser/chromeos/net/network_portal_detector_test_impl.h"
#include "chrome/browser/extensions/external_provider_impl.h"
#include "chrome/browser/prefs/browser_prefs.h"
#include "chrome/browser/prefs/pref_service_mock_factory.h"
#include "chrome/browser/prefs/pref_service_syncable.h"
#include "chrome/browser/ui/app_list/app_list_syncable_service.h"
#include "chrome/browser/ui/app_list/app_list_syncable_service_factory.h"
#include "chrome/test/base/testing_browser_process.h"
#include "chrome/test/base/testing_profile.h"
#include "chromeos/dbus/dbus_thread_manager.h"
#include "chromeos/network/network_handler.h"
#include "chromeos/system/mock_statistics_provider.h"
#include "components/user_prefs/pref_registry_syncable.h"
#include "content/public/test/test_browser_thread_bundle.h"
#include "extensions/common/extension.h"
#include "extensions/common/manifest.h"
#include "net/http/http_response_headers.h"
#include "net/http/http_status_code.h"
#include "net/url_request/test_url_fetcher_factory.h"
#include "net/url_request/url_request_status.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"
using ::testing::Exactly;
using ::testing::Invoke;
using ::testing::Mock;
using ::testing::_;
namespace {
const char kGoodStartupManifest[] =
"{"
" \"version\": \"1.0\","
" \"initial_locale\" : \"en-US\","
" \"initial_timezone\" : \"US/Pacific\","
" \"keyboard_layout\" : \"xkb:us::eng\","
" \"setup_content\" : {"
" \"en-US\" : {"
" \"eula_page\" : \"file:///opt/oem/eula/en-US/eula.html\","
" },"
" \"ru-RU\" : {"
" \"eula_page\" : \"file:///opt/oem/eula/ru-RU/eula.html\","
" },"
" \"default\" : {"
" \"eula_page\" : \"file:///opt/oem/eula/en/eula.html\","
" },"
" },"
" \"hwid_map\" : ["
" {"
" \"hwid_mask\": \"ZGA*34\","
" \"initial_locale\" : \"ja\","
" \"initial_timezone\" : \"Asia/Tokyo\","
" \"keyboard_layout\" : \"mozc-jp\","
" },"
" {"
" \"hwid_mask\": \"Mario 1?3*\","
" \"initial_locale\" : \"ru-RU\","
" \"initial_timezone\" : \"Europe/Moscow\","
" \"keyboard_layout\" : \"xkb:ru::rus\","
" },"
" ],"
"}";
const char kBadManifest[] = "{\"version\": \"1\"}";
const char kGoodServicesManifest[] =
"{"
" \"version\": \"1.0\","
" \"default_wallpaper\": \"http://somedomain.com/image.png\",\n"
" \"default_apps\": [\n"
" \"aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa\",\n"
" \"bbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbb\"\n"
" ],\n"
" \"localized_content\": {\n"
" \"en-US\": {\n"
" \"default_apps_folder_name\": \"EN-US OEM Name\"\n"
" },\n"
" \"en\": {\n"
" \"default_apps_folder_name\": \"EN OEM Name\"\n"
" },\n"
" \"default\": {\n"
" \"default_apps_folder_name\": \"Default OEM Name\"\n"
" }\n"
" }\n"
"}";
const char kDummyCustomizationID[] = "test-dummy";
const char kStubEthernetServicePath[] = "eth1";
}
namespace chromeos {
using ::testing::_;
using ::testing::DoAll;
using ::testing::NotNull;
using ::testing::Return;
using ::testing::SetArgumentPointee;
TEST(StartupCustomizationDocumentTest, Basic) {
system::MockStatisticsProvider mock_statistics_provider;
EXPECT_CALL(mock_statistics_provider, GetMachineStatistic(_, NotNull()))
.WillRepeatedly(Return(false));
EXPECT_CALL(mock_statistics_provider,
GetMachineStatistic(std::string("hardware_class"), NotNull()))
.WillOnce(DoAll(SetArgumentPointee<1>(std::string("Mario 12345")),
Return(true)));
StartupCustomizationDocument customization(&mock_statistics_provider,
kGoodStartupManifest);
EXPECT_EQ("ru-RU", customization.initial_locale());
EXPECT_EQ("Europe/Moscow", customization.initial_timezone());
EXPECT_EQ("xkb:ru::rus", customization.keyboard_layout());
EXPECT_EQ("file:///opt/oem/eula/en-US/eula.html",
customization.GetEULAPage("en-US"));
EXPECT_EQ("file:///opt/oem/eula/ru-RU/eula.html",
customization.GetEULAPage("ru-RU"));
EXPECT_EQ("file:///opt/oem/eula/en/eula.html",
customization.GetEULAPage("ja"));
}
TEST(StartupCustomizationDocumentTest, VPD) {
system::MockStatisticsProvider mock_statistics_provider;
EXPECT_CALL(mock_statistics_provider,
GetMachineStatistic(std::string("hardware_class"), NotNull()))
.WillOnce(DoAll(SetArgumentPointee<1>(std::string("Mario 12345")),
Return(true)));
EXPECT_CALL(mock_statistics_provider,
GetMachineStatistic(std::string("initial_locale"), NotNull()))
.WillOnce(DoAll(SetArgumentPointee<1>(std::string("ja")),
Return(true)));
EXPECT_CALL(mock_statistics_provider,
GetMachineStatistic(std::string("initial_timezone"), NotNull()))
.WillOnce(DoAll(SetArgumentPointee<1>(std::string("Asia/Tokyo")),
Return(true)));
EXPECT_CALL(mock_statistics_provider,
GetMachineStatistic(std::string("keyboard_layout"), NotNull()))
.WillOnce(DoAll(SetArgumentPointee<1>(std::string("mozc-jp")),
Return(true)));
StartupCustomizationDocument customization(&mock_statistics_provider,
kGoodStartupManifest);
EXPECT_TRUE(customization.IsReady());
EXPECT_EQ("ja", customization.initial_locale());
EXPECT_EQ("Asia/Tokyo", customization.initial_timezone());
EXPECT_EQ("mozc-jp", customization.keyboard_layout());
}
TEST(StartupCustomizationDocumentTest, BadManifest) {
system::MockStatisticsProvider mock_statistics_provider;
StartupCustomizationDocument customization(&mock_statistics_provider,
kBadManifest);
EXPECT_FALSE(customization.IsReady());
}
class TestURLFetcherCallback {
public:
scoped_ptr<net::FakeURLFetcher> CreateURLFetcher(
const GURL& url,
net::URLFetcherDelegate* d,
const std::string& response_data,
net::HttpStatusCode response_code,
net::URLRequestStatus::Status status) {
scoped_ptr<net::FakeURLFetcher> fetcher(
new net::FakeURLFetcher(url, d, response_data, response_code, status));
OnRequestCreate(url, fetcher.get());
return fetcher.Pass();
}
MOCK_METHOD2(OnRequestCreate,
void(const GURL&, net::FakeURLFetcher*));
};
void AddMimeHeader(const GURL& url, net::FakeURLFetcher* fetcher) {
scoped_refptr<net::HttpResponseHeaders> download_headers =
new net::HttpResponseHeaders("");
download_headers->AddHeader("Content-Type: application/json");
fetcher->set_response_headers(download_headers);
}
class MockExternalProviderVisitor
: public extensions::ExternalProviderInterface::VisitorInterface {
public:
MockExternalProviderVisitor() {}
MOCK_METHOD6(OnExternalExtensionFileFound,
bool(const std::string&,
const base::Version*,
const base::FilePath&,
extensions::Manifest::Location,
int,
bool));
MOCK_METHOD6(OnExternalExtensionUpdateUrlFound,
bool(const std::string&,
const std::string&,
const GURL&,
extensions::Manifest::Location,
int,
bool));
MOCK_METHOD1(OnExternalProviderReady,
void(const extensions::ExternalProviderInterface* provider));
};
class ServicesCustomizationDocumentTest : public testing::Test {
protected:
ServicesCustomizationDocumentTest()
: factory_(NULL,
base::Bind(&TestURLFetcherCallback::CreateURLFetcher,
base::Unretained(&url_callback_))) {
}
virtual void SetUp() OVERRIDE {
ServicesCustomizationDocument::InitializeForTesting();
EXPECT_CALL(mock_statistics_provider_, GetMachineStatistic(_, NotNull()))
.WillRepeatedly(Return(false));
chromeos::system::StatisticsProvider::SetTestProvider(
&mock_statistics_provider_);
DBusThreadManager::InitializeWithStub();
NetworkHandler::Initialize();
NetworkPortalDetector::InitializeForTesting(&network_portal_detector_);
NetworkPortalDetector::CaptivePortalState online_state;
online_state.status = NetworkPortalDetector::CAPTIVE_PORTAL_STATUS_ONLINE;
online_state.response_code = 204;
network_portal_detector_.SetDefaultNetworkPathForTesting(
kStubEthernetServicePath);
network_portal_detector_.SetDetectionResultsForTesting(
kStubEthernetServicePath, online_state);
TestingBrowserProcess::GetGlobal()->SetLocalState(&local_state_);
ServicesCustomizationDocument::RegisterPrefs(local_state_.registry());
}
virtual void TearDown() OVERRIDE {
TestingBrowserProcess::GetGlobal()->SetLocalState(NULL);
NetworkHandler::Shutdown();
DBusThreadManager::Shutdown();
NetworkPortalDetector::InitializeForTesting(NULL);
chromeos::system::StatisticsProvider::SetTestProvider(NULL);
ServicesCustomizationDocument::ShutdownForTesting();
}
void RunUntilIdle() {
base::RunLoop().RunUntilIdle();
}
void AddCustomizationIdToVp(const std::string& id) {
EXPECT_CALL(mock_statistics_provider_,
GetMachineStatistic(system::kCustomizationIdKey, NotNull()))
.WillOnce(DoAll(SetArgumentPointee<1>(id),
Return(true)));
}
void AddExpectedManifest(const std::string& id,
const std::string& manifest) {
GURL url(base::StringPrintf(ServicesCustomizationDocument::kManifestUrl,
id.c_str()));
factory_.SetFakeResponse(url,
manifest,
net::HTTP_OK,
net::URLRequestStatus::SUCCESS);
EXPECT_CALL(url_callback_, OnRequestCreate(url, _))
.Times(Exactly(1))
.WillRepeatedly(Invoke(AddMimeHeader));
}
void AddManifestNotFound(const std::string& id) {
GURL url(base::StringPrintf(ServicesCustomizationDocument::kManifestUrl,
id.c_str()));
factory_.SetFakeResponse(url,
std::string(),
net::HTTP_NOT_FOUND,
net::URLRequestStatus::SUCCESS);
EXPECT_CALL(url_callback_, OnRequestCreate(url, _))
.Times(Exactly(1))
.WillRepeatedly(Invoke(AddMimeHeader));
}
scoped_ptr<TestingProfile> CreateProfile() {
TestingProfile::Builder profile_builder;
PrefServiceMockFactory factory;
scoped_refptr<user_prefs::PrefRegistrySyncable> registry(
new user_prefs::PrefRegistrySyncable);
scoped_ptr<PrefServiceSyncable> prefs(
factory.CreateSyncable(registry.get()));
chrome::RegisterUserProfilePrefs(registry.get());
profile_builder.SetPrefService(prefs.Pass());
return profile_builder.Build();
}
private:
system::MockStatisticsProvider mock_statistics_provider_;
content::TestBrowserThreadBundle thread_bundle_;
TestingPrefServiceSimple local_state_;
TestURLFetcherCallback url_callback_;
net::FakeURLFetcherFactory factory_;
NetworkPortalDetectorTestImpl network_portal_detector_;
};
TEST_F(ServicesCustomizationDocumentTest, Basic) {
AddCustomizationIdToVp(kDummyCustomizationID);
AddExpectedManifest(kDummyCustomizationID, kGoodServicesManifest);
ServicesCustomizationDocument* doc =
ServicesCustomizationDocument::GetInstance();
EXPECT_FALSE(doc->IsReady());
doc->StartFetching();
RunUntilIdle();
EXPECT_TRUE(doc->IsReady());
EXPECT_EQ(doc->GetDefaultWallpaperUrl().spec(),
"http://somedomain.com/image.png");
std::vector<std::string> default_apps;
EXPECT_TRUE(doc->GetDefaultApps(&default_apps));
ASSERT_EQ(default_apps.size(), 2u);
EXPECT_EQ(default_apps[0], "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa");
EXPECT_EQ(default_apps[1], "bbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbb");
EXPECT_EQ(doc->GetOemAppsFolderName("en-US"), "EN-US OEM Name");
EXPECT_EQ(doc->GetOemAppsFolderName("en"), "EN OEM Name");
EXPECT_EQ(doc->GetOemAppsFolderName("ru"), "Default OEM Name");
}
TEST_F(ServicesCustomizationDocumentTest, NoCustomizationIdInVpd) {
ServicesCustomizationDocument* doc =
ServicesCustomizationDocument::GetInstance();
EXPECT_FALSE(doc->IsReady());
scoped_ptr<TestingProfile> profile = CreateProfile();
extensions::ExternalLoader* loader = doc->CreateExternalLoader(profile.get());
EXPECT_TRUE(loader);
MockExternalProviderVisitor visitor;
scoped_ptr<extensions::ExternalProviderImpl> provider(
new extensions::ExternalProviderImpl(
&visitor,
loader,
profile.get(),
extensions::Manifest::EXTERNAL_PREF,
extensions::Manifest::EXTERNAL_PREF_DOWNLOAD,
extensions::Extension::FROM_WEBSTORE |
extensions::Extension::WAS_INSTALLED_BY_DEFAULT));
EXPECT_CALL(visitor, OnExternalExtensionFileFound(_, _, _, _, _, _))
.Times(0);
EXPECT_CALL(visitor, OnExternalExtensionUpdateUrlFound(_, _, _, _, _, _))
.Times(0);
EXPECT_CALL(visitor, OnExternalProviderReady(_))
.Times(1);
RunUntilIdle();
loader->StartLoading();
Mock::VerifyAndClearExpectations(&visitor);
RunUntilIdle();
EXPECT_FALSE(doc->IsReady());
}
TEST_F(ServicesCustomizationDocumentTest, DefaultApps) {
AddCustomizationIdToVp(kDummyCustomizationID);
AddExpectedManifest(kDummyCustomizationID, kGoodServicesManifest);
ServicesCustomizationDocument* doc =
ServicesCustomizationDocument::GetInstance();
EXPECT_FALSE(doc->IsReady());
scoped_ptr<TestingProfile> profile = CreateProfile();
extensions::ExternalLoader* loader = doc->CreateExternalLoader(profile.get());
EXPECT_TRUE(loader);
app_list::AppListSyncableServiceFactory::GetInstance()->
SetTestingFactoryAndUse(
profile.get(),
&app_list::AppListSyncableServiceFactory::BuildInstanceFor);
MockExternalProviderVisitor visitor;
scoped_ptr<extensions::ExternalProviderImpl> provider(
new extensions::ExternalProviderImpl(
&visitor,
loader,
profile.get(),
extensions::Manifest::EXTERNAL_PREF,
extensions::Manifest::EXTERNAL_PREF_DOWNLOAD,
extensions::Extension::FROM_WEBSTORE |
extensions::Extension::WAS_INSTALLED_BY_DEFAULT));
EXPECT_CALL(visitor, OnExternalExtensionFileFound(_, _, _, _, _, _))
.Times(0);
EXPECT_CALL(visitor, OnExternalExtensionUpdateUrlFound(_, _, _, _, _, _))
.Times(0);
EXPECT_CALL(visitor, OnExternalProviderReady(_))
.Times(1);
loader->StartLoading();
Mock::VerifyAndClearExpectations(&visitor);
EXPECT_CALL(visitor, OnExternalExtensionFileFound(_, _, _, _, _, _))
.Times(0);
EXPECT_CALL(visitor, OnExternalExtensionUpdateUrlFound(_, _, _, _, _, _))
.Times(2);
EXPECT_CALL(visitor, OnExternalProviderReady(_))
.Times(1);
RunUntilIdle();
EXPECT_TRUE(doc->IsReady());
app_list::AppListSyncableService* service =
app_list::AppListSyncableServiceFactory::GetForProfile(profile.get());
ASSERT_TRUE(service);
EXPECT_EQ(service->GetOemFolderNameForTest(), "EN OEM Name");
}
TEST_F(ServicesCustomizationDocumentTest, CustomizationManifestNotFound) {
AddCustomizationIdToVp(kDummyCustomizationID);
AddManifestNotFound(kDummyCustomizationID);
ServicesCustomizationDocument* doc =
ServicesCustomizationDocument::GetInstance();
EXPECT_FALSE(doc->IsReady());
scoped_ptr<TestingProfile> profile = CreateProfile();
extensions::ExternalLoader* loader = doc->CreateExternalLoader(profile.get());
EXPECT_TRUE(loader);
MockExternalProviderVisitor visitor;
scoped_ptr<extensions::ExternalProviderImpl> provider(
new extensions::ExternalProviderImpl(
&visitor,
loader,
profile.get(),
extensions::Manifest::EXTERNAL_PREF,
extensions::Manifest::EXTERNAL_PREF_DOWNLOAD,
extensions::Extension::FROM_WEBSTORE |
extensions::Extension::WAS_INSTALLED_BY_DEFAULT));
EXPECT_CALL(visitor, OnExternalExtensionFileFound(_, _, _, _, _, _))
.Times(0);
EXPECT_CALL(visitor, OnExternalExtensionUpdateUrlFound(_, _, _, _, _, _))
.Times(0);
EXPECT_CALL(visitor, OnExternalProviderReady(_))
.Times(1);
loader->StartLoading();
Mock::VerifyAndClearExpectations(&visitor);
EXPECT_CALL(visitor, OnExternalExtensionFileFound(_, _, _, _, _, _))
.Times(0);
EXPECT_CALL(visitor, OnExternalExtensionUpdateUrlFound(_, _, _, _, _, _))
.Times(0);
EXPECT_CALL(visitor, OnExternalProviderReady(_))
.Times(1);
RunUntilIdle();
EXPECT_TRUE(doc->IsReady());
}
}