[chromedriver] Apply page load timeout to slow cross-process navigations

chrome/test/chromedriver/util_unittest.cc

 // Copyright (c) 2013 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 <string>
 
 #include "base/base64.h"
 #include "base/files/file_path.h"
 #include "base/files/file_util.h"
 #include "base/files/scoped_temp_dir.h"
 #include "chrome/test/chromedriver/chrome/status.h"
 #include "chrome/test/chromedriver/util.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
+using base::TimeDelta;
+
 TEST(UnzipSoleFile, Entry) {
   base::ScopedTempDir temp_dir;
   ASSERT_TRUE(temp_dir.CreateUniqueTempDir());
   std::string data;
   // A zip entry sent from a Java WebDriver client (v2.20) that contains a
   // file with the contents "COW\n".
   const char kBase64ZipEntry[] =
       "UEsDBBQACAAIAJpyXEAAAAAAAAAAAAAAAAAEAAAAdGVzdHP2D+"
       "cCAFBLBwi/wAzGBgAAAAQAAAA=";
   ASSERT_TRUE(base::Base64Decode(kBase64ZipEntry, &data));
@@ skipping 16 matching lines @@
       "wAzGBAAAAAQAAAADAAAAAAAAAAAAAACggQAAAABtb29QSwUGAAAAAAEAAQAxAAAAJQAAAAA"
       "A";
   ASSERT_TRUE(base::Base64Decode(kBase64ZipArchive, &data));
   base::FilePath file;
   Status status = UnzipSoleFile(temp_dir.path(), data, &file);
   ASSERT_EQ(kOk, status.code()) << status.message();
   std::string contents;
   ASSERT_TRUE(base::ReadFileToString(file, &contents));
   ASSERT_STREQ("COW\n", contents.c_str());
 }
+
+TEST(TimeoutTest, Basics) {
+  Timeout timeout;
+  EXPECT_FALSE(timeout.is_set());
+  EXPECT_FALSE(timeout.IsExpired());
+  EXPECT_EQ(TimeDelta::Max(), timeout.GetDuration());
+  EXPECT_EQ(TimeDelta::Max(), timeout.GetRemainingTime());
+
+  timeout.SetDuration(TimeDelta());
+  EXPECT_TRUE(timeout.is_set());
+  EXPECT_TRUE(timeout.IsExpired());
+  EXPECT_EQ(TimeDelta(), timeout.GetDuration());
+  EXPECT_GE(TimeDelta(), timeout.GetRemainingTime());
+}
+
+TEST(TimeoutTest, Derive) {
+  Timeout timeout(TimeDelta::FromMinutes(5));
+  EXPECT_TRUE(timeout.is_set());
+  EXPECT_FALSE(timeout.IsExpired());
+  EXPECT_EQ(TimeDelta::FromMinutes(5), timeout.GetDuration());
+  EXPECT_GE(TimeDelta::FromMinutes(5), timeout.GetRemainingTime());
+
+  Timeout small = Timeout(TimeDelta::FromSeconds(10), &timeout);
+  EXPECT_TRUE(small.is_set());
+  EXPECT_FALSE(small.IsExpired());
+  EXPECT_EQ(TimeDelta::FromSeconds(10), small.GetDuration());
+
+  Timeout large = Timeout(TimeDelta::FromMinutes(30), &timeout);
+  EXPECT_TRUE(large.is_set());
+  EXPECT_FALSE(large.IsExpired());
+  EXPECT_GE(timeout.GetDuration(), large.GetDuration());
+}
+
+TEST(TimeoutTest, DeriveExpired) {
+  Timeout timeout((TimeDelta()));
+  EXPECT_TRUE(timeout.is_set());
+  EXPECT_TRUE(timeout.IsExpired());
+
+  Timeout derived = Timeout(TimeDelta::FromSeconds(10), &timeout);
+  EXPECT_TRUE(derived.is_set());
+  EXPECT_TRUE(derived.IsExpired());
+  EXPECT_GE(TimeDelta(), derived.GetDuration());
+}