| OLD | NEW |
| 1 // Copyright 2015 The Chromium Authors. All rights reserved. | 1 // Copyright 2015 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "components/feedback/feedback_uploader_chrome.h" | 5 #include "components/feedback/feedback_uploader_chrome.h" |
| 6 | 6 |
| 7 #include <string> | 7 #include <string> |
| 8 | 8 |
| 9 #include "base/macros.h" | 9 #include "base/macros.h" |
| 10 #include "base/message_loop/message_loop.h" |
| 10 #include "base/metrics/field_trial.h" | 11 #include "base/metrics/field_trial.h" |
| 11 #include "components/variations/variations_associated_data.h" | 12 #include "components/variations/variations_associated_data.h" |
| 12 #include "components/variations/variations_http_header_provider.h" | 13 #include "components/variations/variations_http_header_provider.h" |
| 13 #include "content/public/test/test_browser_context.h" | 14 #include "content/public/test/test_browser_context.h" |
| 14 #include "content/public/test/test_browser_thread_bundle.h" | |
| 15 #include "net/url_request/test_url_fetcher_factory.h" | 15 #include "net/url_request/test_url_fetcher_factory.h" |
| 16 #include "net/url_request/url_fetcher_delegate.h" | 16 #include "net/url_request/url_fetcher_delegate.h" |
| 17 #include "testing/gtest/include/gtest/gtest.h" | 17 #include "testing/gtest/include/gtest/gtest.h" |
| 18 | 18 |
| 19 namespace feedback { | 19 namespace feedback { |
| 20 | 20 |
| 21 class FeedbackUploaderChromeTest : public ::testing::Test { | 21 class FeedbackUploaderChromeTest : public ::testing::Test { |
| 22 protected: | 22 protected: |
| 23 FeedbackUploaderChromeTest() | 23 FeedbackUploaderChromeTest() {} |
| 24 : browser_thread_bundle_(content::TestBrowserThreadBundle::IO_MAINLOOP) {} | |
| 25 | 24 |
| 26 ~FeedbackUploaderChromeTest() override { | 25 ~FeedbackUploaderChromeTest() override { |
| 27 // Clean up registered ids. | 26 // Clean up registered ids. |
| 28 variations::testing::ClearAllVariationIDs(); | 27 variations::testing::ClearAllVariationIDs(); |
| 29 } | 28 } |
| 30 | 29 |
| 31 // Registers a field trial with the specified name and group and an associated | 30 // Registers a field trial with the specified name and group and an associated |
| 32 // google web property variation id. | 31 // google web property variation id. |
| 33 void CreateFieldTrialWithId(const std::string& trial_name, | 32 void CreateFieldTrialWithId(const std::string& trial_name, |
| 34 const std::string& group_name, | 33 const std::string& group_name, |
| 35 int variation_id) { | 34 int variation_id) { |
| 36 variations::AssociateGoogleVariationID( | 35 variations::AssociateGoogleVariationID( |
| 37 variations::GOOGLE_WEB_PROPERTIES, trial_name, group_name, | 36 variations::GOOGLE_WEB_PROPERTIES, trial_name, group_name, |
| 38 static_cast<variations::VariationID>(variation_id)); | 37 static_cast<variations::VariationID>(variation_id)); |
| 39 base::FieldTrialList::CreateFieldTrial(trial_name, group_name)->group(); | 38 base::FieldTrialList::CreateFieldTrial(trial_name, group_name)->group(); |
| 40 } | 39 } |
| 41 | 40 |
| 42 private: | 41 private: |
| 43 content::TestBrowserThreadBundle browser_thread_bundle_; | 42 base::MessageLoopForUI message_loop_; |
| 44 | 43 |
| 45 DISALLOW_COPY_AND_ASSIGN(FeedbackUploaderChromeTest); | 44 DISALLOW_COPY_AND_ASSIGN(FeedbackUploaderChromeTest); |
| 46 }; | 45 }; |
| 47 | 46 |
| 48 TEST_F(FeedbackUploaderChromeTest, VariationHeaders) { | 47 TEST_F(FeedbackUploaderChromeTest, VariationHeaders) { |
| 49 // Register a trial and variation id, so that there is data in variations | 48 // Register a trial and variation id, so that there is data in variations |
| 50 // headers. Also, the variations header provider may have been registered to | 49 // headers. Also, the variations header provider may have been registered to |
| 51 // observe some other field trial list, so reset it. | 50 // observe some other field trial list, so reset it. |
| 52 variations::VariationsHttpHeaderProvider::GetInstance()->ResetForTesting(); | 51 variations::VariationsHttpHeaderProvider::GetInstance()->ResetForTesting(); |
| 53 base::FieldTrialList field_trial_list_(NULL); | 52 base::FieldTrialList field_trial_list_(NULL); |
| (...skipping 11 matching lines...) Expand all Loading... |
| 65 std::string value; | 64 std::string value; |
| 66 EXPECT_TRUE(headers.GetHeader("X-Client-Data", &value)); | 65 EXPECT_TRUE(headers.GetHeader("X-Client-Data", &value)); |
| 67 EXPECT_FALSE(value.empty()); | 66 EXPECT_FALSE(value.empty()); |
| 68 // The fetcher's delegate is responsible for freeing the fetcher (and itself). | 67 // The fetcher's delegate is responsible for freeing the fetcher (and itself). |
| 69 fetcher->delegate()->OnURLFetchComplete(fetcher); | 68 fetcher->delegate()->OnURLFetchComplete(fetcher); |
| 70 | 69 |
| 71 variations::VariationsHttpHeaderProvider::GetInstance()->ResetForTesting(); | 70 variations::VariationsHttpHeaderProvider::GetInstance()->ResetForTesting(); |
| 72 } | 71 } |
| 73 | 72 |
| 74 } // namespace feedback | 73 } // namespace feedback |
| OLD | NEW |