Skip to content

Commit 344e604

Browse files
committed
fix vm tester for Node runtime
1 parent a178d84 commit 344e604

File tree

1 file changed

+6
-5
lines changed

1 file changed

+6
-5
lines changed

test/lib/vm.spec.ts

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,8 @@
11
import { expect } from 'chai';
22
import { Context, Request } from '../../src/common';
3-
import { VmRunner } from '../../src/lib/vm';
4-
import { wrapCode, wrapCodeBrowser } from '../../src/plugins/built-in';
3+
import { VmRunner as VmRunnerBrowser } from '../../src/lib/vm';
4+
import { VmRunner as VmRunnerNode } from '../../src/lib/vm2';
5+
import { wrapCode as wrapCodeNode, wrapCodeBrowser } from '../../src/plugins/built-in';
56
import * as utils from '../../src/lib/utils';
67
import { Logger } from '../../src/lib/logger';
78

@@ -14,7 +15,7 @@ describe('VmRunner', () => {
1415
const req = new Request();
1516
const ctx = new Context();
1617
const logger = new Logger('TESTER');
17-
const vPreProcess = await VmRunner.run(vCode, { req, ctx, utils, logger });
18+
const vPreProcess = await VmRunnerBrowser.run(vCode, { req, ctx, utils, logger });
1819
const vPostProcessingCallback = await vPreProcess();
1920
expect(vPreProcess).not.eq(undefined);
2021
expect(req.message).not.eq(123);
@@ -26,12 +27,12 @@ describe('VmRunner', () => {
2627

2728
describe('VM for Node', () => {
2829
it('should run the code', async () => {
29-
const vCode = wrapCodeBrowser(`logger.info('Ok'); return () => {req.message = 123}`);
30+
const vCode = wrapCodeNode(`logger.info('Ok'); return () => {req.message = 123}`);
3031

3132
const req = new Request();
3233
const ctx = new Context();
3334
const logger = new Logger('TESTER');
34-
const vPreProcess = await VmRunner.run(vCode, { req, ctx, utils, logger });
35+
const vPreProcess = await VmRunnerNode.run(vCode, { req, ctx, utils, logger });
3536
const vPostProcessingCallback = await vPreProcess();
3637
expect(vPreProcess).not.eq(undefined);
3738
expect(req.message).not.eq(123);

0 commit comments

Comments
 (0)