Add test cases for cobalt::loader::FetcherCache (#1832)

Test FetcherCache with corner cases, and try to catch crash issue. b/270993319
youtube · Apr 14, 2024 · 89a8945 · 89a8945
1 parent 8b22ec9
commit 89a8945
Show file tree

Hide file tree

Showing 3 changed files with 433 additions and 0 deletions.
diff --git a/cobalt/loader/BUILD.gn b/cobalt/loader/BUILD.gn
@@ -175,6 +175,7 @@ target(gtest_target_type, "loader_test") {
 
   sources = [
     "blob_fetcher_test.cc",
+    "fetcher_cache_test.cc",
     "fetcher_factory_test.cc",
     "fetcher_test.h",
     "file_fetcher_test.cc",
@@ -201,6 +202,7 @@ target(gtest_target_type, "loader_test") {
     "//testing/gmock",
     "//testing/gtest",
     "//third_party/ots:ots",
+    "//url",
   ]
 
   data_deps = [ ":copy_loader_test_data" ]

diff --git a/cobalt/loader/fetcher_cache.h b/cobalt/loader/fetcher_cache.h
@@ -42,6 +42,16 @@ class FetcherCache : public base::RefCountedThreadSafe<FetcherCache> {
   Loader::FetcherCreator GetFetcherCreator(
       const GURL& url, const Loader::FetcherCreator& real_fetcher_creator);
   void NotifyResourceRequested(const std::string& url);
+  size_t size() const {
+    DCHECK_CALLED_ON_VALID_THREAD(thread_checker_);
+    CHECK_EQ(thread_id_, SbThreadGetId());
+    return total_size_;
+  }
+  size_t capacity() const {
+    DCHECK_CALLED_ON_VALID_THREAD(thread_checker_);
+    CHECK_EQ(thread_id_, SbThreadGetId());
+    return capacity_;
+  }
 
   // To signal the imminent destruction of this object.  If everything is
   // working as expected, there shouldn't be any other reference of this object,