1 // Copyright 2013 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 #include "chromeos/dbus/cros_disks_client.h"
6
7 #include "base/memory/scoped_ptr.h"
8 #include "dbus/message.h"
9 #include "testing/gtest/include/gtest/gtest.h"
10 #include "third_party/cros_system_api/dbus/service_constants.h"
11
12 namespace chromeos {
13
14 namespace {
15
16 // Appends a boolean entry to a dictionary of type "a{sv}"
AppendBoolDictEntry(dbus::MessageWriter * array_writer,const std::string & key,bool value)17 void AppendBoolDictEntry(dbus::MessageWriter* array_writer,
18 const std::string& key,
19 bool value) {
20 dbus::MessageWriter entry_writer(NULL);
21 array_writer->OpenDictEntry(&entry_writer);
22 entry_writer.AppendString(key);
23 entry_writer.AppendVariantOfBool(value);
24 array_writer->CloseContainer(&entry_writer);
25 }
26
27 // Appends a string entry to a dictionary of type "a{sv}"
AppendStringDictEntry(dbus::MessageWriter * array_writer,const std::string & key,const std::string & value)28 void AppendStringDictEntry(dbus::MessageWriter* array_writer,
29 const std::string& key,
30 const std::string& value) {
31 dbus::MessageWriter entry_writer(NULL);
32 array_writer->OpenDictEntry(&entry_writer);
33 entry_writer.AppendString(key);
34 entry_writer.AppendVariantOfString(value);
35 array_writer->CloseContainer(&entry_writer);
36 }
37
38 } // namespace
39
TEST(CrosDisksClientTest,DiskInfo)40 TEST(CrosDisksClientTest, DiskInfo) {
41 const std::string kDeviceFile = "/dev/sdb1";
42 const bool kDeviceIsDrive = true;
43 const bool kDeviceIsMediaAvailable = true;
44 const bool kDeviceIsMounted = true;
45 const bool kDeviceIsOnBootDevice = true;
46 const bool kDeviceIsReadOnly = true;
47 const uint32 kDeviceMediaType = cros_disks::DEVICE_MEDIA_SD;
48 const std::string kMountPath = "/media/removable/UNTITLED";
49 const bool kDevicePresentationHide = false;
50 const uint64 kDeviceSize = 16005464064;
51 const std::string kDriveModel = "DriveModel";
52 const std::string kIdLabel = "UNTITLED";
53 const std::string kIdUuid = "XXXX-YYYY";
54 const std::string kNativePath = "/sys/devices/.../sdb/sdb1";
55 const std::string kProductId = "1234";
56 const std::string kProductName = "Product Name";
57 const std::string kVendorId = "0000";
58 const std::string kVendorName = "Vendor Name";
59
60 // Construct a fake response of GetDeviceProperties().
61 scoped_ptr<dbus::Response> response(dbus::Response::CreateEmpty());
62 {
63 dbus::MessageWriter writer(response.get());
64 dbus::MessageWriter array_writer(NULL);
65 writer.OpenArray("{sv}", &array_writer);
66
67 AppendStringDictEntry(&array_writer, cros_disks::kDeviceFile, kDeviceFile);
68 AppendBoolDictEntry(&array_writer, cros_disks::kDeviceIsDrive,
69 kDeviceIsDrive);
70 AppendBoolDictEntry(&array_writer, cros_disks::kDeviceIsMediaAvailable,
71 kDeviceIsMediaAvailable);
72 AppendBoolDictEntry(&array_writer, cros_disks::kDeviceIsMounted,
73 kDeviceIsMounted);
74 AppendBoolDictEntry(&array_writer, cros_disks::kDeviceIsOnBootDevice,
75 kDeviceIsOnBootDevice);
76 AppendBoolDictEntry(&array_writer, cros_disks::kDeviceIsReadOnly,
77 kDeviceIsReadOnly);
78 {
79 dbus::MessageWriter entry_writer(NULL);
80 array_writer.OpenDictEntry(&entry_writer);
81 entry_writer.AppendString(cros_disks::kDeviceMediaType);
82 entry_writer.AppendVariantOfUint32(kDeviceMediaType);
83 array_writer.CloseContainer(&entry_writer);
84 }
85 {
86 std::vector<std::string> mounted_paths;
87 mounted_paths.push_back(kMountPath);
88
89 dbus::MessageWriter entry_writer(NULL);
90 array_writer.OpenDictEntry(&entry_writer);
91 entry_writer.AppendString(cros_disks::kDeviceMountPaths);
92 dbus::MessageWriter variant_writer(NULL);
93 entry_writer.OpenVariant("as", &variant_writer);
94 variant_writer.AppendArrayOfStrings(mounted_paths);
95 entry_writer.CloseContainer(&variant_writer);
96 array_writer.CloseContainer(&entry_writer);
97 }
98 AppendBoolDictEntry(&array_writer, cros_disks::kDevicePresentationHide,
99 kDevicePresentationHide);
100 {
101 dbus::MessageWriter entry_writer(NULL);
102 array_writer.OpenDictEntry(&entry_writer);
103 entry_writer.AppendString(cros_disks::kDeviceSize);
104 entry_writer.AppendVariantOfUint64(kDeviceSize);
105 array_writer.CloseContainer(&entry_writer);
106 }
107 AppendStringDictEntry(&array_writer, cros_disks::kDriveModel, kDriveModel);
108 AppendStringDictEntry(&array_writer, cros_disks::kIdLabel, kIdLabel);
109 AppendStringDictEntry(&array_writer, cros_disks::kIdUuid, kIdUuid);
110 AppendStringDictEntry(&array_writer, cros_disks::kNativePath, kNativePath);
111 AppendStringDictEntry(&array_writer, cros_disks::kProductId, kProductId);
112 AppendStringDictEntry(&array_writer, cros_disks::kProductName,
113 kProductName);
114 AppendStringDictEntry(&array_writer, cros_disks::kVendorId, kVendorId);
115 AppendStringDictEntry(&array_writer, cros_disks::kVendorName, kVendorName);
116
117 writer.CloseContainer(&array_writer);
118 }
119
120 // Construct DiskInfo.
121 DiskInfo result(kDeviceFile, response.get());
122 EXPECT_EQ(kDeviceFile, result.device_path());
123 EXPECT_EQ(kDeviceIsDrive, result.is_drive());
124 EXPECT_EQ(kDeviceIsReadOnly, result.is_read_only());
125 // Use EXPECT_TRUE(kDevicePresentationHide == result.is_hidden()) instead of
126 // EXPECT_EQ(kDevicePresentationHide, result.is_hidden()) as gcc 4.7 issues
127 // the following warning on EXPECT_EQ(false, x), which is turned into an error
128 // with -Werror=conversion-null:
129 //
130 // converting 'false' to pointer type for argument 1 of
131 // 'char testing::internal::IsNullLiteralHelper(testing::internal::Secret*)'
132 EXPECT_TRUE(kDevicePresentationHide == result.is_hidden());
133 EXPECT_EQ(kDeviceIsMediaAvailable, result.has_media());
134 EXPECT_EQ(kDeviceIsOnBootDevice, result.on_boot_device());
135 EXPECT_EQ(kNativePath, result.system_path());
136 EXPECT_EQ(kDeviceFile, result.file_path());
137 EXPECT_EQ(kVendorId, result.vendor_id());
138 EXPECT_EQ(kVendorName, result.vendor_name());
139 EXPECT_EQ(kProductId, result.product_id());
140 EXPECT_EQ(kProductName, result.product_name());
141 EXPECT_EQ(kDriveModel, result.drive_label());
142 EXPECT_EQ(kIdLabel, result.label());
143 EXPECT_EQ(kIdUuid, result.uuid());
144 EXPECT_EQ(kDeviceSize, result.total_size_in_bytes());
145 EXPECT_EQ(DEVICE_TYPE_SD, result.device_type());
146 EXPECT_EQ(kMountPath, result.mount_path());
147 }
148
149 } // namespace chromeos
150