Index: components/payments/core/subkey_requester_unittest.cc |
diff --git a/components/payments/core/subkey_requester_unittest.cc b/components/payments/core/subkey_requester_unittest.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..8af7801a08999f2d56af546426e76d9d9d7f5cbe |
--- /dev/null |
+++ b/components/payments/core/subkey_requester_unittest.cc |
@@ -0,0 +1,184 @@ |
+// Copyright 2017 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 "components/payments/core/subkey_requester.h" |
+ |
+#include <utility> |
+ |
+#include "base/run_loop.h" |
+#include "base/strings/utf_string_conversions.h" |
+#include "base/test/scoped_task_scheduler.h" |
+#include "testing/gtest/include/gtest/gtest.h" |
+#include "third_party/libaddressinput/src/cpp/include/libaddressinput/null_storage.h" |
+#include "third_party/libaddressinput/src/cpp/include/libaddressinput/source.h" |
+#include "third_party/libaddressinput/src/cpp/include/libaddressinput/storage.h" |
+#include "third_party/libaddressinput/src/cpp/test/testdata_source.h" |
+ |
+namespace payments { |
+namespace { |
+ |
+using ::i18n::addressinput::NullStorage; |
+using ::i18n::addressinput::Source; |
+using ::i18n::addressinput::Storage; |
+using ::i18n::addressinput::TestdataSource; |
+ |
+const char kLocale[] = "OZ"; |
+const int kInvalidSize = -1; |
+const int kCorrectSize = 2; // for subkeys = Do, Re |
+const int kEmptySize = 0; |
+ |
+// The requester of normalization for this test. |
+class RequestDelegate : public SubKeyRequester::Delegate { |
+ public: |
+ RequestDelegate() : subkeys_size_(kInvalidSize) {} |
+ |
+ ~RequestDelegate() override {} |
+ |
+ void OnSubKeysReceived(std::vector<std::string> subkeys) override { |
+ subkeys_size_ = subkeys.size(); |
+ } |
+ |
+ int subkeys_size() const { return subkeys_size_; } |
+ |
+ private: |
+ int subkeys_size_; |
+ |
+ DISALLOW_COPY_AND_ASSIGN(RequestDelegate); |
+}; |
+ |
+// Used to load region rules for this test. |
+class ChromiumTestdataSource : public TestdataSource { |
+ public: |
+ ChromiumTestdataSource() : TestdataSource(true) {} |
+ |
+ ~ChromiumTestdataSource() override {} |
+ |
+ // For this test, only load the rules for the kLocale. |
+ void Get(const std::string& key, const Callback& data_ready) const override { |
+ data_ready( |
+ true, key, |
+ new std::string( |
+ "{\"data/OZ\": " |
+ "{\"id\":\"data/OZ\",\"key\":\"OZ\",\"name\":\"Oz \", " |
+ "\"lang\":\"en\",\"languages\":\"en\",\"sub_keys\":\"DO~Re\"}}")); |
+ } |
+ |
+ private: |
+ DISALLOW_COPY_AND_ASSIGN(ChromiumTestdataSource); |
+}; |
+ |
+// A test subclass of the SubKeyRequesterImpl. Used to simulate rules not |
+// being loaded. |
+class TestSubKeyRequester : public SubKeyRequester { |
+ public: |
+ TestSubKeyRequester(std::unique_ptr<::i18n::addressinput::Source> source, |
+ std::unique_ptr<::i18n::addressinput::Storage> storage) |
+ : SubKeyRequester(std::move(source), std::move(storage)), |
+ should_load_rules_(true) {} |
+ |
+ ~TestSubKeyRequester() override {} |
+ |
+ void ShouldLoadRules(bool should_load_rules) { |
+ should_load_rules_ = should_load_rules; |
+ } |
+ |
+ void LoadRulesForRegion(const std::string& region_code) override { |
+ if (should_load_rules_) { |
+ SubKeyRequester::LoadRulesForRegion(region_code); |
+ } |
+ } |
+ |
+ private: |
+ bool should_load_rules_; |
+ base::test::ScopedTaskScheduler scoped_task_scheduler_; |
+ |
+ DISALLOW_COPY_AND_ASSIGN(TestSubKeyRequester); |
+}; |
+ |
+} // namespace |
+ |
+class SubKeyRequesterTest : public testing::Test { |
+ protected: |
+ SubKeyRequesterTest() |
+ : requester_(new TestSubKeyRequester( |
+ std::unique_ptr<Source>(new ChromiumTestdataSource), |
+ std::unique_ptr<Storage>(new NullStorage))) {} |
+ |
+ ~SubKeyRequesterTest() override {} |
+ |
+ const std::unique_ptr<TestSubKeyRequester> requester_; |
+ |
+ private: |
+ DISALLOW_COPY_AND_ASSIGN(SubKeyRequesterTest); |
+}; |
+ |
+// Tests that rules are not loaded by default. |
+TEST_F(SubKeyRequesterTest, AreRulesLoadedForRegion_NotLoaded) { |
+ EXPECT_FALSE(requester_->AreRulesLoadedForRegion(kLocale)); |
+} |
+ |
+// Tests that the rules are loaded correctly. |
+TEST_F(SubKeyRequesterTest, AreRulesLoadedForRegion_Loaded) { |
+ requester_->LoadRulesForRegion(kLocale); |
+ EXPECT_TRUE(requester_->AreRulesLoadedForRegion(kLocale)); |
+} |
+ |
+// Tests that if the rules are loaded before the subkey request is started, the |
+// received subkeys will be returned to the delegate synchronously. |
+TEST_F(SubKeyRequesterTest, StartRequest_RulesLoaded) { |
+ RequestDelegate delegate; |
+ |
+ // Load the rules. |
+ requester_->LoadRulesForRegion(kLocale); |
+ EXPECT_TRUE(requester_->AreRulesLoadedForRegion(kLocale)); |
+ |
+ // Start the request. |
+ requester_->StartRegionSubKeysRequest(kLocale, 0, &delegate); |
+ |
+ // Since the rules are already loaded, the subkeys should be received |
+ // synchronously. |
+ EXPECT_EQ(delegate.subkeys_size(), kCorrectSize); |
+} |
+ |
+// Tests that if the rules are not loaded before the request and cannot be |
+// loaded after, the subkeys will not be received and the delegate will be |
+// notified. |
+TEST_F(SubKeyRequesterTest, StartRequest_RulesNotLoaded_WillNotLoad) { |
+ RequestDelegate delegate; |
+ |
+ // Make sure the rules will not be loaded in the StartRegionSubKeysRequest |
+ // call. |
+ requester_->ShouldLoadRules(false); |
+ |
+ // Start the normalization. |
+ requester_->StartRegionSubKeysRequest(kLocale, 0, &delegate); |
+ |
+ // Let the timeout execute. |
+ base::RunLoop().RunUntilIdle(); |
+ |
+ // Since the rules are never loaded and the timeout is 0, the delegate should |
+ // get notified that the subkeys could not be received. |
+ EXPECT_EQ(delegate.subkeys_size(), kEmptySize); |
+} |
+ |
+// Tests that if the rules are not loaded before the call to |
+// StartRegionSubKeysRequest, they will be loaded in the call. |
+TEST_F(SubKeyRequesterTest, StartRequest_RulesNotLoaded_WillLoad) { |
+ RequestDelegate delegate; |
+ |
+ // Make sure the rules will not be loaded in the StartRegionSubKeysRequest |
+ // call. |
+ requester_->ShouldLoadRules(true); |
+ // Start the request. |
+ requester_->StartRegionSubKeysRequest(kLocale, 0, &delegate); |
+ |
+ // Even if the rules are not loaded before the call to |
+ // StartRegionSubKeysRequest, they should get loaded in the call. Since our |
+ // test source is synchronous, the request will happen synchronously |
+ // too. |
+ EXPECT_TRUE(requester_->AreRulesLoadedForRegion(kLocale)); |
+ EXPECT_EQ(delegate.subkeys_size(), kCorrectSize); |
+} |
+ |
+} // namespace payments |