Index: remoting/test/app_remoting_test_driver_environment.h |
diff --git a/remoting/test/app_remoting_test_driver_environment.h b/remoting/test/app_remoting_test_driver_environment.h |
new file mode 100644 |
index 0000000000000000000000000000000000000000..b48a0233e77c07b9a5fc382e58db4fb893fc2730 |
--- /dev/null |
+++ b/remoting/test/app_remoting_test_driver_environment.h |
@@ -0,0 +1,94 @@ |
+// 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_APP_REMOTING_TEST_DRIVER_ENVIRONMENT_H_ |
+#define REMOTING_TEST_APP_REMOTING_TEST_DRIVER_ENVIRONMENT_H_ |
+ |
+#include "base/memory/scoped_ptr.h" |
+#include "remoting/test/access_token_fetcher.h" |
+#include "remoting/test/refresh_token_store.h" |
+#include "testing/gtest/include/gtest/gtest.h" |
+ |
+namespace remoting { |
+namespace test { |
+ |
+// 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 caching remote host connection information. |
+// TODO(joedow) Add remote host connection functionality. |
+class AppRemotingTestDriverEnvironment : public testing::Environment { |
+ public: |
+ AppRemotingTestDriverEnvironment( |
+ const std::string& user_name, |
+ const std::string& service_environment); |
+ ~AppRemotingTestDriverEnvironment() override; |
+ |
+ // Returns false if a valid access token cannot be retrieved. |
+ bool Initialize(const std::string& auth_code); |
+ |
+ // Synchronously request a new access token using |refresh_token_|. |
+ // Returns true if a valid access token has been retrieved. |
+ bool RefreshAccessToken(); |
+ |
+ // Used to set a mock version of the fetcher for testing. |
Wez
2015/02/19 22:00:24
nit: Clarify "for testing the driver environment i
joedow
2015/02/20 02:58:36
Done.
|
+ void SetAccessTokenFetcherForTest( |
+ scoped_ptr<remoting::test::AccessTokenFetcher> mock_fetcher); |
Wez
2015/02/19 22:00:24
nit: mock_fetcher->fetcher - it'd be valid for a c
joedow
2015/02/20 02:58:36
I've updated this code to use a friend class. Tha
Wez
2015/02/23 20:17:14
Acknowledged.
|
+ |
+ // Used to set a mock version of the refresh token store object for testing. |
Wez
2015/02/19 22:00:23
nit: If you make the comment on the fetcher setter
joedow
2015/02/20 02:58:36
Done.
|
+ void SetRefreshTokenStoreForTest( |
+ scoped_ptr<remoting::test::RefreshTokenStoreInterface> mock_token_store); |
+ |
+ // Accessors for fields used by tests. |
+ const std::string& access_token() const { return access_token_; } |
+ const std::string& user_name() const { return user_name_; } |
+ |
+ private: |
+ // Environment interface. |
+ void SetUp() override; |
+ 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 run_loop_quit_closure, |
+ const std::string& access_token, |
+ const std::string& refresh_token); |
+ |
+ // Used for authenticating with the app remoting service API. |
+ std::string access_token_; |
+ |
+ // Used to retrieve an access token. |
+ std::string refresh_token_; |
+ |
+ // Used for authentication. |
+ std::string user_name_; |
+ |
+ // Service API to target when retrieving remote host connection information. |
+ // TODO(joedow): Turn this into an enum when remote_host code is added. |
+ std::string service_environment_; |
+ |
+ // Access token fetcher to use for testing. |
+ scoped_ptr<remoting::test::AccessTokenFetcher> mock_access_token_fetcher_; |
+ |
+ // Used to read/write the refresh token from the disk. |
+ scoped_ptr<remoting::test::RefreshTokenStoreInterface> |
+ refresh_token_store_; |
+ |
+ DISALLOW_COPY_AND_ASSIGN(AppRemotingTestDriverEnvironment); |
+}; |
+ |
+// Unfortunately a global var is how the GTEST framework handles sharing data |
+// between tests and keeping long-lived objects around. Used to share auth |
+// tokens and remote host connection information across tests. |
+extern AppRemotingTestDriverEnvironment* AppRemotingSharedData; |
+ |
+} // namespace test |
+} // namespace remoting |
+ |
+#endif // REMOTING_TEST_APP_REMOTING_TEST_DRIVER_ENVIRONMENT_H_ |