1 // Copyright (c) 2012 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.
13 #include "base/file_util.h"
14 #include "base/files/file_path.h"
15 #include "base/files/scoped_temp_dir.h"
16 #include "base/logging.h"
17 #include "base/strings/string_split.h"
18 #include "content/browser/download/file_metadata_linux.h"
19 #include "testing/gtest/include/gtest/gtest.h"
25 using std::istringstream
;
29 class FileMetadataLinuxTest
: public testing::Test
{
31 FileMetadataLinuxTest()
32 : source_url_("http://www.source.com"),
33 referrer_url_("http://www.referrer.com"),
34 is_xattr_supported_(false) {}
36 const base::FilePath
& test_file() const {
40 const GURL
& source_url() const {
44 const GURL
& referrer_url() const {
48 bool is_xattr_supported() const {
49 return is_xattr_supported_
;
53 virtual void SetUp() OVERRIDE
{
54 ASSERT_TRUE(temp_dir_
.CreateUniqueTempDir());
55 ASSERT_TRUE(base::CreateTemporaryFileInDir(temp_dir_
.path(), &test_file_
));
56 int result
= setxattr(test_file_
.value().c_str(),
57 "user.test", "test", 4, 0);
58 is_xattr_supported_
= (!result
) || (errno
!= ENOTSUP
);
59 if (!is_xattr_supported_
) {
60 VLOG(0) << "Test will be skipped because extended attributes are not "
61 << "supported on this OS/file system.";
65 void CheckExtendedAttributeValue(const string attr_name
,
66 const string expected_value
) const {
67 ssize_t len
= getxattr(test_file().value().c_str(), attr_name
.c_str(),
69 if (len
<= static_cast<ssize_t
>(0)) {
70 FAIL() << "Attribute '" << attr_name
<< "' does not exist";
72 char* buffer
= new char[len
];
73 len
= getxattr(test_file().value().c_str(), attr_name
.c_str(), buffer
, len
);
74 EXPECT_EQ(expected_value
.size(), static_cast<size_t>(len
));
75 string
real_value(buffer
, len
);
77 EXPECT_EQ(expected_value
, real_value
);
80 void GetExtendedAttributeNames(vector
<string
>* attr_names
) const {
81 ssize_t len
= listxattr(test_file().value().c_str(), NULL
, 0);
82 if (len
<= static_cast<ssize_t
>(0)) return;
83 char* buffer
= new char[len
];
84 len
= listxattr(test_file().value().c_str(), buffer
, len
);
86 base::SplitString(string(buffer
, len
), '\0', attr_names
);
90 void VerifyAttributesAreSetCorrectly() const {
91 vector
<string
> attr_names
;
92 GetExtendedAttributeNames(&attr_names
);
94 // Check if the attributes are set on the file
95 vector
<string
>::const_iterator pos
= find(attr_names
.begin(),
96 attr_names
.end(), kSourceURLAttrName
);
97 EXPECT_NE(pos
, attr_names
.end());
98 pos
= find(attr_names
.begin(), attr_names
.end(), kReferrerURLAttrName
);
99 EXPECT_NE(pos
, attr_names
.end());
101 // Check if the attribute values are set correctly
102 CheckExtendedAttributeValue(kSourceURLAttrName
, source_url().spec());
103 CheckExtendedAttributeValue(kReferrerURLAttrName
, referrer_url().spec());
107 base::ScopedTempDir temp_dir_
;
108 base::FilePath test_file_
;
111 bool is_xattr_supported_
;
114 TEST_F(FileMetadataLinuxTest
, CheckMetadataSetCorrectly
) {
115 if (!is_xattr_supported()) return;
116 AddOriginMetadataToFile(test_file(), source_url(), referrer_url());
117 VerifyAttributesAreSetCorrectly();
120 TEST_F(FileMetadataLinuxTest
, SetMetadataMultipleTimes
) {
121 if (!is_xattr_supported()) return;
122 GURL
dummy_url("http://www.dummy.com");
123 AddOriginMetadataToFile(test_file(), dummy_url
, dummy_url
);
124 AddOriginMetadataToFile(test_file(), source_url(), referrer_url());
125 VerifyAttributesAreSetCorrectly();
128 TEST_F(FileMetadataLinuxTest
, InvalidSourceURLTest
) {
129 if (!is_xattr_supported()) return;
131 vector
<string
> attr_names
;
132 AddOriginMetadataToFile(test_file(), invalid_url
, referrer_url());
133 GetExtendedAttributeNames(&attr_names
);
134 EXPECT_EQ(attr_names
.end(), find(attr_names
.begin(), attr_names
.end(),
135 kSourceURLAttrName
));
136 CheckExtendedAttributeValue(kReferrerURLAttrName
, referrer_url().spec());
139 TEST_F(FileMetadataLinuxTest
, InvalidReferrerURLTest
) {
140 if (!is_xattr_supported()) return;
142 vector
<string
> attr_names
;
143 AddOriginMetadataToFile(test_file(), source_url(), invalid_url
);
144 GetExtendedAttributeNames(&attr_names
);
145 EXPECT_EQ(attr_names
.end(), find(attr_names
.begin(), attr_names
.end(),
146 kReferrerURLAttrName
));
147 CheckExtendedAttributeValue(kSourceURLAttrName
, source_url().spec());
150 TEST_F(FileMetadataLinuxTest
, InvalidURLsTest
) {
151 if (!is_xattr_supported()) return;
153 vector
<string
> attr_names
;
154 AddOriginMetadataToFile(test_file(), invalid_url
, invalid_url
);
155 GetExtendedAttributeNames(&attr_names
);
156 EXPECT_EQ(attr_names
.end(), find(attr_names
.begin(), attr_names
.end(),
157 kSourceURLAttrName
));
158 EXPECT_EQ(attr_names
.end(), find(attr_names
.begin(), attr_names
.end(),
159 kReferrerURLAttrName
));
163 } // namespace content