test: Scheduler

3 files changed, 65 insertions, 2 deletions

diff --git a/.mocharc.json b/.mocharc.json
index dbf2db2..6511ecb 100644
--- a/.mocharc.json
+++ b/.mocharc.json
@@ -3,5 +3,6 @@
     "chai/register-should",
     "ts-node/register",
     "test/utils/dbConnection.ts"
-  ]
+  ],
+  "timeout": 30000
 }
diff --git a/test/model.ts b/test/model.ts
index 4d189dc..f8f76a6 100644
--- a/test/model.ts
+++ b/test/model.ts
@@ -1,10 +1,12 @@
 import { Schema } from 'mongoose';
-import model from '../lib/event.model';
+import model, { Event } from '../lib/event.model';
 
 interface Context {
   message: string;
 }
 
+export type CustomEvent = Event<Context>;
+
 const contextSchema = new Schema({
   message: String
 });
diff --git a/test/scheduler.test.ts b/test/scheduler.test.ts
new file mode 100644
index 0000000..ffcd623
--- /dev/null
+++ b/test/scheduler.test.ts
@@ -0,0 +1,60 @@
+import { expect } from 'chai';
+import Scheduler from '../lib/scheduler';
+import Model, { CustomEvent } from './model';
+import connection from './utils/dbConnection';
+
+const sleep = async (time: number) => new Promise<void>(res => setTimeout(res, time));
+
+describe('Scheduler', async () => {
+  let event: CustomEvent;
+  let scheduler: Scheduler;
+
+  beforeEach(() => {
+    scheduler = new Scheduler(Model);
+  });
+
+  afterEach(() => {
+    scheduler.stopPolling();
+  });
+
+  before(async () => {
+    event = await Model.create({
+      type: 'test',
+      schedule: '*/15 * * * * *', // Every 15 seconds
+      context: {
+        message: 'Hello, world!'
+      },
+    });
+  });
+
+  after(async () => {
+    connection.dropCollection('customevents');
+  });
+
+
+  it('Should run event', async () => {
+    // Wait until job is run
+    await new Promise<void>(res => {
+      scheduler.registerHandler('test', async (event: CustomEvent) => {
+        await sleep(2000);
+        await event.log(event.context.message)
+        res();
+      });
+    });
+
+    // Wait for status to change
+    await sleep(100);
+
+    const updatedEvent = await Model.findOne();
+    expect(updatedEvent).to.have.property('status').equal('complete');
+  });
+
+
+  it('Should fail an event if no handler is present', async () => {
+    await sleep(25000);
+
+    const updatedEvent = await Model.findOne();
+    expect(updatedEvent).to.have.property('status').equal('failed');
+    expect(updatedEvent).to.have.property('error').equal('Error: No handler found');
+  });
+});