[StorageMonitor] Move gallery name generation to StorageInfo.

chrome/browser/storage_monitor/storage_info_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/strings/utf_string_conversions.h"
+#include "chrome/browser/storage_monitor/media_storage_util.h"
 #include "chrome/browser/storage_monitor/storage_info.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 const char kMtpDeviceId[] = "mtp:VendorModelSerial:ABC:1233:1237912873";
 const char kUniqueId[] = "VendorModelSerial:ABC:1233:1237912873";
 const char kImageCaptureDeviceId[] = "ic:xyz";
 
 // Test to verify |MakeDeviceId| functionality using a sample
 // mtp device unique id.
 TEST(StorageInfoTest, MakeMtpDeviceId) {
@@ skipping 12 matching lines @@
   EXPECT_EQ(StorageInfo::MTP_OR_PTP, type);
 }
 
 TEST(StorageInfoTest, TestImageCaptureDeviceId) {
   StorageInfo::Type type;
   std::string id;
   ASSERT_TRUE(StorageInfo::CrackDeviceId(kImageCaptureDeviceId, &type, &id));
   EXPECT_EQ(StorageInfo::MAC_IMAGE_CAPTURE, type);
   EXPECT_EQ("xyz", id);
 }
+
+TEST(StorageInfoTest, NameGenerationRemovableDevice) {
+  StorageInfo storageInfo(
+     StorageInfo::MakeDeviceId(StorageInfo::REMOVABLE_MASS_STORAGE_WITH_DCIM,
+                               "unique"),
+     ASCIIToUTF16("override"), // device_name
+     base::FilePath::StringType(), // no location
+     string16(), // no storage label
+     string16(), // no storage vendor
+     string16(), // no storage model
+     0 // 0 capacity
+  );
+  EXPECT_EQ(ASCIIToUTF16("override"),
+            storageInfo.GetDisplayName());
+  storageInfo.set_name(ASCIIToUTF16("o2"));
+  EXPECT_EQ(ASCIIToUTF16("o2"),
+            storageInfo.GetDisplayName());
+
+  storageInfo.set_storage_label(ASCIIToUTF16("vol"));
+  storageInfo.set_vendor_name(ASCIIToUTF16("vendor"));
+  storageInfo.set_model_name(ASCIIToUTF16("model"));
+  EXPECT_EQ(ASCIIToUTF16("o2"),
+            storageInfo.GetDisplayName());
+
+  storageInfo.set_name(string16());
+  EXPECT_EQ(ASCIIToUTF16("vol"),
+            storageInfo.GetDisplayName());
+  storageInfo.set_storage_label(string16());
+  EXPECT_EQ(ASCIIToUTF16("vendor, model"),
+            storageInfo.GetDisplayName());
+
+  storageInfo.set_total_size_in_bytes(1000000);
+  EXPECT_EQ(ASCIIToUTF16("977 KB vendor, model"),
+            storageInfo.GetDisplayName());
+
+  // Root directory
+  storageInfo.set_location(FILE_PATH_LITERAL(
+      "/"));
+  EXPECT_EQ(ASCIIToUTF16("977 KB vendor, model"),
+            storageInfo.GetDisplayName());
+  // Subfolder directory
+  storageInfo.set_location(FILE_PATH_LITERAL(
+      "/sub/path"));
+  EXPECT_EQ(ASCIIToUTF16("path - 977 KB vendor, model"),
+            storageInfo.GetDisplayName());
+  // drive letter directory
+  storageInfo.set_location(FILE_PATH_LITERAL(
+      "C:\\"));
+  EXPECT_EQ(ASCIIToUTF16("\\ - 977 KB vendor, model"),
+            storageInfo.GetDisplayName());
+}
+
+TEST(StorageInfoTest, NameGenerationFixedDevice) {
+  std::string device_id = StorageInfo::MakeDeviceId(
+      StorageInfo::FIXED_MASS_STORAGE, "/path/to/gallery");
+  StorageInfo storageInfo(
+     device_id,
+     string16(), // no device_name
+     MediaStorageUtil::FindDevicePathById(device_id).value(),
+     //base::FilePath::StringType(), // no location
+     string16(), // no storage label
+     string16(), // no storage vendor
+     string16(), // no storage model
+     0 // 0 capacity
+  );
+  std::string galleryName("/path/to/gallery");
+#if defined(OS_CHROMEOS)
+  galleryName = "gallery";
+#endif
+  EXPECT_EQ(ASCIIToUTF16(galleryName), storageInfo.GetDisplayName());
+
+  storageInfo.set_name(ASCIIToUTF16("override"));
+  EXPECT_EQ(ASCIIToUTF16("override"),
+            storageInfo.GetDisplayName());
+
+  storageInfo.set_name(string16());
+  storageInfo.set_storage_label(ASCIIToUTF16("label"));
+  EXPECT_EQ(ASCIIToUTF16(galleryName), storageInfo.GetDisplayName());
+
+  galleryName = "/path/to/gallery/sub/gallery2";
+  storageInfo.set_location(FILE_PATH_LITERAL(
+      "/path/to/gallery/sub/gallery2"));
+#if defined(OS_CHROMEOS)
+  galleryName = "gallery2";
+#endif
+  EXPECT_EQ(ASCIIToUTF16(galleryName), storageInfo.GetDisplayName());
+}