finish

Author: Alexoner03

exercises/tiered_pricing/solutions/aortiz_typescript/src/InvalidUnitException.ts

@@ -0,0 +1,5 @@
+export class InvalidUnitException extends Error {
+  constructor(message: string) {
+    super(message);
+  }
+}

exercises/tiered_pricing/solutions/aortiz_typescript/src/Subscription.ts

@@ -1,3 +1,5 @@
+import {InvalidUnitException} from "./InvalidUnitException";
+
 export class Subscription {
   private tieredPrice!: number;
 
@@ -6,7 +8,9 @@ export class Subscription {
   }
 
   private setTieredPrice() {
-    if (this.units < 3) {
+    if (this.units <= 0) {
+      throw new InvalidUnitException("Unit should be greater than zero");
+    } else if (this.units < 3) {
       this.tieredPrice = 299;
     } else if (this.units < 11) {
       this.tieredPrice = 239;

exercises/tiered_pricing/solutions/aortiz_typescript/tests/Subscription.test.ts

@@ -1,3 +1,4 @@
+import { InvalidUnitException } from "../src/InvalidUnitException";
 import { Subscription } from "../src/Subscription";
 
 describe("Subscription", () => {
@@ -12,7 +13,7 @@ describe("Subscription", () => {
   });
 
   it("can be calculate subscription price of 2 unit", () => {
-    const subscription = new Subscription(1);
+    const subscription = new Subscription(2);
 
     expect(subscription.getTotalPrice()).toBe(598);
   });
@@ -40,4 +41,8 @@ describe("Subscription", () => {
 
     expect(subscription.getTotalPrice()).toBe(10132);
   });
+
+  it("should be throw InvalidUnitException", () => {
+    expect(() => new Subscription(-1)).toThrow(InvalidUnitException);
+  });
 });