From 3b01ea12e63f9fb7180739bc1b23eeccdd4a0b0b Mon Sep 17 00:00:00 2001
From: Skillz <gankstarsleague@gmail.com>
Date: Sun, 26 Mar 2023 11:31:31 -0500
Subject: [PATCH] fix: more unit tests coverage for utils

---
 packages/utils/src/bucket.ts           |  5 +---
 packages/utils/tests/bucket.spec.ts    | 36 ++++++++++++++++++++++++++
 packages/utils/tests/color.spec.ts     | 14 ++++++++++
 packages/utils/tests/logger.spec.ts    | 28 ++++++++++++++++++++
 packages/utils/tests/reactions.spec.ts | 20 ++++++++++++++
 5 files changed, 99 insertions(+), 4 deletions(-)
 create mode 100644 packages/utils/tests/logger.spec.ts
 create mode 100644 packages/utils/tests/reactions.spec.ts

diff --git a/packages/utils/src/bucket.ts b/packages/utils/src/bucket.ts
index 05a8ba0c4..5f542cea3 100644
--- a/packages/utils/src/bucket.ts
+++ b/packages/utils/src/bucket.ts
@@ -32,10 +32,7 @@ export class LeakyBucket implements LeakyBucketOptions {
   async processQueue(): Promise<void> {
     logger.debug('[Gateway] Processing queue')
     // There is already a queue that is processing
-    if (this.processing) {
-      logger.debug('[Gateway] Queue is already processing.')
-      return
-    }
+    if (this.processing) return logger.debug('[Gateway] Queue is already processing.')
 
     // Begin going through the queue.
     while (this.queue.length) {
diff --git a/packages/utils/tests/bucket.spec.ts b/packages/utils/tests/bucket.spec.ts
index 6275306df..1a63660e3 100644
--- a/packages/utils/tests/bucket.spec.ts
+++ b/packages/utils/tests/bucket.spec.ts
@@ -62,5 +62,41 @@ describe('bucket.ts', () => {
       })
       expect(bucket.queue).to.deep.equal([])
     })
+
+    it('will acquire a request', async () => {
+      const bucket = new LeakyBucket({
+        max: 120,
+        refillInterval: 60000,
+        refillAmount: 120,
+      })
+
+      await bucket.acquire()
+      expect(bucket.remaining).to.be.equal(119)
+      expect(bucket.used).to.be.equal(1)
+    })
+
+    it('will handle multiple requests at once', async () => {
+      const bucket = new LeakyBucket({
+        max: 120,
+        refillInterval: 60000,
+        refillAmount: 120,
+      })
+
+      for (let i = 0; i < 10; i++) {
+        bucket.acquire()
+      }
+    })
+
+    it('will handle too many requests', async () => {
+      const bucket = new LeakyBucket({
+        max: 5,
+        refillInterval: 10000,
+        refillAmount: 5,
+      })
+
+      for (let i = 0; i < 10; i++) {
+        bucket.acquire()
+      }
+    })
   })
 })
diff --git a/packages/utils/tests/color.spec.ts b/packages/utils/tests/color.spec.ts
index 2b547ff8b..e99e93477 100644
--- a/packages/utils/tests/color.spec.ts
+++ b/packages/utils/tests/color.spec.ts
@@ -47,4 +47,18 @@ describe('color.ts', () => {
         })
       })
     })
+
+  describe('get and set color enabled', () => {
+    it('by default color should be on', () => {
+      expect(colors.getColorEnabled()).to.equal(true)
+    })
+
+    it('Set it off and on', () => {
+      expect(colors.getColorEnabled()).to.equal(true)
+      colors.setColorEnabled(false)
+      expect(colors.getColorEnabled()).to.equal(false)
+      colors.setColorEnabled(true)
+      expect(colors.getColorEnabled()).to.equal(true)
+    })
+  })
 })
diff --git a/packages/utils/tests/logger.spec.ts b/packages/utils/tests/logger.spec.ts
new file mode 100644
index 000000000..f3680abc8
--- /dev/null
+++ b/packages/utils/tests/logger.spec.ts
@@ -0,0 +1,28 @@
+import { describe, it } from 'mocha'
+import { LogDepth, LogLevels, createLogger } from '../src/logger.js'
+import { expect } from 'chai'
+
+describe('Logger', () => {
+  it('create logger with default options', () => {
+    const loggy = createLogger()
+    loggy.setLevel(LogLevels.Debug)
+    loggy.setDepth(LogDepth.Full)
+
+    loggy.debug('debugging')
+    loggy.error('error')
+    loggy.fatal('fatal')
+    loggy.info('info')
+    loggy.warn('warn')
+  })
+  
+  it('create logger with a name', () => {
+    const loggy = createLogger({ name: 'loggy' })
+    expect(loggy).to.exist
+  })
+
+  it('Handle fake level', () => {
+    const loggy = createLogger({ name: 'fake level' })
+    // @ts-expect-error intentionally high value for level to bypass ts check and run the `default` handler
+    loggy.log(123, 'idk')
+  })
+})
diff --git a/packages/utils/tests/reactions.spec.ts b/packages/utils/tests/reactions.spec.ts
new file mode 100644
index 000000000..0c3c90cba
--- /dev/null
+++ b/packages/utils/tests/reactions.spec.ts
@@ -0,0 +1,20 @@
+import { expect } from 'chai'
+import { describe, it } from 'mocha'
+import { processReactionString } from '../src/reactions.js'
+
+describe('Reactions', () => {
+  it('Convert a unicode emoji to discord form', () => {
+    const reaction = processReactionString('😄')
+    expect(reaction).to.be.equal('😄')
+  })
+
+  it('Convert a custom emoji to discord form', () => {
+    const reaction = processReactionString('<:discordeno:785403373817823272>')
+    expect(reaction).to.be.equal('discordeno:785403373817823272')
+  })
+
+  it('Convert an animated custom emoji to discord form', () => {
+    const reaction = processReactionString('<a:discordeno:785403373817823272>')
+    expect(reaction).to.be.equal('discordeno:785403373817823272')
+  })
+})