| OLD | NEW |
| 1 // Copyright 2012 The Chromium Authors. All rights reserved. | 1 // Copyright 2012 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #import "ios/chrome/browser/snapshots/snapshot_cache.h" | 5 #import "ios/chrome/browser/snapshots/snapshot_cache.h" |
| 6 | 6 |
| 7 #import <Foundation/Foundation.h> | 7 #import <Foundation/Foundation.h> |
| 8 | 8 |
| 9 #include "base/files/file_path.h" | 9 #include "base/files/file_path.h" |
| 10 #include "base/files/file_util.h" | 10 #include "base/files/file_util.h" |
| 11 #include "base/format_macros.h" | 11 #include "base/format_macros.h" |
| 12 #include "base/location.h" | 12 #include "base/location.h" |
| 13 #include "base/mac/bind_objc_block.h" | 13 #include "base/mac/bind_objc_block.h" |
| 14 #include "base/mac/scoped_cftyperef.h" | 14 #include "base/mac/scoped_cftyperef.h" |
| 15 #include "base/mac/scoped_nsautorelease_pool.h" | 15 #include "base/mac/scoped_nsautorelease_pool.h" |
| 16 #include "base/mac/scoped_nsobject.h" | 16 #include "base/mac/scoped_nsobject.h" |
| 17 #include "base/run_loop.h" | 17 #include "base/run_loop.h" |
| 18 #include "base/strings/sys_string_conversions.h" | 18 #include "base/strings/sys_string_conversions.h" |
| 19 #include "base/task_scheduler/task_scheduler.h" | 19 #include "base/threading/sequenced_worker_pool.h" |
| 20 #include "base/time/time.h" | 20 #include "base/time/time.h" |
| 21 #import "ios/chrome/browser/snapshots/snapshot_cache_internal.h" | 21 #import "ios/chrome/browser/snapshots/snapshot_cache_internal.h" |
| 22 #include "ios/web/public/test/test_web_thread_bundle.h" | 22 #include "ios/web/public/test/test_web_thread_bundle.h" |
| 23 #include "ios/web/public/web_thread.h" |
| 23 #include "testing/gtest/include/gtest/gtest.h" | 24 #include "testing/gtest/include/gtest/gtest.h" |
| 24 #include "testing/gtest_mac.h" | 25 #include "testing/gtest_mac.h" |
| 25 #include "testing/platform_test.h" | 26 #include "testing/platform_test.h" |
| 26 | 27 |
| 27 static const NSUInteger kSessionCount = 10; | 28 static const NSUInteger kSessionCount = 10; |
| 28 static const NSUInteger kSnapshotPixelSize = 8; | 29 static const NSUInteger kSnapshotPixelSize = 8; |
| 29 | 30 |
| 30 namespace { | 31 namespace { |
| 31 | 32 |
| 32 class SnapshotCacheTest : public PlatformTest { | 33 class SnapshotCacheTest : public PlatformTest { |
| (...skipping 43 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 76 CGContextSetRGBStrokeColor(context, r, g, b, 1.0); | 77 CGContextSetRGBStrokeColor(context, r, g, b, 1.0); |
| 77 CGContextSetRGBFillColor(context, r, g, b, 1.0); | 78 CGContextSetRGBFillColor(context, r, g, b, 1.0); |
| 78 CGContextFillRect( | 79 CGContextFillRect( |
| 79 context, CGRectMake(0.0, 0.0, kSnapshotPixelSize, kSnapshotPixelSize)); | 80 context, CGRectMake(0.0, 0.0, kSnapshotPixelSize, kSnapshotPixelSize)); |
| 80 return UIGraphicsGetImageFromCurrentImageContext(); | 81 return UIGraphicsGetImageFromCurrentImageContext(); |
| 81 } | 82 } |
| 82 | 83 |
| 83 // Flushes all the runloops internally used by the snapshot cache. | 84 // Flushes all the runloops internally used by the snapshot cache. |
| 84 void FlushRunLoops() { | 85 void FlushRunLoops() { |
| 85 base::RunLoop().RunUntilIdle(); | 86 base::RunLoop().RunUntilIdle(); |
| 86 base::TaskScheduler::GetInstance()->FlushForTesting(); | 87 web::WebThread::GetBlockingPool()->FlushForTesting(); |
| 87 base::RunLoop().RunUntilIdle(); | 88 base::RunLoop().RunUntilIdle(); |
| 88 } | 89 } |
| 89 | 90 |
| 90 // This function removes the snapshots both from dictionary and from disk. | 91 // This function removes the snapshots both from dictionary and from disk. |
| 91 void ClearDumpedImages() { | 92 void ClearDumpedImages() { |
| 92 SnapshotCache* cache = GetSnapshotCache(); | 93 SnapshotCache* cache = GetSnapshotCache(); |
| 93 | 94 |
| 94 NSString* sessionID; | 95 NSString* sessionID; |
| 95 for (sessionID in testSessions_.get()) | 96 for (sessionID in testSessions_.get()) |
| 96 [cache removeImageWithSessionID:sessionID]; | 97 [cache removeImageWithSessionID:sessionID]; |
| (...skipping 466 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 563 EXPECT_TRUE(base::PathExists(retinaFile)); | 564 EXPECT_TRUE(base::PathExists(retinaFile)); |
| 564 | 565 |
| 565 // Delete the image. | 566 // Delete the image. |
| 566 [cache removeImageWithSessionID:kSession]; | 567 [cache removeImageWithSessionID:kSession]; |
| 567 FlushRunLoops(); // ensure the file is removed. | 568 FlushRunLoops(); // ensure the file is removed. |
| 568 | 569 |
| 569 EXPECT_FALSE(base::PathExists(retinaFile)); | 570 EXPECT_FALSE(base::PathExists(retinaFile)); |
| 570 } | 571 } |
| 571 | 572 |
| 572 } // namespace | 573 } // namespace |
| OLD | NEW |