feat: add setAvatar method

WickrInc · Oct 20, 2023 · 89c8e02 · 89c8e02
1 parent 5f57cfc
commit 89c8e02
Show file tree

Hide file tree

Showing 4 changed files with 28 additions and 0 deletions.
diff --git a/lib/bot.js b/lib/bot.js
@@ -1,3 +1,4 @@
+const fs = require('fs')
 const EventEmitter = require('events')
 const mergician = require('mergician')
 const WickrBrain = require('./brain')
@@ -299,6 +300,12 @@ class WickrBot extends EventEmitter {
     this.defaultHandler = fn
   }
 
+  setAvatar(imgPath) {
+    if (!fs.existsSync(imgPath)) throw new Error(`Unable to set avatar. File ${imgPath} not found.`)
+
+    return this.wickr.cmdSetAvatar(imgPath)
+  }
+
   async waitForState(state, retries=10, interval=1000, exponential=true) {
     const curState = this.wickr.getClientState()
 

diff --git a/test/fakes/wickr.js b/test/fakes/wickr.js
@@ -15,6 +15,7 @@ class FakeWickr {
   cmdStartAsyncRecvMessages() {}
   cmdSendRoomMessage() {}
   cmdSend1to1Message() {}
+  cmdSetAvatar() {}
   isConnected() {}
 }
 

diff --git a/test/fixtures/1x1.png b/test/fixtures/1x1.png
diff --git a/test/test-wickr-bot.js b/test/test-wickr-bot.js
@@ -439,4 +439,24 @@ describe('wickr-bot', function() {
       expect(this.wickr.cmdLeaveRoom.calledWith('Sfakevgroupid'))
     })
   })
+
+  describe('#setAvatar', function() {
+    it('calls cmdSetAvatar', function() {
+      let bot = new WickrBot(this.wickr, 'foo')
+      const avatar = __dirname + '/fixtures/1x1.png'
+      sinon.spy(this.wickr, 'cmdSetAvatar')
+
+      bot.setAvatar(avatar)
+      sinon.assert.calledWith(this.wickr.cmdSetAvatar, avatar)
+    })
+
+    it('throws an error when the avatar path does not exist', function() {
+      let fn = () => {
+        const bot = new WickrBot(this.wickr, 'foo')
+        bot.setAvatar('lol.png')
+      }
+
+      expect(fn).to.throw(/File lol.png not found/)
+    })
+  })
 })