Merge pull request #8 from machinefi/add-custom-event-type

allow emit msgs with custom events

README.md

@@ -87,7 +87,9 @@ const message = simulator.generateEvents(numberOfDataPoints);
 To send a single message to the server, call the `sendSingleMessage` method:
 
 ```ts
-const { res, msg } = await simulator.sendSingleMessage();
+const eventType = "TEMPERATURE";
+
+const { res, msg } = await simulator.sendSingleMessage(eventType);
 
 console.log("response: ", res?.data);
 console.log("w3bstream message: ", msg);
@@ -96,7 +98,10 @@ console.log("w3bstream message: ", msg);
 To send messages at a specified interval, use the `powerOn` method:
 
 ```ts
-simulator.powerOn(intervalInSec);
+const intervalInSec = 1;
+const eventType = "TEMPERATURE";
+
+simulator.powerOn(intervalInSec, eventType);
 ```
 
 To stop sending messages, call the `powerOff` method:

src/Simulator/index.ts

@@ -38,12 +38,12 @@ export class Simulator {
     };
   }
 
-  powerOn(intervalInSec: number): void {
+  powerOn(intervalInSec: number, eventType?: string): void {
     const intervalInMs = intervalInSec * 1000;
 
     this._interval = setInterval(async () => {
       try {
-        await this.sendSingleMessage();
+        await this.sendSingleMessage(eventType);
       } catch (e) {
         console.log(e);
         console.log("Stopping simulator due to error");
@@ -61,14 +61,15 @@ export class Simulator {
     }
   }
 
-  async sendSingleMessage(): Promise<{
+  async sendSingleMessage(eventType?: string): Promise<{
     res: AxiosResponse | undefined;
     msg: W3bStreamMessage;
   }> {
     const message = this.generateSingleMessage();
 
     const header: WSHeader = {
       device_id: message.deviceId,
+      event_type: eventType,
     };
     const res = await this._client?.publishDirect(header, message);
 

src/Simulator/simulator.test.ts

@@ -136,7 +136,6 @@ describe("simulator", () => {
       );
     });
   });
-
   describe("simulation", () => {
     interface TemperatureDataPoint extends DataPoint {
       temperature: number;
@@ -161,6 +160,10 @@ describe("simulator", () => {
     });
     afterEach(() => {
       mockSendMsg.mockRestore();
+      fs.rmSync(path.join("./", "private.key"), { force: true });
+    });
+    it("should send a message to a custom event", async () => {
+      await simulator1.sendSingleMessage("TEMPERATURE_EVENT");
     });
     it("should power on", async () => {
       jest.useFakeTimers();
@@ -181,6 +184,21 @@ describe("simulator", () => {
 
       jest.useRealTimers();
     });
+    it("should power on with a custom event", async () => {
+      jest.useFakeTimers();
+
+      const eventType = "TEMPERATURE_EVENT";
+      simulator1.powerOn(1, eventType);
+      jest.spyOn(simulator1, "sendSingleMessage");
+
+      await jest.advanceTimersByTimeAsync(1_000);
+
+      expect(simulator1.sendSingleMessage).toHaveBeenCalledTimes(1);
+      expect(simulator1.sendSingleMessage).toHaveBeenCalledWith(eventType);
+
+      simulator1.powerOff();
+      jest.useRealTimers();
+    });
     it("should power off if msg sending fails", async () => {
       jest.useFakeTimers();
 
@@ -196,7 +214,6 @@ describe("simulator", () => {
 
       jest.useRealTimers();
     });
-
   });
 });