Simplify multiple addXXX methods into a single add method for Footer

src/file/footer-wrapper.spec.ts

@@ -7,21 +7,19 @@ import { Paragraph } from "./paragraph";
 import { Table } from "./table";
 
 describe("FooterWrapper", () => {
-    describe("#addParagraph", () => {
+    describe("#add", () => {
         it("should call the underlying footer's addParagraph", () => {
             const file = new FooterWrapper(new Media(), 1);
-            const spy = sinon.spy(file.Footer, "addParagraph");
+            const spy = sinon.spy(file.Footer, "add");
-            file.addParagraph(new Paragraph({}));
+            file.add(new Paragraph({}));
 
             expect(spy.called).to.equal(true);
         });
-    });
 
-    describe("#addTable", () => {
         it("should call the underlying footer's addParagraph", () => {
             const file = new FooterWrapper(new Media(), 1);
-            const spy = sinon.spy(file.Footer, "addTable");
+            const spy = sinon.spy(file.Footer, "add");
-            file.addTable(
+            file.add(
                 new Table({
                     rows: 1,
                     columns: 1,
@@ -30,12 +28,10 @@ describe("FooterWrapper", () => {
 
             expect(spy.called).to.equal(true);
         });
-    });
 
-    describe("#addImage", () => {
         it("should call the underlying footer's addImage", () => {
             const file = new FooterWrapper(new Media(), 1);
-            const spy = sinon.spy(file.Footer, "addParagraph");
+            const spy = sinon.spy(file.Footer, "add");
             // tslint:disable-next-line:no-any
             file.addImage({} as any);
 

src/file/footer-wrapper.ts

@@ -22,16 +22,12 @@ export class FooterWrapper {
         this.relationships = new Relationships();
     }
 
-    public addParagraph(paragraph: Paragraph): void {
+    public add(item: Paragraph | Table): void {
-        this.footer.addParagraph(paragraph);
+        this.footer.add(item);
-    }
-
-    public addTable(table: Table): void {
-        this.footer.addTable(table);
     }
 
     public addImage(image: Image): FooterWrapper {
-        this.footer.addParagraph(image.Paragraph);
+        this.footer.add(image.Paragraph);
         return this;
     }
 
@@ -47,7 +43,7 @@ export class FooterWrapper {
     ): Paragraph {
         const image = Media.addImage(this, buffer, width, height, drawingOptions);
         const paragraph = new Paragraph(image);
-        this.addParagraph(paragraph);
+        this.add(paragraph);
 
         return paragraph;
     }

src/file/footer/footer.ts

@@ -36,13 +36,7 @@ export class Footer extends InitializableXmlComponent {
         return this.refId;
     }
 
-    public addParagraph(paragraph: Paragraph): Footer {
+    public add(item: Paragraph | Table): void {
-        this.root.push(paragraph);
+        this.root.push(item);
-
-        return this;
-    }
-
-    public addTable(table: Table): void {
-        this.root.push(table);
     }
 }

src/file/header-wrapper.spec.ts

@@ -10,7 +10,7 @@ describe("HeaderWrapper", () => {
     describe("#add", () => {
         it("should call the underlying header's addChildElement for Paragraph", () => {
             const wrapper = new HeaderWrapper(new Media(), 1);
-            const spy = sinon.spy(wrapper.Header, "addChildElement");
+            const spy = sinon.spy(wrapper.Header, "add");
             wrapper.add(new Paragraph({}));
 
             expect(spy.called).to.equal(true);
@@ -18,7 +18,7 @@ describe("HeaderWrapper", () => {
 
         it("should call the underlying header's addChildElement for Table", () => {
             const wrapper = new HeaderWrapper(new Media(), 1);
-            const spy = sinon.spy(wrapper.Header, "addChildElement");
+            const spy = sinon.spy(wrapper.Header, "add");
             wrapper.add(
                 new Table({
                     rows: 1,

src/file/header-wrapper.ts

@@ -23,7 +23,7 @@ export class HeaderWrapper {
     }
 
     public add(item: Paragraph | Table): HeaderWrapper {
-        this.header.addChildElement(item);
+        this.header.add(item);
 
         return this;
     }