Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(903)

Unified Diff: remoting/test/refresh_token_store.cc

Issue 880273006: Adding the AccessTokenFetcher and Environment class to the app remoting test (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: Updating AccessTokenFetcher Unittests to use the FakeURLFetcherFactory Created 5 years, 10 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
« remoting/test/refresh_token_store.h ('K') | « remoting/test/refresh_token_store.h ('k') | no next file » | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
Index: remoting/test/refresh_token_store.cc
diff --git a/remoting/test/refresh_token_store.cc b/remoting/test/refresh_token_store.cc
new file mode 100644
index 0000000000000000000000000000000000000000..63e91b1225df2f1f7cb0c3685f4e6fe59c98b474
--- /dev/null
+++ b/remoting/test/refresh_token_store.cc
@@ -0,0 +1,133 @@
+// Copyright 2015 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 "remoting/test/refresh_token_store.h"
+
+#include "base/files/file_util.h"
+#include "base/logging.h"
+
+namespace {
+const char kTokenFileName[] = "refresh_token.txt";
+
+// Generates a FilePath using the user_name for use with the read/write
+// functions in this file.
Wez 2015/02/19 22:00:24 "Returns the FilePath of the token store file for
joedow 2015/02/20 02:58:36 Done.
+base::FilePath GetRefreshTokenDirPath(const std::string& user_name) {
Wez 2015/02/19 22:00:24 You never use the directory path on it's own; you
joedow 2015/02/20 02:58:36 I use the path w/o file name to see if the directo
Wez 2015/02/23 20:17:14 Ah yes. It may be clearer to construct the file f
+ base::FilePath refresh_token_dir_path;
+ if (!GetTempDir(&refresh_token_dir_path)) {
+ LOG(WARNING) << "Failed to retrieve temporary directory path.";
+ return base::FilePath();
+ }
+
+ // Build up the token path by appending the sub-directories.
Wez 2015/02/19 22:00:24 We can see that you're appending directories; no n
joedow 2015/02/20 02:58:36 Done.
+ refresh_token_dir_path = refresh_token_dir_path.Append("remoting");
+ refresh_token_dir_path = refresh_token_dir_path.Append("test_driver");
+ refresh_token_dir_path = refresh_token_dir_path.Append(user_name);
+
+ return refresh_token_dir_path;
+}
+
+} // namespace
+
+namespace remoting {
+namespace test {
+
+// Provides functionality to write a refresh token to a local folder on disk and
+// read it back during subsequent tool runs.
+class RefreshTokenStore : public RefreshTokenStoreInterface {
+ public:
+ RefreshTokenStore();
+ ~RefreshTokenStore() override;
+
+ // Reads the refresh token from a file in a user unique directory if it exists
+ // and returns the refresh token value if successful or empty string if not.
Wez 2015/02/19 22:00:24 This and the method below are inherited from the i
joedow 2015/02/20 02:58:36 Done.
+ std::string ReadRefreshTokenFromDisk(const std::string& user_name) override;
+
+ // Stores the given refresh_token in a folder on the user's local disk. It
+ // will create the directory and file if they do not exist and set permissions
+ // on the local file so that it is only readable by the current user. Returns
+ // true if the token was successfully written otherwise false.
+ bool WriteRefreshTokenToDisk(const std::string& user_name,
+ const std::string& refresh_token) override;
+};
+
+scoped_ptr<RefreshTokenStoreInterface> GetRefreshTokenStoreForDisk() {
+ return make_scoped_ptr<RefreshTokenStore>(new RefreshTokenStore());
+}
+
+RefreshTokenStore::RefreshTokenStore() {}
+
+RefreshTokenStore::~RefreshTokenStore() {}
+
+std::string RefreshTokenStore::ReadRefreshTokenFromDisk(
+ const std::string& user_name) {
+ DCHECK(!user_name.empty());
+
+ base::FilePath token_dir_path(GetRefreshTokenDirPath(user_name));
+ DCHECK(!token_dir_path.empty());
+
+ DVLOG(2) << "Reading token from path: " << token_dir_path.value();
+ base::FilePath token_file_path(token_dir_path.Append(kTokenFileName));
+
+ std::string refresh_token;
+ if (!base::ReadFileToString(token_file_path, &refresh_token)) {
+ DVLOG(1) << "Failed to read token file from: " << token_dir_path.value();
+ return std::string();
+ }
+
+ return refresh_token;
+}
+
+bool RefreshTokenStore::WriteRefreshTokenToDisk(
+ const std::string& user_name,
+ const std::string& refresh_token) {
+ DCHECK(!user_name.empty());
+ DCHECK(!refresh_token.empty());
+
+ base::FilePath token_dir_path(GetRefreshTokenDirPath(user_name));
+ if (token_dir_path.empty()) {
+ return false;
+ }
+
+ base::FilePath token_file_path(token_dir_path.Append(kTokenFileName));
+ if (!base::DirectoryExists(token_dir_path) &&
+ !base::CreateDirectory(token_dir_path)) {
+ LOG(ERROR) << "Failed to create directory, refresh token not stored.";
+ return false;
+ }
+
+#if defined(OS_POSIX)
+ // For POSIX we can set permissions on the token file so we do so here.
+ // The test code should not run on other platforms since the code to safely
+ // store the token has not been implemented yet.
+
+ // Create an empty stub file if one does not exist.
+ if (!base::PathExists(token_file_path) &&
+ base::WriteFile(token_file_path, "", 0) < 0) {
+ LOG(ERROR) << "Failed to create stub file, refresh token not stored.";
+ return false;
+ }
+
+ // Set permissions on the stub file.
+ int mode =
+ base::FILE_PERMISSION_READ_BY_USER | base::FILE_PERMISSION_WRITE_BY_USER;
+ if (!SetPosixFilePermissions(token_file_path, mode)) {
+ LOG(ERROR) << "Failed to set file permissions, refresh token not stored.";
+ return false;
+ }
+
+ // Write the refresh token to our newly created file.
+ if (base::WriteFile(token_file_path, refresh_token.c_str(),
+ refresh_token.size()) < 0) {
+ LOG(ERROR) << "Failed to save refresh token to the file on disk.";
+ return false;
+ }
+
+ return true;
+#else
+ NOTIMPLEMENTED()
+#endif // OS_POSIX
+}
+
+} // namespace test
+} // namespace remoting
« remoting/test/refresh_token_store.h ('K') | « remoting/test/refresh_token_store.h ('k') | no next file » | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698