1 files changed, 98 insertions, 0 deletions
diff --git a/cpp/test/region_data_builder_test.cc b/cpp/test/region_data_builder_test.cc
new file mode 100644
index 0000000..088a73e
--- /dev/null
+++ b/cpp/test/region_data_builder_test.cc
@@ -0,0 +1,98 @@
+// Copyright (C) 2014 Google Inc.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+#include <libaddressinput/region_data_builder.h>
+
+#include <libaddressinput/callback.h>
+#include <libaddressinput/null_storage.h>
+#include <libaddressinput/preload_supplier.h>
+#include <libaddressinput/region_data.h>
+#include <libaddressinput/util/basictypes.h>
+#include <libaddressinput/util/scoped_ptr.h>
+
+#include "fake_downloader.h"
+
+#include <string>
+
+#include <gtest/gtest.h>
+
+namespace {
+
+using i18n::addressinput::BuildCallback;
+using i18n::addressinput::FakeDownloader;
+using i18n::addressinput::NullStorage;
+using i18n::addressinput::PreloadSupplier;
+using i18n::addressinput::RegionData;
+using i18n::addressinput::RegionDataBuilder;
+using i18n::addressinput::scoped_ptr;
+
+class RegionDataBuilderTest : public testing::Test {
+ protected:
+  RegionDataBuilderTest()
+      : supplier_(FakeDownloader::kFakeAggregateDataUrl,
+                  new FakeDownloader,
+                  new NullStorage),
+        builder_(&supplier_),
+        loaded_callback_(BuildCallback(this, &RegionDataBuilderTest::OnLoaded)),
+        best_language_() {}
+
+  virtual ~RegionDataBuilderTest() {}
+
+  PreloadSupplier supplier_;
+  RegionDataBuilder builder_;
+  scoped_ptr<PreloadSupplier::Callback> loaded_callback_;
+  std::string best_language_;
+
+ private:
+  void OnLoaded(bool success,
+                const std::string& region_code,
+                const int& num_rules) {
+    ASSERT_TRUE(success);
+    ASSERT_FALSE(region_code.empty());
+    ASSERT_LT(0, num_rules);
+    ASSERT_TRUE(supplier_.IsLoaded(region_code));
+  }
+
+  DISALLOW_COPY_AND_ASSIGN(RegionDataBuilderTest);
+};
+
+TEST_F(RegionDataBuilderTest, BuildUsRegionTree) {
+  supplier_.LoadRules("US", *loaded_callback_);
+  const RegionData& tree = builder_.Build("US", "en-US", &best_language_);
+  EXPECT_FALSE(tree.sub_regions().empty());
+}
+
+TEST_F(RegionDataBuilderTest, BuildCnRegionTree) {
+  supplier_.LoadRules("CN", *loaded_callback_);
+  const RegionData& tree = builder_.Build("CN", "zh-Hans", &best_language_);
+  ASSERT_FALSE(tree.sub_regions().empty());
+  EXPECT_FALSE(tree.sub_regions().front()->sub_regions().empty());
+}
+
+TEST_F(RegionDataBuilderTest, BuildChRegionTree) {
+  supplier_.LoadRules("CH", *loaded_callback_);
+  const RegionData& tree = builder_.Build("CH", "de-CH", &best_language_);
+  // Although "CH" has information for its administrative divisions, the
+  // administrative area field is not used, which results in an empty tree of
+  // sub-regions.
+  EXPECT_TRUE(tree.sub_regions().empty());
+}
+
+TEST_F(RegionDataBuilderTest, BuildZwRegionTree) {
+  supplier_.LoadRules("ZW", *loaded_callback_);
+  const RegionData& tree = builder_.Build("ZW", "en-ZW", &best_language_);
+  EXPECT_TRUE(tree.sub_regions().empty());
+}
+
+}  // namespace