feat(factory): add way to extend drivers

src/factories/ConnectionFactory.ts

@@ -8,6 +8,7 @@
  */
 
 import { debug } from '#src/debug'
+import type { Driver } from '#src/drivers/Driver'
 import { FakeDriver } from '#src/drivers/FakeDriver'
 import { VanillaDriver } from '#src/drivers/VanillaDriver'
 import { DatabaseDriver } from '#src/drivers/DatabaseDriver'
@@ -115,6 +116,23 @@ export class ConnectionFactory {
     return availableConnections
   }
 
+  /**
+   * Define your own queue driver implementation to use
+   * within Queue facade.
+   *
+   * @example
+   * ```ts
+   * import { Driver } from '@athenna/queue'
+   *
+   * class TestDriver extends Driver {}
+   *
+   * ConnectionFactory.createDriver('test', TestDriver)
+   * ```
+   */
+  public static createDriver(name: string, impl: typeof Driver) {
+    this.drivers.set(name, impl)
+  }
+
   /**
    * Parse connection config name if is default
    */

tests/fixtures/config/queue.ts

@@ -63,6 +63,10 @@ export default {
 
     'not-found': {
       driver: 'not-found'
+    },
+
+    test: {
+      driver: 'test'
     }
   }
 }

tests/fixtures/consoles/console-mock-dest-import.ts

@@ -1,5 +1,5 @@
 /**
- * @athenna/validator
+ * @athenna/queue
  *
  * (c) João Lenon <[email protected]>
  *

tests/fixtures/constants/products.ts

@@ -1,5 +1,5 @@
 /**
- * @athenna/worker
+ * @athenna/queue
  *
  * (c) João Lenon <[email protected]>
  *

tests/unit/workers/BaseWorkerTest.ts → tests/fixtures/drivers/TestDriver.ts

@@ -6,3 +6,7 @@
  * For the full copyright and license information, please view the LICENSE
  * file that was distributed with this source code.
  */
+
+import { VanillaDriver } from '#src'
+
+export class TestDriver extends VanillaDriver {}

tests/fixtures/workers/AnnotatedWorker.ts

@@ -1,5 +1,5 @@
 /**
- * @athenna/worker
+ * @athenna/queue
  *
  * (c) João Lenon <[email protected]>
  *

tests/unit/factories/ConnectionFactoryTest.ts

@@ -8,6 +8,7 @@
  */
 
 import { Path } from '@athenna/common'
+import { TestDriver } from '#tests/fixtures/drivers/TestDriver'
 import { AfterEach, BeforeEach, Test, type Context } from '@athenna/test'
 import { NotFoundDriverException } from '#src/exceptions/NotFoundDriverException'
 import { ConnectionFactory, FakeDriver, DatabaseDriver, VanillaDriver } from '#src'
@@ -18,6 +19,7 @@ export class ConnectionFactoryTest {
   public async beforeEach() {
     await Config.loadAll(Path.fixtures('config'))
     ConnectionFactory.connections = new Map()
+    ConnectionFactory.drivers.delete('test')
   }
 
   @AfterEach()
@@ -78,4 +80,16 @@ export class ConnectionFactoryTest {
   public async shouldThrowNotImplementedConfigExceptionWhenTryingToUseANotImplementedDriver({ assert }: Context) {
     assert.throws(() => ConnectionFactory.fabricate('not-found-con'), NotImplementedConfigException)
   }
+
+  @Test()
+  public async shouldBeAbleToCreateOwnDriverImplementationToUseWithinQueueFacade({ assert }: Context) {
+    ConnectionFactory.createDriver('test', TestDriver)
+
+    const testDriver = ConnectionFactory.fabricate('test')
+
+    assert.instanceOf(testDriver, TestDriver)
+
+    ConnectionFactory.drivers.delete('test')
+    ConnectionFactory.connections.delete('test')
+  }
 }