Chromium Code Reviews| 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..7ce9e1f0e6a2b55b05dff46c289478e28febd235 |
| --- /dev/null |
| +++ b/remoting/test/chromoting_test_driver_environment.h |
| @@ -0,0 +1,130 @@ |
| +// 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 lost. |
|
anandc
2015/07/15 23:51:58
lost=list
tonychun
2015/07/16 18:46:53
Done.
|
| + std::string host_name_; |
| + |
| + // Used for authentication. |
| + std::string user_name_; |
|
anandc
2015/07/15 23:51:58
Change to: "The test-account for a test case".
Ye
tonychun
2015/07/16 18:46:53
Done.
|
| + |
| + // Path to a JSON file containing refresh tokens. |
| + base::FilePath refresh_token_file_path_; |
| + |
| + // Host info in host list used to connect to remote host. |
|
anandc
2015/07/15 23:51:58
? Did you mean: "List of remote-hosts for the spec
tonychun
2015/07/16 18:46:53
Done.
|
| + 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); |
| +}; |
| + |
| +// Unfortunately a global var is how the GTEST framework handles sharing data |
| +// between tests and keeping long-lived objects around. Used to share access |
| +// tokens and a host list across tests. |
| +extern ChromotingTestDriverEnvironment* ChromotingSharedData; |
| + |
| +} // namespace test |
| +} // namespace remoting |
| + |
| +#endif // REMOTING_TEST_CHROMOTING_TEST_DRIVER_ENVIRONMENT_H_ |