test: Improve test coverage.

Author: xiCO2k

src/utils/get-plural-index.ts

@@ -7,41 +7,6 @@
  */
 export function getPluralIndex(lang: string, number: number): number {
   switch (lang.replace('_', '-')) {
-    case 'az':
-    case 'az-AZ':
-    case 'bo':
-    case 'bo-CN':
-    case 'bo-IN':
-    case 'dz':
-    case 'dz-BT':
-    case 'id':
-    case 'id-ID':
-    case 'ja':
-    case 'ja-JP':
-    case 'jv':
-    case 'ka':
-    case 'ka-GE':
-    case 'km':
-    case 'km-KH':
-    case 'kn':
-    case 'kn-IN':
-    case 'ko':
-    case 'ko-KR':
-    case 'ms':
-    case 'ms-MY':
-    case 'th':
-    case 'th-TH':
-    case 'tr':
-    case 'tr-CY':
-    case 'tr-TR':
-    case 'vi':
-    case 'vi-VN':
-    case 'zh':
-    case 'zh-CN':
-    case 'zh-HK':
-    case 'zh-SG':
-    case 'zh-TW':
-      return 0
     case 'af':
     case 'af-ZA':
     case 'bn':

test/plurization.test.ts

@@ -1,4 +1,5 @@
 import { trans_choice, transChoice, wTransChoice, loadLanguageAsync } from '../src';
+import { choose } from '../src/pluralization';
 
 it.each([
   ['first', 'first', 1],
@@ -72,7 +73,6 @@ it('translates even using an alias "trans_choice"', async () => {
     .toBe('há 3 minutos');
 })
 
-
 it('translates "wTransChoice" and test language change values', async () => {
   await global.mountPlugin()
 
@@ -81,8 +81,64 @@ it('translates "wTransChoice" and test language change values', async () => {
     .toBe('há 3 minutos');
 
   await loadLanguageAsync('en');
-  
+
   expect(translation.value)
     .toBe('3 minutes ago');
-
 })
+
+it.each([
+    ['en-US', 0, 'second'],
+    ['en_US', 1, 'first'],
+    ['en', 2, 'second'],
+    ['fr', 0, 'first'],
+    ['fr', 1, 'first'],
+    ['fr', 2, 'second'],
+    ['be', 0, 'third'],
+    ['be', 1, 'first'],
+    ['be', 3, 'second'],
+    ['sk', 0, 'third'],
+    ['sk', 1, 'first'],
+    ['sk', 2, 'second'],
+    ['ga', 0, 'third'],
+    ['ga', 1, 'first'],
+    ['ga', 2, 'second'],
+    ['lt', 0, 'third'],
+    ['lt', 1, 'first'],
+    ['lt', 2, 'second'],
+    ['sl', 0, 'fourth'],
+    ['sl', 1, 'first'],
+    ['sl', 2, 'second'],
+    ['sl', 4, 'third'],
+    ['mk', 0, 'second'],
+    ['mk', 1, 'first'],
+    ['mt', 0, 'second'],
+    ['mt', 1, 'first'],
+    ['mt', 11, 'third'],
+    ['mt', 21, 'fourth'],
+    ['lv', 0, 'first'],
+    ['lv', 1, 'second'],
+    ['lv', 2, 'third'],
+    ['pl', 0, 'third'],
+    ['pl', 1, 'first'],
+    ['pl', 2, 'second'],
+    ['cy', 0, 'fourth'],
+    ['cy', 1, 'first'],
+    ['cy', 2, 'second'],
+    ['cy', 8, 'third'],
+    ['ro', 0, 'second'],
+    ['ro', 1, 'first'],
+    ['ro', 21, 'third'],
+    ['ar', 0, 'first'],
+    ['ar', 1, 'second'],
+    ['ar', 2, 'third'],
+    ['ar', 3, 'fourth'],
+    ['ar', 12, 'fifth'],
+    ['ar', 99.1, 'sixth'],
+    ['az', 0, 'first'],
+    ['az', 1, 'first'],
+    ['random', 0, 'first'],
+])('translates each lang with the correct plural', (lang, number, correctMessage) => {
+  const message = 'first|second|third|fourth|fifth|sixth';
+
+  expect(choose(message, number, lang)).toBe(correctMessage);
+});