From afa67c729b03e798ccaccd6b0c8e8b0a2c975b29 Mon Sep 17 00:00:00 2001
From: Giacomo Ferlaino <jack@Giacomos-MacBook-Pro.local>
Date: Fri, 13 Mar 2020 21:26:45 +0100
Subject: [PATCH] feat: Added calculateFontSize method test

---
 src/lib/directives/ng2-fittext.directive.ts            | 2 +-
 src/lib/directives/specs/ng2-fittext.directive.spec.ts | 8 ++++++++
 2 files changed, 9 insertions(+), 1 deletion(-)

diff --git a/src/lib/directives/ng2-fittext.directive.ts b/src/lib/directives/ng2-fittext.directive.ts
index 3b7813a..8922fbd 100644
--- a/src/lib/directives/ng2-fittext.directive.ts
+++ b/src/lib/directives/ng2-fittext.directive.ts
@@ -58,7 +58,7 @@ export class Ng2FittextDirective
     return this.fontSize;
   }
 
-  calculateFontSize(fontSize: number, speed: number) {
+  calculateFontSize(fontSize: number, speed: number): number {
     return Math.floor(fontSize / speed);
   }
 
diff --git a/src/lib/directives/specs/ng2-fittext.directive.spec.ts b/src/lib/directives/specs/ng2-fittext.directive.spec.ts
index 11bceaf..4dca2a1 100644
--- a/src/lib/directives/specs/ng2-fittext.directive.spec.ts
+++ b/src/lib/directives/specs/ng2-fittext.directive.spec.ts
@@ -93,4 +93,12 @@ describe('Class: Ng2FittextDirective', () => {
       expect(ng2FittextDirective.getFontSize()).toEqual(1000);
     });
   });
+
+  describe('Method: calculateFontSize', () => {
+    it('Should return the font size rounded down', () => {
+      expect(ng2FittextDirective.calculateFontSize(10, 3)).toEqual(3);
+      expect(ng2FittextDirective.calculateFontSize(9, 3)).toEqual(3);
+      expect(ng2FittextDirective.calculateFontSize(8, 3)).toEqual(2);
+    });
+  });
 });