Upstream iOS UI utilities.

ios/chrome/browser/ui/snapshots_util_unittest.mm

Index: ios/chrome/browser/ui/snapshots_util_unittest.mm
diff --git a/ios/chrome/browser/ui/snapshots_util_unittest.mm b/ios/chrome/browser/ui/snapshots_util_unittest.mm
new file mode 100644
index 0000000000000000000000000000000000000000..71063021b18991b0c5e57a9e9acd4ff4bce929ba
--- /dev/null
+++ b/ios/chrome/browser/ui/snapshots_util_unittest.mm
@@ -0,0 +1,57 @@
+// Copyright 2011 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.
+
+#import "ios/chrome/browser/ui/snapshots_util.h"
+
+#import <Foundation/Foundation.h>
+#import <UIKit/UIKit.h>
+
+#include "base/ios/ios_util.h"
+#include "base/mac/foundation_util.h"
+#include "base/path_service.h"
+#include "base/strings/sys_string_conversions.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace {
+
+bool RegexMatchesOneSnapshotPath(NSString* regexString) {
+  NSRegularExpression* regex =
+      [NSRegularExpression regularExpressionWithPattern:regexString
+                                                options:0
+                                                  error:NULL];
+  std::vector<base::FilePath> snapshotsPaths;
+  GetSnapshotsPaths(&snapshotsPaths);
+  int numberOfMatches = 0;
+  for (const base::FilePath& path : snapshotsPaths) {
+    NSString* string =
+        [NSString stringWithCString:path.value().c_str()
+                           encoding:[NSString defaultCStringEncoding]];
+    if ([regex numberOfMatchesInString:string
+                               options:0
+                                 range:NSMakeRange(0, [string length])])
+      numberOfMatches++;
+  }
+  return numberOfMatches == 1;
+}
+
+TEST(SnapshotsUtilTest, TestSnapshotList) {
+  NSString* scaleModifier = @"";
+  CGFloat scale = [UIScreen mainScreen].scale;
+  if (scale > 1) {
+    scaleModifier = [NSString stringWithFormat:@"@%.0fx", scale];
+  }
+  NSString* path = @"Main";
+  if (base::ios::IsRunningOnIOS8OrLater()) {
+    path = base::SysUTF8ToNSString(base::mac::BaseBundleID());
+  }
+  NSString* filename = @"UIApplicationAutomaticSnapshotDefault-LandscapeRight";
+  NSString* regex = [NSString
+      stringWithFormat:@".*/%@/%@%@.png$", path, filename, scaleModifier];
+  EXPECT_FALSE(RegexMatchesOneSnapshotPath(@".*"));
+  EXPECT_FALSE(RegexMatchesOneSnapshotPath(@"foo"));
+  EXPECT_TRUE(RegexMatchesOneSnapshotPath(@".*LandscapeRight.*"));
+  EXPECT_TRUE(RegexMatchesOneSnapshotPath(regex));
+}
+
+}  // anonymous namespace