Add unit test

Author: FrankYFTang

inflection/test/src/inflection/message2/MF2FactoryTest.cpp

@@ -10,37 +10,86 @@
 #include "unicode/locid.h"
 #include "unicode/messageformat2.h"
 #include "unicode/messageformat2_function_registry.h"
-
+#include "unicode/messageformat2_formattable.h"
 
 using icu::Locale;
+using icu::UnicodeString;
+using icu::message2::Formattable;
 using icu::message2::data_model::FunctionName;
 using icu::message2::MFFunctionRegistry;
+using icu::message2::MessageArguments;
 using icu::message2::MessageFormatter;
 
 
-MFFunctionRegistry SetupCustomBuilder(UErrorCode& errorCode) {
+TEST_CASE("MF2Factory#testBasic")
+{
+    UErrorCode errorCode = U_ZERO_ERROR;
     auto functionName = FunctionName("inflection");
-    return MFFunctionRegistry::Builder(errorCode)
+    MFFunctionRegistry::Builder(errorCode)
         .adoptFormatter(functionName, inflection::message2::createFormatterFactory(), errorCode)
         .adoptSelector(functionName, inflection::message2::createSelectorFactory(), errorCode)
         .build();
+    REQUIRE(U_SUCCESS(errorCode));
 }
 
-TEST_CASE("MF2Factory#testBasic")
+TEST_CASE("MF2Factory#testCreateFormatter")
 {
     UErrorCode errorCode = U_ZERO_ERROR;
-    auto customRegistry = SetupCustomBuilder(errorCode);
+    auto functionName = FunctionName("inflection");
+    auto customRegistry = MFFunctionRegistry::Builder(errorCode)
+        .adoptFormatter(functionName, inflection::message2::createFormatterFactory(), errorCode)
+        .build();
+    REQUIRE(U_SUCCESS(errorCode));
+
+    UParseError pe;
+    MessageFormatter mf = MessageFormatter::Builder(errorCode)
+                .setFunctionRegistry(customRegistry)
+                .setLocale(Locale::forLanguageTag("es-MX", errorCode))
+                .setPattern(
+                    "Foo {$name :inflection hello=world definiteness=definite number=plural gender=feminine} Bar",
+                    pe, errorCode)
+                .build(errorCode);
     REQUIRE(U_SUCCESS(errorCode));
+
+    std::map<UnicodeString, Formattable> arguments;
+    arguments["name"]= Formattable("gato");
+
+    UnicodeString ret = mf.formatToString(MessageArguments(arguments, errorCode), errorCode);
+    REQUIRE(U_SUCCESS(errorCode));
+    REQUIRE(ret == u"Foo las gatas Bar");
 }
 
-TEST_CASE("MF2Factory#testCreateFormatter")
+TEST_CASE("MF2Factory#testCreateSelector")
 {
     UErrorCode errorCode = U_ZERO_ERROR;
+    auto functionName = FunctionName("inflection");
+    auto customRegistry = MFFunctionRegistry::Builder(errorCode)
+        .adoptSelector(functionName, inflection::message2::createSelectorFactory(), errorCode)
+        .build();
+    REQUIRE(U_SUCCESS(errorCode));
+
     UParseError pe;
     MessageFormatter mf = MessageFormatter::Builder(errorCode)
-                .setFunctionRegistry(SetupCustomBuilder(errorCode))
+                .setFunctionRegistry(customRegistry)
                 .setLocale(Locale::forLanguageTag("es-MX", errorCode))
-                .setPattern("Foo {$name :inflection hello=world definiteness=definite number=plural gender=feminine} Bar", pe, errorCode)
+                .setPattern(
+                    ".local $var1 = {$name :inflection feature=gender} \
+                     .local $var2 = {$name :inflection feature=number} \
+                     .match $var1 $var2\
+                     masculine 2 {{{$name} is Masculine & 2}} \
+                     feminine singular {{{$name} is Feminine & Singular}} \
+                     foo 4 {{{$name} is Foo & 4}} \
+                     masculine singular {{{$name} is Masculine & Singular}} \
+                     hello singular {{{$name} is Hello & Singular}} \
+                     * * {{{$name} is other}}\n",
+                    pe, errorCode)
                 .build(errorCode);
     REQUIRE(U_SUCCESS(errorCode));
+
+    std::map<UnicodeString, Formattable> arguments;
+    arguments["name"]= Formattable("gato");
+
+    UnicodeString ret = mf.formatToString(MessageArguments(arguments, errorCode), errorCode);
+    REQUIRE(U_SUCCESS(errorCode));
+    REQUIRE(ret == u"gato is Masculine & Singular");
 }