Fix number as string for count

ditrit · Jul 10, 2024 · 0f350ca · 0f350ca
1 parent eecf1b3
commit 0f350ca
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/src/drawer/render/TerraformComponentRenderer.js b/src/drawer/render/TerraformComponentRenderer.js
@@ -11,11 +11,12 @@ class TerraformComponentRenderer extends ComponentRenderer {
    */
   getTemplateData(component) {
     const countAttribute = component.getAttributeByName('count');
-    const hasCount = !!countAttribute && countAttribute.value !== 1;
+    const countValue = parseInt(countAttribute?.value, 10);
+    const hasCount = !!countAttribute && countValue !== 1;
     let count = '?';
 
-    if (hasCount && countAttribute.type === 'Number') {
-      count = countAttribute.value;
+    if (hasCount && Number.isInteger(countValue)) {
+      count = countValue;
     }
 
     return {

diff --git a/tests/unit/drawer/render/TerraformComponentRenderer.spec.js b/tests/unit/drawer/render/TerraformComponentRenderer.spec.js
@@ -55,12 +55,13 @@ describe('Test class: TerraformComponentRenderer', () => {
         definition: new TerraformComponentDefinition(),
         attributes: [new TerraformComponentAttribute({
           name: 'count',
-          type: 'Number',
-          value: 2,
+          type: 'String',
+          value: '2',
         })],
       });
       const result = renderer.getTemplateData(component);
 
+      expect(result.hasCount).toEqual(true);
       expect(result.count).toBe(2);
     });