add(getString): add third argument to getString for fallback message

projectfluent · Feb 6, 2018 · 402d83c · 402d83c
1 parent f676215
commit 402d83c
Show file tree

Hide file tree

Showing 3 changed files with 48 additions and 5 deletions.
diff --git a/fluent-react/src/localization.js b/fluent-react/src/localization.js
@@ -65,10 +65,13 @@ export default class ReactLocalization {
   /*
    * Find a translation by `id` and format it to a string using `args`.
    */
-  getString(id, args) {
+  getString(id, args, fallback) {
     const mcx = this.getMessageContext(id);
 
     if (mcx === null) {
+      if (fallback) {
+        return fallback;
+      }
       return id;
     }
 

diff --git a/fluent-react/src/with_localization.js b/fluent-react/src/with_localization.js
@@ -12,14 +12,17 @@ export default function withLocalization(Inner) {
     /*
      * Find a translation by `id` and format it to a string using `args`.
      */
-    getString(id, args) {
+    getString(id, args, fallback) {
       const { l10n } = this.context;
 
       if (!l10n) {
+        if (fallback) {
+          return fallback;
+        }
         return id;
       }
 
-      return l10n.getString(id, args);
+      return l10n.getString(id, args, fallback);
     }
 
     render() {

diff --git a/fluent-react/test/with_localization_test.js b/fluent-react/test/with_localization_test.js
@@ -46,7 +46,26 @@ foo = FOO
 
     const getString = wrapper.prop('getString');
     // Returns the translation.
-    assert.equal(getString('foo'), 'FOO');
+    assert.equal(getString('foo', {}), 'FOO');
+  });
+
+  test('getString with access to the l10n context, with fallback value', function() {
+    const mcx = new MessageContext();
+    const l10n = new ReactLocalization([mcx]);
+    const EnhancedComponent = withLocalization(DummyComponent);
+
+    mcx.addMessages(`
+foo = FOO
+`);
+
+    const wrapper = shallow(
+      <EnhancedComponent />,
+      { context: { l10n } }
+    );
+
+    const getString = wrapper.prop('getString');
+    // Returns the translation, even if fallback value provided.
+    assert.equal(getString('bar', {}, 'fallback'), 'fallback');
   });
 
   test('getString without access to the l10n context', function() {
@@ -63,7 +82,25 @@ foo = FOO
     );
 
     const getString = wrapper.prop('getString');
-    // Returns the id.
+    // Returns the id if no fallback.
     assert.equal(getString('foo', {arg: 1}), 'foo');
   });
+
+  test('getString without access to the l10n context, with fallback value', function() {
+    const mcx = new MessageContext();
+    const l10n = new ReactLocalization([mcx]);
+    const EnhancedComponent = withLocalization(DummyComponent);
+
+    mcx.addMessages(`
+foo = FOO
+`);
+
+    const wrapper = shallow(
+      <EnhancedComponent />
+    );
+
+    const getString = wrapper.prop('getString');
+    // Returns the fallback if provided.
+    assert.equal(getString('foo', {arg: 1}, 'fallback message'), 'fallback message');
+  });
 });