aboutsummaryrefslogtreecommitdiff
path: root/packages/taler-wallet-core/src/crypto/workers/crypto-dispatcher.test.ts
blob: d8d53a839872b471369890e202bae23198ccb3dc (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
/*
 This file is part of GNU Taler
 (C) 2023 Taler Systems S.A.

 GNU Taler is free software; you can redistribute it and/or modify it under the
 terms of the GNU General Public License as published by the Free Software
 Foundation; either version 3, or (at your option) any later version.

 GNU Taler is distributed in the hope that it will be useful, but WITHOUT ANY
 WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
 A PARTICULAR PURPOSE.  See the GNU General Public License for more details.

 You should have received a copy of the GNU General Public License along with
 GNU Taler; see the file COPYING.  If not, see <http://www.gnu.org/licenses/>
 */

import { AbsoluteTime } from "@gnu-taler/taler-util";
import test from "ava";
import { CryptoDispatcher, CryptoWorkerFactory } from "./crypto-dispatcher.js";
import {
  CryptoWorker,
  CryptoWorkerResponseMessage,
} from "./cryptoWorkerInterface.js";
import { SynchronousCryptoWorkerFactoryNode } from "./synchronousWorkerFactoryNode.js";
import { processRequestWithImpl } from "./worker-common.js";

export class MyCryptoWorker implements CryptoWorker {
  /**
   * Function to be called when we receive a message from the worker thread.
   */
  onmessage: undefined | ((m: any) => void) = undefined;

  /**
   * Function to be called when we receive an error from the worker thread.
   */
  onerror: undefined | ((m: any) => void) = undefined;

  /**
   * Add an event listener for either an "error" or "message" event.
   */
  addEventListener(event: "message" | "error", fn: (x: any) => void): void {
    switch (event) {
      case "message":
        this.onmessage = fn;
        break;
      case "error":
        this.onerror = fn;
        break;
    }
  }

  private dispatchMessage(msg: any): void {
    if (this.onmessage) {
      this.onmessage(msg);
    }
  }

  /**
   * Send a message to the worker thread.
   */
  postMessage(msg: any): void {
    const handleRequest = async () => {
      let responseMsg: CryptoWorkerResponseMessage;
      if (msg.operation === "testSuccess") {
        responseMsg = {
          id: msg.id,
          type: "success",
          result: {
            testResult: 42,
          },
        };
      } else if (msg.operation === "testError") {
        responseMsg = {
          id: msg.id,
          type: "error",
          error: {
            code: 42,
            when: AbsoluteTime.now(),
            hint: "bla",
          },
        };
      } else if (msg.operation === "testTimeout") {
        // Don't respond
        return;
      }
      try {
        setTimeout(() => this.dispatchMessage(responseMsg), 0);
      } catch (e) {
        console.error("got error during dispatch", e);
      }
    };
    handleRequest().catch((e) => {
      console.error("Error while handling crypto request:", e);
    });
  }

  /**
   * Forcibly terminate the worker thread.
   */
  terminate(): void {
    // This is a no-op.
  }
}

export class MyCryptoWorkerFactory implements CryptoWorkerFactory {
  startWorker(): CryptoWorker {
    return new MyCryptoWorker();
  }

  getConcurrency(): number {
    return 1;
  }
}

test("continues after error", async (t) => {
  const cryptoDisp = new CryptoDispatcher(new MyCryptoWorkerFactory());
  const resp1 = await cryptoDisp.doRpc("testSuccess", 0, {});
  t.assert((resp1 as any).testResult === 42);
  const exc = await t.throwsAsync(async () => {
    const resp2 = await cryptoDisp.doRpc("testError", 0, {});
  });

  // Check that it still works after one error.
  const resp2 = await cryptoDisp.doRpc("testSuccess", 0, {});
  t.assert((resp2 as any).testResult === 42);

  // Check that it still works after timeout.
  const resp3 = await cryptoDisp.doRpc("testSuccess", 0, {});
  t.assert((resp3 as any).testResult === 42);
});