1// Copyright (c) 2012 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/common/chrome_content_client.h"
6
7#include "base/command_line.h"
8#include "base/strings/string_split.h"
9#include "content/public/common/content_switches.h"
10#include "testing/gtest/include/gtest/gtest.h"
11
12namespace {
13
14void CheckUserAgentStringOrdering(bool mobile_device) {
15  std::vector<std::string> pieces;
16
17  // Check if the pieces of the user agent string come in the correct order.
18  ChromeContentClient content_client;
19  std::string buffer = content_client.GetUserAgent();
20
21  base::SplitStringUsingSubstr(buffer, "Mozilla/5.0 (", &pieces);
22  ASSERT_EQ(2u, pieces.size());
23  buffer = pieces[1];
24  EXPECT_EQ("", pieces[0]);
25
26  base::SplitStringUsingSubstr(buffer, ") AppleWebKit/", &pieces);
27  ASSERT_EQ(2u, pieces.size());
28  buffer = pieces[1];
29  std::string os_str = pieces[0];
30
31  base::SplitStringUsingSubstr(buffer, " (KHTML, like Gecko) ", &pieces);
32  ASSERT_EQ(2u, pieces.size());
33  buffer = pieces[1];
34  std::string webkit_version_str = pieces[0];
35
36  base::SplitStringUsingSubstr(buffer, " Safari/", &pieces);
37  ASSERT_EQ(2u, pieces.size());
38  std::string product_str = pieces[0];
39  std::string safari_version_str = pieces[1];
40
41  // Not sure what can be done to better check the OS string, since it's highly
42  // platform-dependent.
43  EXPECT_TRUE(os_str.size() > 0);
44
45  // Check that the version numbers match.
46  EXPECT_TRUE(webkit_version_str.size() > 0);
47  EXPECT_TRUE(safari_version_str.size() > 0);
48  EXPECT_EQ(webkit_version_str, safari_version_str);
49
50  EXPECT_EQ(0u, product_str.find("Chrome/"));
51  if (mobile_device) {
52    // "Mobile" gets tacked on to the end for mobile devices, like phones.
53    const std::string kMobileStr = " Mobile";
54    EXPECT_EQ(kMobileStr,
55              product_str.substr(product_str.size() - kMobileStr.size()));
56  }
57}
58
59}  // namespace
60
61
62namespace chrome_common {
63
64TEST(ChromeContentClientTest, Basic) {
65#if !defined(OS_ANDROID)
66  CheckUserAgentStringOrdering(false);
67#else
68  const char* kArguments[] = {"chrome"};
69  CommandLine::Reset();
70  CommandLine::Init(1, kArguments);
71  CommandLine* command_line = CommandLine::ForCurrentProcess();
72
73  // Do it for regular devices.
74  ASSERT_FALSE(command_line->HasSwitch(switches::kUseMobileUserAgent));
75  CheckUserAgentStringOrdering(false);
76
77  // Do it for mobile devices.
78  command_line->AppendSwitch(switches::kUseMobileUserAgent);
79  ASSERT_TRUE(command_line->HasSwitch(switches::kUseMobileUserAgent));
80  CheckUserAgentStringOrdering(true);
81#endif
82}
83
84}  // namespace chrome_common
85