fix: fixed a few mistakes & fix tests

src/button.ts

@@ -1,5 +1,5 @@
-import { ButtonParams, IButton } from './types/button'
-const button = (params: ButtonParams | string): IButton => {
+import { IButton } from './types/button'
+const button = (params: IButton | string): IButton => {
   // Button has been created from string
   if (typeof params === 'string') {
     return {
@@ -10,19 +10,18 @@ const button = (params: ButtonParams | string): IButton => {
   if (typeof params === 'object') {
     const {
       title,
-      text,
       tts,
       url,
       hide = false,
       payload,
     } = params
 
-    if (!title && !text) {
+    if (!title) {
       throw new Error('text is a required parameter')
     }
 
     return {
-      title: title || text,
+      title: title,
       tts,
       url,
       hide,

src/buttonBuilder.ts

@@ -1,39 +1,18 @@
+import { IButton } from './types/button'
 
-interface ButtonConstructor {
-  title?: string // title and text — same
-  text?: string
-}
 export default class ButtonBuilder {
-  public button: {
-    title?: string,
-    url?: string,
-    hide?: boolean,
-    payload?: {},
-  }
-  constructor(buttonConstructor?: ButtonConstructor) {
+  button: IButton
+  constructor(buttonConstructor?: IButton) {
     /* No button object passed to the constructor */
     if (!buttonConstructor) {
-      this.button = {}
+      this.button = {
+        title: ''
+      }
       return this
     }
 
-    /* Object-constructor passed */
-    if (typeof buttonConstructor !== 'object') {
-      throw new Error('Invalid ButtonBuilder constructor type. Should be object')
-    }
-    const {
-      title, text,
-    } = buttonConstructor
-    if (!title && !text) {
-      throw new Error('Button [title] or [text] is required for ButtonBuilder constructor.')
-    }
-
     this.button = buttonConstructor
-    return this.button
-  }
-
-  public text(text) {
-    return this._setTitle(text)
+    return this
   }
 
   public title(title) {

src/tests/button.spec.ts

@@ -1,5 +1,6 @@
 
 import button from '../button'
+import ButtonBuilder from '../buttonBuilder'
 
 test('create button with string constructor', () => {
   const expected = {
@@ -20,5 +21,14 @@ test('create button with object constructor', () => {
     payload: expected.payload,
     hide: expected.hide,
   })
+
+  const btn2 = new ButtonBuilder()
+    .title(expected.title)
+    .payload(expected.payload)
+    .shouldHide(expected.hide)
+    .get()
+
   expect(btn).toEqual(expected)
+  expect(btn2).toEqual(expected)
 })
+

src/tests/buttonBuilder.spec.ts

@@ -15,7 +15,7 @@ test('common test for buttonBuilder', () => {
 
   // test using button builder factory
   button
-    .text(expected.title)
+    .title(expected.title)
     .url(expected.url)
     .shouldHide(expected.hide)
     .payload(expected.payload)
@@ -25,5 +25,5 @@ test('common test for buttonBuilder', () => {
   // test using button builder constructor
 
   const button2 = new ButtonBuilder(expected)
-  expect(button2).toEqual(expected)
+  expect(button2.get()).toEqual(expected)
 })