Support for wasTerminated

jest/__mocks__/request.js

@@ -4,6 +4,7 @@ import CacheKey from './cacheKey';
 import { ReadableStream } from './streams'
 
 export const mockRespondWith = jest.fn();
+export const mockWasTerminated = jest.fn();
 export const mockGetHeader = jest.fn();
 export const mockSetHeader = jest.fn();
 export const mockAddHeader = jest.fn();
@@ -29,6 +30,7 @@ const Request = jest.fn().mockImplementation(() => {
     cpCode: 1191398,
     cacheKey: new CacheKey(),
     respondWith: mockRespondWith,
+    wasTerminated: mockWasTerminated,
     getHeader: mockGetHeader,
     setHeader: mockSetHeader,
     addHeader: mockAddHeader,
@@ -41,7 +43,7 @@ const Request = jest.fn().mockImplementation(() => {
     text: mockText,
     arrayBuffer: mockArrayBuffer,
     body: new ReadableStream()
-    };
+  };
 });
 
 export default Request;

jest/edgeworkers/examples/respond-from-edgeworkers/respondwith/was-terminated/bundle.json

@@ -0,0 +1,4 @@
+{
+    "edgeworker-version": "0.1",
+    "description" : "Returns true if respondWith() has been called in the current handler and false otherwise."
+  }

jest/edgeworkers/examples/respond-from-edgeworkers/respondwith/was-terminated/main.js

@@ -0,0 +1,5 @@
+export function onClientResponse(request, response) {
+  request.respondWith(333, {}, "");
+  request.wasTerminated();
+  request.respondWith(222, {}, "GOOD: wasTerminated worked!");
+}

jest/package.json

@@ -1,6 +1,6 @@
 {
   "name": "edgeworkers-jest-mocks",
-  "version": "1.0.24",
+  "version": "1.0.25",
   "description": "Akamai EdgeWorkers Jest mocks",
   "dependencies": {
     "@babel/core": "^7.17.8",

jest/test/examples/was-terminated/main.test.js

@@ -0,0 +1,28 @@
+import { onClientResponse } from "../../../edgeworkers/examples/respond-from-edgeworkers/respondwith/was-terminated/main";
+import Request from "request";
+
+describe("demonstrates how an EdgeWorker can be used to see if respondWith() has been called.", () => {
+  beforeEach(() => {
+    jest.clearAllMocks();
+  });
+
+  test("Simple onClientResponse call to check that wasTerminated was called", () => {
+    let requestMock = new Request();
+    requestMock.mockWasTerminated = true;
+
+    onClientResponse(requestMock);
+    expect(requestMock.respondWith).toHaveBeenCalledTimes(2);
+    expect(requestMock.wasTerminated).toHaveBeenCalledTimes(1);
+    expect(requestMock.mockWasTerminated).toBeTruthy();
+  });
+
+  test("Simple onClientResponse call to check that wasTerminated was not called", () => {
+    let requestMock = new Request();
+    requestMock.mockWasTerminated = false;
+
+    onClientResponse(requestMock);
+    expect(requestMock.respondWith).toHaveBeenCalledTimes(2);
+    expect(requestMock.wasTerminated).toHaveBeenCalledTimes(1);
+    expect(requestMock.mockWasTerminated).not.toBeTruthy();
+  });
+});

mocha/__mocks__/request.js

@@ -6,6 +6,7 @@ import CacheKey from './cacheKey';
 import { ReadableStream } from './streams'
 
 export const mockRespondWith = sinon.stub();
+export const mockWasTerminated = sinon.stub();
 export const mockGetHeader = sinon.stub();
 export const mockSetHeader = sinon.stub();
 export const mockAddHeader = sinon.stub();
@@ -31,6 +32,7 @@ export default class Request {
     this.cpCode = 1191398;
     this.cacheKey = new CacheKey();
     this.respondWith = mockRespondWith;
+    this.wasTerminated = mockWasTerminated;
     this.getHeader = mockGetHeader;
     this.setHeader = mockSetHeader;
     this.addHeader = mockAddHeader;

mocha/edgeworkers/examples/respond-from-edgeworkers/respondwith/was-terminated/bundle.json

@@ -0,0 +1,4 @@
+{
+    "edgeworker-version": "0.1",
+    "description" : "Returns true if respondWith() has been called in the current handler and false otherwise."
+  }

mocha/edgeworkers/examples/respond-from-edgeworkers/respondwith/was-terminated/main.js

@@ -0,0 +1,5 @@
+export function onClientResponse(request, response) {
+  request.respondWith(333, {}, "");
+  request.wasTerminated();
+  request.respondWith(222, {}, "GOOD: wasTerminated worked!");
+}

mocha/package.json

@@ -1,6 +1,6 @@
 {
   "name": "edgeworkers-mocha-mocks",
-  "version": "1.0.24",
+  "version": "1.0.25",
   "description": "Akamai EdgeWorkers Mocha mocks",
   "dependencies": {
     "@babel/core": "^7.17.8",

mocha/test/examples/was-terminated/main.test.js

@@ -0,0 +1,31 @@
+import { onClientResponse } from "../../../edgeworkers/examples/respond-from-edgeworkers/respondwith/was-terminated/main";
+import Request from "request";
+
+const sinon = require("sinon");
+const expect = require('expect.js');
+
+describe("demonstrates how an EdgeWorker can be used to see if respondWith() has been called.", () => {
+  afterEach(function () {
+    sinon.reset();
+  });
+
+  it("Simple onClientResponse call to check that wasTerminated was called", () => {
+    let requestMock = new Request();
+    requestMock.mockWasTerminated = true;
+
+    onClientResponse(requestMock);
+    expect(requestMock.respondWith.callCount).to.be(2);
+    expect(requestMock.wasTerminated.callCount).to.be(1);
+    expect(requestMock.mockWasTerminated).to.be.true;
+  });
+
+  it("Simple onClientResponse call to check that wasTerminated was not called", () => {
+    let requestMock = new Request();
+    requestMock.mockWasTerminated = false;
+
+    onClientResponse(requestMock);
+    expect(requestMock.respondWith.callCount).to.be(2);
+    expect(requestMock.wasTerminated.callCount).to.be(1);
+    expect(requestMock.mockWasTerminated).to.be.false;
+  });
+});