refactor(optim): add the missing 'grille' mecanism for serialization

publicodes · Jan 17, 2024 · f93551a · f93551a
1 parent fed1c1e
commit f93551a
Show file tree

Hide file tree

Showing 2 changed files with 32 additions and 2 deletions.
diff --git a/source/serializeParsedRules.ts b/source/serializeParsedRules.ts
@@ -172,8 +172,9 @@ function serializeASTNode(node: ASTNode): SerializedRule {
           }
         }
 
-        case 'taux progressif':
-        case 'barème': {
+        case 'barème':
+        case 'grille':
+        case 'taux progressif': {
           const serializedNode = {
             assiette: serializeASTNode(node.explanation.assiette),
             tranches: node.explanation.tranches.map((tranche) => {

diff --git a/test/serializeParsedRules.test.ts b/test/serializeParsedRules.test.ts
@@ -366,6 +366,35 @@ describe('API > mecanisms list', () => {
     expect(serializedRules).toStrictEqual(rules)
   })
 
+  it('should serialize rule with [grille]', () => {
+    const rules = {
+      'SMIC horaire': {
+        valeur: '10 €/heures',
+      },
+      'revenu cotisé': {
+        valeur: '1900 €/an',
+      },
+      'trimestres validés': {
+        unité: 'trimestres validés/an',
+        grille: {
+          assiette: 'revenu cotisé',
+          multiplicateur: 'SMIC horaire',
+          tranches: [
+            { montant: 0, plafond: '150 heures/an' },
+            { montant: 1, plafond: '300 heures/an' },
+            { montant: 2, plafond: '450 heures/an' },
+            { montant: 3, plafond: '600 heures/an' },
+            { montant: 4 },
+          ],
+        },
+      },
+    }
+    const serializedRules = serializeParsedRules(
+      new Engine(rules).getParsedRules(),
+    )
+    expect(serializedRules).toStrictEqual(rules)
+  })
+
   it("should serialize rule with [barème] and a custom 'multiplicateur'", () => {
     const rules = {
       'revenu imposable': {