Index: remoting/test/chromoting_test_driver_environment.h |
diff --git a/remoting/test/chromoting_test_driver_environment.h b/remoting/test/chromoting_test_driver_environment.h |
new file mode 100644 |
index 0000000000000000000000000000000000000000..dd22446d36c2a91a4510130292e3789507199ae9 |
--- /dev/null |
+++ b/remoting/test/chromoting_test_driver_environment.h |
@@ -0,0 +1,125 @@ |
+// 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. |
+ |
+#ifndef REMOTING_TEST_CHROMOTING_TEST_DRIVER_ENVIRONMENT_H_ |
+#define REMOTING_TEST_CHROMOTING_TEST_DRIVER_ENVIRONMENT_H_ |
+ |
+#include <string> |
+#include <vector> |
+ |
+#include "base/callback.h" |
+#include "base/files/file_path.h" |
+#include "base/memory/scoped_ptr.h" |
+#include "testing/gtest/include/gtest/gtest.h" |
+ |
+namespace base { |
+class MessageLoopForIO; |
+} |
+ |
+namespace remoting { |
+namespace test { |
+ |
+class AccessTokenFetcher; |
+class RefreshTokenStore; |
+class HostListFetcher; |
+struct HostInfo; |
+ |
+// Globally accessible to all test fixtures and cases and has its |
+// lifetime managed by the GTest framework. It is responsible for managing |
+// access tokens and retrieving the host list. |
+class ChromotingTestDriverEnvironment : public testing::Environment { |
+ public: |
+ struct EnvironmentOptions { |
+ EnvironmentOptions(); |
+ ~EnvironmentOptions(); |
+ |
+ std::string user_name; |
+ std::string host_name; |
+ base::FilePath refresh_token_file_path; |
+ }; |
+ |
+ explicit ChromotingTestDriverEnvironment(const EnvironmentOptions& options); |
+ ~ChromotingTestDriverEnvironment() override; |
+ |
+ // Returns false if a valid access token cannot be retrieved. |
+ bool Initialize(const std::string& auth_code); |
+ |
+ // Retrieves connection information for all known hosts and displays |
+ // their availability to STDOUT. |
+ void DisplayHostList(); |
+ |
+ // Used to set fake/mock objects for ChromotingTestDriverEnvironment tests. |
+ // The caller retains ownership of the supplied objects, and must ensure that |
+ // they remain valid until the ChromotingTestDriverEnvironment instance has |
+ // been destroyed. |
+ void SetAccessTokenFetcherForTest(AccessTokenFetcher* access_token_fetcher); |
+ void SetRefreshTokenStoreForTest(RefreshTokenStore* refresh_token_store); |
+ void SetHostListFetcherForTest(HostListFetcher* host_list_fetcher); |
+ |
+ // Accessors for fields used by tests. |
+ const std::string& access_token() const { return access_token_; } |
+ const std::string& host_name() const { return host_name_; } |
+ const std::string& user_name() const { return user_name_; } |
+ const std::vector<HostInfo>& host_list() const { return host_list_; } |
+ |
+ private: |
+ // testing::Environment interface. |
+ void TearDown() override; |
+ |
+ // Used to retrieve an access token. If |auth_code| is empty, then the stored |
+ // refresh_token will be used instead of |auth_code|. |
+ // Returns true if a new, valid access token has been retrieved. |
+ bool RetrieveAccessToken(const std::string& auth_code); |
+ |
+ // Called after the access token fetcher completes. |
+ // The tokens will be empty on failure. |
+ void OnAccessTokenRetrieved(base::Closure done_closure, |
+ const std::string& retrieved_access_token, |
+ const std::string& retrieved_refresh_token); |
+ |
+ // Used to retrieve a host list from the directory service. |
+ // Returns true if the request was successful and |host_list_| is valid. |
+ bool RetrieveHostList(); |
+ |
+ // Called after the host info fetcher completes. |
+ void OnHostListRetrieved(base::Closure done_closure, |
+ const std::vector<HostInfo>& retrieved_host_list); |
+ |
+ // Used for authenticating with the directory service. |
+ std::string access_token_; |
+ |
+ // Used to retrieve an access token. |
+ std::string refresh_token_; |
+ |
+ // Used to find remote host in host list. |
+ std::string host_name_; |
+ |
+ // The test account for a test case. |
+ std::string user_name_; |
+ |
+ // Path to a JSON file containing refresh tokens. |
+ base::FilePath refresh_token_file_path_; |
+ |
+ // List of remote hosts for the specified user/test-account. |
+ std::vector<HostInfo> host_list_; |
+ |
+ // Access token fetcher used by TestDriverEnvironment tests. |
+ remoting::test::AccessTokenFetcher* test_access_token_fetcher_; |
+ |
+ // RefreshTokenStore used by TestDriverEnvironment tests. |
+ remoting::test::RefreshTokenStore* test_refresh_token_store_; |
+ |
+ // HostListFetcher used by TestDriverEnvironment tests. |
+ remoting::test::HostListFetcher* test_host_list_fetcher_; |
+ |
+ // Used for running network request tasks. |
+ scoped_ptr<base::MessageLoopForIO> message_loop_; |
+ |
+ DISALLOW_COPY_AND_ASSIGN(ChromotingTestDriverEnvironment); |
+}; |
+ |
+} // namespace test |
+} // namespace remoting |
+ |
+#endif // REMOTING_TEST_CHROMOTING_TEST_DRIVER_ENVIRONMENT_H_ |