1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
|
// Copyright 2018 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "chrome/browser/ash/smb_client/smb_file_system_id.h"
#include <string>
#include "base/files/file_path.h"
#include "base/strings/strcat.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"
using ::testing::MatchesRegex;
namespace ash::smb_client {
namespace {
// gmock "regex" support is very basic and doesn't support [] or + operations.
// So write out the \w 16 times. This will incorrectly match _, but we can live
// with that.
constexpr char kRandomIdRegex[] =
"^\\w\\w\\w\\w\\w\\w\\w\\w\\w\\w\\w\\w\\w\\w\\w\\w";
} // namespace
constexpr char kTestUsername[] = "foobar";
constexpr char kTestWorkgroup[] = "accounts.baz.com";
class SmbFileSystemIdTest : public testing::Test {
public:
SmbFileSystemIdTest() = default;
SmbFileSystemIdTest(const SmbFileSystemIdTest&) = delete;
SmbFileSystemIdTest& operator=(const SmbFileSystemIdTest&) = delete;
~SmbFileSystemIdTest() override = default;
};
TEST_F(SmbFileSystemIdTest, ShouldCreateFileSystemIdCorrectly) {
const base::FilePath share_path("smb://192.168.0.0/test");
EXPECT_THAT(
CreateFileSystemId(share_path, false /* is_kerberos */),
MatchesRegex(std::string(kRandomIdRegex) + "@@smb://192.168.0.0/test"));
EXPECT_THAT(CreateFileSystemId(share_path, true /* is_kerberos */),
MatchesRegex(std::string(kRandomIdRegex) +
"@@smb://192.168.0.0/test@@kerberos_chromad"));
}
TEST_F(SmbFileSystemIdTest, CreateFileSystemIdForUser) {
const base::FilePath share_path("smb://192.168.0.0/test");
EXPECT_THAT(CreateFileSystemIdForUser(share_path, kTestUsername),
MatchesRegex(std::string(kRandomIdRegex) +
"@@smb://192.168.0.0/test@@user=" + kTestUsername));
std::string user_workgroup =
base::StrCat({kTestUsername, "@", kTestWorkgroup});
EXPECT_THAT(CreateFileSystemIdForUser(share_path, user_workgroup),
MatchesRegex(std::string(kRandomIdRegex) +
"@@smb://192.168.0.0/test@@user=" + user_workgroup));
}
TEST_F(SmbFileSystemIdTest, ShouldParseSharePathCorrectly) {
// Note: These are still valid because existing users might have saved shares.
const std::string file_system_id_1 = "12@@smb://192.168.0.0/test";
const base::FilePath expected_share_path_1("smb://192.168.0.0/test");
const std::string file_system_id_2 =
"13@@smb://192.168.0.1/test@@kerberos_chromad";
const base::FilePath expected_share_path_2("smb://192.168.0.1/test");
const std::string file_system_id_3 =
"EFAFF3864D0FE389@@smb://192.168.0.1/test@@kerberos_chromad";
const base::FilePath expected_share_path_3("smb://192.168.0.1/test");
EXPECT_EQ(expected_share_path_1,
GetSharePathFromFileSystemId(file_system_id_1));
EXPECT_EQ(expected_share_path_2,
GetSharePathFromFileSystemId(file_system_id_2));
EXPECT_EQ(expected_share_path_3,
GetSharePathFromFileSystemId(file_system_id_3));
}
TEST_F(SmbFileSystemIdTest, IsKerberosChromadReturnsCorrectly) {
const std::string kerberos_file_system_id =
"13@@smb://192.168.0.1/test@@kerberos_chromad";
const std::string non_kerberos_file_system_id = "12@@smb://192.168.0.0/test";
EXPECT_TRUE(IsKerberosChromadFileSystemId(kerberos_file_system_id));
EXPECT_FALSE(IsKerberosChromadFileSystemId(non_kerberos_file_system_id));
}
TEST_F(SmbFileSystemIdTest, GetUserFromFileSystemId) {
const std::string file_system_id_1 = base::StrCat(
{"EFAFF3864D0FE389@@smb://192.168.0.1/test@@user=", kTestUsername});
const std::string user_workgroup =
base::StrCat({kTestUsername, "@", kTestWorkgroup});
const std::string file_system_id_2 = base::StrCat(
{"EFAFF3864D0FE389@@smb://192.168.0.1/test@@user=", user_workgroup});
std::optional<std::string> actual_user =
GetUserFromFileSystemId(file_system_id_1);
ASSERT_TRUE(actual_user);
EXPECT_EQ(kTestUsername, *actual_user);
actual_user = GetUserFromFileSystemId(file_system_id_2);
ASSERT_TRUE(actual_user);
EXPECT_EQ(user_workgroup, *actual_user);
}
TEST_F(SmbFileSystemIdTest, GetUserFromFileSystemId_NoUser) {
const std::string file_system_id_1 =
"EFAFF3864D0FE389@@smb://192.168.0.1/test";
const std::string file_system_id_2 =
"EFAFF3864D0FE389@@smb://192.168.0.1/test@@kerberos_chromad";
const std::string file_system_id_3 =
"EFAFF3864D0FE389@@smb://192.168.0.1/test@@unrecognised_field";
EXPECT_FALSE(GetUserFromFileSystemId(file_system_id_1));
EXPECT_FALSE(GetUserFromFileSystemId(file_system_id_2));
EXPECT_FALSE(GetUserFromFileSystemId(file_system_id_3));
}
} // namespace ash::smb_client
|
