diff --git a/main.js b/main.js index fe241f6..28ef8cf 100644 --- a/main.js +++ b/main.js @@ -7,6 +7,11 @@ class House { addRoom(room) { this.rooms.push(room); } + removeRoom(room) { + let index = this.rooms.indexOf(room); + this.rooms.splice(index, 1); + return this; + } area() { return this.rooms.reduce((acc, e) => { return acc + e.area(); @@ -87,4 +92,13 @@ describe("House", () => { }); }); + describe("Removing rooms from houses", () => { + it("removes a single room", () => { + let house1 = new House("Red"); + let room1 = new Room({ width: 7, length: 8 }); + house1.addRoom(room1); + house1.removeRoom(room1); + expect(house1.rooms.length).to.equal(0); + }); + }); });