Fix for URL opening code

chrome/browser/sessions/session_restore_browsertest.cc

 // Copyright 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include <stddef.h>
 
+#include <algorithm>
 #include <set>
 #include <vector>
 
 #include "base/base_switches.h"
 #include "base/command_line.h"
 #include "base/files/file_path.h"
 #include "base/files/scoped_temp_dir.h"
 #include "base/macros.h"
 #include "base/memory/memory_pressure_listener.h"
 #include "base/memory/ptr_util.h"
@@ skipping 1598 matching lines @@
   // automatically at the start of the test.
   for (size_t i = 1; i < web_contents().size(); i++) {
     GURL expected_url = GURL(kUrls[activation_order[kExpectedNumTabs - i]]);
     ASSERT_EQ(expected_url, web_contents()[i]->GetLastCommittedURL());
     if (i > 0) {
       ASSERT_GT(web_contents()[i - 1]->GetLastActiveTime(),
                 web_contents()[i]->GetLastActiveTime());
     }
   }
 }
+
+namespace {
+
+const char kStartupURL[] = "http://example.com/";
+const char kSessionURL[] = "http://localhost/";
+
+template <class Container>
+bool BeginsWith(const Container& container, const Container& prefix) {

[sky, 2017/05/04 23:40:46]

Why are you using a template here? Can't you use substr()?

[eugenebng, 2017/05/15 06:17:09]

Done.

+  return container.size() >= prefix.size() &&
+         equal(prefix.begin(), prefix.end(), container.begin());
+}
+
+}  // namespace
+
+class SessionRestoreWithStartupURLTest : public SessionRestoreTest {
+ protected:
+  void SetUpCommandLine(base::CommandLine* command_line) override {
+    // Add startup URL to command line
+    const testing::TestInfo* const test_info =
+        testing::UnitTest::GetInstance()->current_test_info();
+    std::string test_name = test_info->name();
+    // Do not add startup URL in PRE_ test to prevent it from opening twice
+    // in main test body.
+    if (!BeginsWith(test_name, std::string("PRE_"))) {
+      command_line->AppendArg(kStartupURL);
+    }
+    SessionRestoreTest::SetUpCommandLine(command_line);
+  }
+};

[sky, 2017/05/04 23:40:46]

private: DISALLOW...

[eugenebng, 2017/05/15 06:17:08]

Done.

+
+IN_PROC_BROWSER_TEST_F(SessionRestoreWithStartupURLTest,
+                       PRE_StartupURLsWithSessionRestore) {
+  // Prepare a session to restore in main test body.
+  ui_test_utils::NavigateToURL(browser(), GURL(kSessionURL));
+}
+
+IN_PROC_BROWSER_TEST_F(SessionRestoreWithStartupURLTest,
+                       StartupURLsWithSessionRestore) {
+  // Check that browser is started with restored session and
+  // tab with startup URL next to it.
+  // ui_test_utils::NavigateToURL(browser(),GURL(kSessionURL));

[sky, 2017/05/04 23:40:46]

Why is this commented out?

[eugenebng, 2017/05/15 06:17:08]

Done.

+  TabStripModel* tab_strip = browser()->tab_strip_model();
+  ASSERT_EQ(2, tab_strip->count());
+  EXPECT_EQ(kSessionURL,
+            tab_strip->GetWebContentsAt(0)->GetURL().possibly_invalid_spec());
+  EXPECT_EQ(kStartupURL,
+            tab_strip->GetWebContentsAt(1)->GetURL().possibly_invalid_spec());

[eugenebng, 2017/05/03 16:05:30]

Added test.

+}